SmoothHess: ReLU Network Feature Interactions via Stein's Lemma
Not on the map. Why?
Abstract (arXiv)
Several recent methods for interpretability model feature interactions by looking at the Hessian of a neural network. This poses a challenge for ReLU networks, which are piecewise-linear and thus have a zero Hessian almost everywhere. We propose SmoothHess, a method of estimating second-order interactions through Stein's Lemma. In particular, we estimate the Hessian of the network convolved with a Gaussian through an efficient sampling algorithm, requiring only network gradient calls. SmoothHess is applied post-hoc, requires no modifications to the ReLU network architecture, and the extent of smoothing can be controlled explicitly. We provide a non-asymptotic bound on the sample complexity of our estimation procedure. We validate the superior ability of SmoothHess to capture interactions on benchmark datasets and a real-world medical spirometry dataset.
Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 51 of its 94 references that are in the map. First posted on arXiv 1 Nov 2023.
Builds on
51 of 94 references.
References · 51
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
- 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,070
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
- 2017Axiomatic Attribution for Deep Networkscited by 3,384
- 2021Convex optimizationcited by 2,515
- 2015Learning Deep Features for Discriminative Localizationcited by 2,436
- 2013Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Mapscited by 2,371
- 2017SmoothGrad: removing noise by adding noisecited by 1,109
- 2019Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)cited by 669
38 more
- 2017Interpretation of Neural Networks is Fragilecited by 444
- 2014On the Number of Linear Regions of Deep Neural Networkscited by 381
- 2018Why ReLU networks yield high-confidence predictions far away from the training data and how to mitigate the problemcited by 323
- 2018Regularisation of Neural Networks by Enforcing Lipschitz Continuitycited by 314
- 2018On the Robustness of Interpretability Methodscited by 314
- 2016On the Expressive Power of Deep Neural Networkscited by 306
- 2019Explanations can be manipulated and geometry is to blamecited by 216
- 2017Bounding and Counting Linear Regions of Deep Neural Networkscited by 161
- 2019Deep ReLU Networks Have Surprisingly Few Activation Patternscited by 161
- 2019The Shapley Taylor Interaction Indexcited by 143
- 2017On the Properties of the Softmax Function with Application in Game Theory and Reinforcement Learningcited by 142
- 2017Procedural Content Generation via Machine Learning (PCGML)cited by 137
- 2020Reliable Post hoc Explanations: Modeling Uncertainty in Explainabilitycited by 125
- 2019Smooth Grad-CAM++: An Enhanced Inference Level Visualization Technique for Deep Convolutional Neural Network Modelscited by 118
- 2020Explaining Explanations: Axiomatic Feature Interactions for Deep Networkscited by 103
- 2019CXPlain: Causal Explanations for Model Interpretation under Uncertaintycited by 98
- 2017Detecting Statistical Interactions from Neural Network Weightscited by 80
- 2020How does this interaction affect me? Interpretable attribution for feature interactionscited by 73
- 2021Does Your Dermatology Classifier Know What It Doesn't Know? Detecting the Long-Tail of Unseen Conditionscited by 68
- 2018Mad Max: Affine Spline Insights into Deep Learningcited by 55
- 2020Evaluations and Methods for Explanation through Robustness Analysiscited by 51
- 2021Towards the Unification and Robustness of Perturbation and Gradient Based Explanationscited by 47
- 2020Towards Robust Explanations for Deep Neural Networkscited by 43
- 2020Smoothed Geometry for Robust Attributioncited by 38
- 2020Feature Interaction Interpretability: A Case for Explaining Ad-Recommendation Systems via Neural Interaction Detectioncited by 32
- 2019Understanding Impacts of High-Order Loss Approximations and Features in Deep Learning Interpretationcited by 22
- 2020Feature Removal Is a Unifying Principle for Model Explanation Methodscited by 21
- 2019Stein's Lemma for the Reparameterization Trick with Exponential Family Mixturescited by 18
- 2021Deep ReLU Networks Preserve Expected Lengthcited by 17
- 2023Explanations of Black-Box Models based on Directional Feature Interactionscited by 15
- 2021Context-Aware Legal Citation Recommendation using Deep Learningcited by 12
- 2022Match Cutting: Finding Cuts with Smooth Visual Transitionscited by 12
- 2020Explaining Local, Global, And Higher-Order Interactions In Deep Learningcited by 8
- 2021Hessian Estimation via Stein's Identity in Black-Box Problemscited by 8
- 2019Learning Global Pairwise Interactions with Bayesian Neural Networkscited by 6
- 2015Newton-Stein Method: An optimization method for GLMs via Stein's Lemmacited by 3
- 2020Hierarchical Verification for Adversarial Robustnesscited by 3
- 2023Active Learning for Classifying 2D Grid-Based Level Completabilitycited by 2
50 most cited of 51 All 51 (JSON)
Built on by
Citing papers per year: 2024: 1; 2025: 3; 2026: 1.
5 papers.
- 2024Axiomatic Explainer Globalness via Optimal Transportcited by 2
- 2025Leveraging Machine Unlearning for Cost-Efficient Preference Alignmentcited by 2
- 2025OrdShap: Feature Position Importance for Sequential Black-Box Modelscited by 0
- 2025Feature-Function Curvature Analysis: A Geometric Framework for Explaining Differentiable Modelscited by 0
- 2026Local Stability and Gaussian Smoothing of Quantized Neural Networkscited by 0
- 2026Local Stability and Gaussian Smoothing of Quantized Neural Networks22 Jul 2026
- 2025Feature-Function Curvature Analysis: A Geometric Framework for Explaining Differentiable Models31 Oct 2025
- 2025OrdShap: Feature Position Importance for Sequential Black-Box Models16 Jul 2025
- 2025Leveraging Machine Unlearning for Cost-Efficient Preference Alignment9 Apr 2025
- 2024Axiomatic Explainer Globalness via Optimal Transport2 Nov 2024