ExplainabilityTransparencyAccountabilitycs.LG
Faithful and Efficient Explanations for Neural Networks via Neural Tangent Kernel Surrogate Models
Andrew Engel, Zhichao Wang, Natalie S. Frank, Ioana Dumitriu, Sutanay Choudhury, Anand Sarwate, Tony Chiang
May 23, 2023
10 citations
Key Takeaways
-
Surrogate models can make neural networks more interpretable: The research demonstrates that complex neural networks can be approximated using simpler "kernel machine" models that are easier to understand and explain, helping address the "black box" problem in AI systems.
-
Computational efficiency matters for practical explainability: The paper introduces new methods to calculate approximate neural tangent kernels with tunable computational costs, making it feasible for practitioners to generate explanations without requiring prohibitive computing resources—a key barrier to responsible AI adoption.
-
The "trace NTK" method shows promise for consistent explanations: Among the approximation techniques tested, the trace NTK variant performed most reliably, suggesting researchers have identified a practical approach that could be standardized for explaining neural network decisions across different applications.
-
Open-source tools democratize AI explanation: By releasing PyTorch software that implements these methods, the authors enable broader access to explanation techniques beyond specialized research labs, supporting more organizations' ability to audit and understand their AI systems.
-
Surrogate models can accurately reflect neural network behavior: The research validates that kernel machine surrogates correlate well with the original neural networks they approximate, meaning explanations generated from these simpler models are likely faithful representations of how the complex model actually makes decisions.