Source-linked AI summary
CXPlain: Causal Explanations for Model Interpretation under Uncertainty
Patrick Schwab, Walter Karlen
TL;DR
Fast, accurate, and uncertainty-aware feature-importance estimation remains challenging for diverse, high-dimensional machine-learning models. CXPlain treats explanation as causal learning by training a separate model to explain a target model, then uses bootstrap ensembles for uncertainty. Experiments report higher accuracy and faster evaluation than existing model-agnostic methods, with uncertainty estimates correlated with held-out estimation accuracy.
Problem
Fast and accurate feature-importance estimation and uncertainty quantification remain challenging across diverse machine-learning models and high-dimensional data.
Method
CXPlain trains a separate explanation model with an adapted causal objective to estimate feature importance for any predictive model, using bootstrap resampling for uncertainty estimates.
Results
CXPlain was significantly more accurate than leading model-agnostic methods on MNIST and ImageNet, faster than LIME and SHAP, and produced uncertainty estimates correlated with held-out accuracy.
Takeaways & Limitations
CXPlain provides fast post-training explanations and uncertainty estimates that can help assess the reliability of feature-importance scores.
Takeaways & Limitations
CXPlain requires a separate training stage, and its causal associations concern marginal predictive-performance improvements rather than real-world causality between inputs and outputs.
Abstract
from arXiv · showhide
Feature importance estimates that inform users about the degree to which given inputs influence the output of a predictive model are crucial for understanding, validating, and interpreting machine-learning models. However, providing fast and accurate estimates of feature importance for high-dimensional data, and quantifying the uncertainty of such estimates remain open challenges. Here, we frame the task of providing explanations for the decisions of machine-learning models as a causal learning task, and train causal explanation (CXPlain) models that learn to estimate to what degree certain inputs cause outputs in another machine-learning model. CXPlain can, once trained, be used to explain the target model in little time, and enables the quantification of the uncertainty associated with its feature importance estimates via bootstrap ensembling. We present experiments that demonstrate that CXPlain is significantly more accurate and faster than existing model-agnostic methods for estimating feature importance. In addition, we confirm that the uncertainty estimates provided by CXPlain ensembles are strongly correlated with their ability to accurately estimate feature importance on held-out data.
1 Introduction
Feature-importance explanations help users understand and validate model decisions, but fast, accurate, and uncertainty-aware attribution remains difficult for complex and high-dimensional models. CXPlain addresses this challenge with causal explanation models and reports improved accuracy, speed, and uncertainty estimation.
- Feature-importance estimates help users understand, validate, and interpret predictive-model decisions.
- Fast and accurate attribution is difficult because machine-learning models use varied structures, algorithms, and decision functions, while existing estimates often carry substantial uncertainty.
- CXPlain trains a supervised explanation model with a causal objective to estimate feature importance for another model without requiring changes to the predictive model.
- CXPlain combines bootstrap-based uncertainty estimation with experiments showing greater accuracy and faster evaluation than existing model-agnostic methods.
2 Related Work
Existing explanation methods differ in model compatibility and reliability, and uncertainty in their attributions can make explanations difficult to trust. CXPlain is presented as a causal, model-agnostic explanation model designed to address these limitations while enabling uncertainty estimation and fast evaluation.
- Existing feature-importance methods include gradient-based, sensitivity-based, feature-removal, and mimic-model approaches.
- Explanation reliability is understudied, although attribution methods can be sensitive to input shifts and produce substantially uncertain estimates.
- Table 1 compares CXPlain with representative feature-importance estimation methods.
- CXPlain learns to explain any machine-learning model with a causal objective, without retraining, adapting, or requiring detailed knowledge of the explained model.
3 Methodology
CXPlain reframes feature-importance estimation as supervised causal explanation: a separate model learns to explain any predictive model without retraining or adapting it. Its objective derives relative feature contributions from prediction errors, while bootstrap ensembles quantify uncertainty.
- Causal Explanations (CXPlain): CXPlain trains a separate explanation model to estimate the importance scores of a target predictive model from input features.The framework leaves the predictive model unchanged and learns explanations using supervised training.
- Causal Objective: CXPlain computes predictive errors with and without each feature by comparing target-model outputs for the full and feature-masked inputs.Masking can use zeroes, mean replacement, or more distribution-aware schemes depending on the input data.
- Causal Objective: The causal objective measures each feature’s contribution through the decrease in predictive error associated with adding that feature to the available information.This adapts a Granger-style causal definition to feature-importance estimation.
- Causal Objective: The resulting feature contributions are normalised into relative importance scores between 0 and 1 that sum to 1.These scores form the target importance distribution used to train the explanation model.
- Causal Objective: Training minimises the Kullback–Leibler divergence between target importance distributions and the explanation model’s predicted importance distributions.Precomputing target importances for each training sample requires N(p + 1) evaluations of the predictive model.
- Uncertainty of Importance Estimates: Bootstrap ensembles quantify uncertainty by using confidence-interval width for each feature’s importance estimate.Each ensemble member is trained on a bootstrap-resampled training set, and confidence-interval quantiles provide lower and upper bounds.
4 Experiments
The experiments compare CXPlain with existing methods on feature-importance accuracy, computation time, and uncertainty calibration across image benchmarks. They evaluate masked-image behavior, attribution maps, runtime, and correlations between uncertainty and held-out ranking errors.
- Experimental design: The experiments compare CXPlain with state-of-the-art feature-importance methods on MNIST and ImageNet image-classification benchmarks.They focus mainly on image classification and neural-network models to visualize and quantify estimation performance.
- Feature-importance evaluation: CXPlain (U-net) attribution maps are reported as more semantically focused on salient image regions than SHAP and LIME for the Gorilla-versus-Zebra benchmark.For MNIST, accurate masks should make an 8-versus-3 image more closely resemble a 3 after the pixels distinguishing an 8 are removed.
- Experimental design: Top-importance pixel masks are evaluated by removing 10% of MNIST pixels and 30% of ImageNet pixels, then measuring the resulting confidence change.The experiments also plot importance assignments and masked images for qualitative assessment.
- Uncertainty evaluation: Uncertainty calibration is assessed by correlating bootstrap uncertainty ui with per-feature rank error REi on held-out MNIST images.The uncertainty is the width between the 95th and 5th percentile bootstrap confidence limits for importance estimates among the top 2.5% of pixels by ∆log-odds.
- Computational performance: Compute time is compared in log2 seconds for N = 100 test images on identical hardware, with lower values indicating faster estimation.Figures 6 and 7 report significance using *** for p < 0.001 under the MWW test.
5 Results and Discussion
CXPlain matched or exceeded competing feature-importance methods while evaluating faster, and its bootstrap uncertainty estimates tracked explanation errors. The authors also note that CXPlain requires one-off training and that its causal interpretation is limited to predictive-model performance, not real-world causality.
- Predictive Performance: CXPlain was competitive with DeepSHAP on MNIST and outperformed LIME on ImageNet.On MNIST, the U-net variant was competitive with DeepSHAP; on ImageNet, it significantly outperformed LIME.
- Computational Performance: CXPlain computed feature-importance estimates significantly faster than LIME and SHAP on MNIST and ImageNet.The speed gap was considerably larger on ImageNet because higher-dimensional images made LIME and SHAP’s many model evaluations slower.
- Quality of Uncertainty Estimates: M = 5 bootstrap models produced uncertainty estimates significantly correlated with feature-importance accuracy on N = 100 unseen images.Increasing the ensemble from M = 5 to M = 100 significantly increased this correlation, while visual uncertainty patterns resembled rank errors.
- Limitations: CXPlain’s one-off training requirement is a limitation, although the cost typically amortises through faster evaluation.The authors also restrict its causal interpretation to marginal improvements in the predictive model’s performance.
- Limitations: CXPlain’s reported associations do not indicate real-world causal relationships between a model’s inputs and outputs.Its causal objective concerns the degree to which input features improve the explained predictive model’s performance.
6 Conclusion
The conclusion presents CXPlain as a model-agnostic explanation model that produces fast feature-importance estimates and bootstrap uncertainty estimates. Experiments on MNIST and ImageNet found higher accuracy than existing model-agnostic methods and much faster evaluation.
- CXPlain trains a separate explanation model with a causal objective to estimate feature importance for a target predictive model.
- CXPlain is compatible with any machine-learning model, produces importance estimates quickly after training, and supports bootstrap uncertainty estimates.
- CXPlain was significantly more accurate than existing model-agnostic methods on MNIST and ImageNet while providing importance estimates orders of magnitude faster.
Supplementary Material for: "CXPlain: Causal Explanations for Model
The supplementary material identifies the paper as “CXPlain: Causal Explanations for Model Interpretation under Uncertainty” and lists Patrick Schwab and Walter Karlen as authors from ETH Zurich.
- The supplementary material is for “CXPlain: Causal Explanations for Model Interpretation under Uncertainty.”
- The listed authors are Patrick Schwab and Walter Karlen.
- The authors are affiliated with the Institute of Robotics and Intelligent Systems at ETH Zurich.
A Hyperparameters
The experiments used Python and TensorFlow, standardized hardware, image benchmarks with ResNet targets, and a Twitter sentiment benchmark to test CXPlain across modalities and model types.
- Experiments used Python, TensorFlow, standardized compute hardware, and 10000 perturbed samples per image for LIME and SHAP.
- MNIST Benchmark: The MNIST target was a binary ResNet-20 classifier distinguishing digits 8 and 3 with 99.85% test accuracy.
- ImageNet and Twitter Benchmarks: The ImageNet target was a binary ResNet-32 classifier with 96.73% test accuracy, while Twitter experiments used 100000 balanced short messages.
B Training Bootstrap Ensembles of Causal Explanation Models
The bootstrap ensemble trains multiple causal explanation models on resampled training data, producing an ensemble for estimating feature importance.
- Algorithm 1 trains an ensemble of M causal explanation models for a target predictive model.
- Each ensemble model uses a bootstrap sample containing N training pairs drawn with replacement from the original dataset.
- Every explanation model is trained until convergence with the causal loss before being added to the ensemble.
C Computing Infrastructure
All experiments used the same specified computing hardware, including an Intel processor, an Nvidia GPU, and 32 GB of RAM.
- All experiments used an Intel Core i5 7600K processor.
- All experiments used an Nvidia GeForce Titan Xp GPU.
- The experimental system had 32 GB of RAM.
D Qualitative Samples for the MNIST and ImageNet Benchmarks
Additional qualitative samples are presented for the MNIST and ImageNet benchmarks to illustrate feature-importance outputs.
- MNIST: Figure S1 provides additional qualitative samples from the MNIST benchmark.
- ImageNet: Figure S2 provides additional qualitative samples from the ImageNet benchmark.
- The supplementary figures extend the qualitative benchmark examples beyond the main presentation.
E Qualitative Samples for the Twitter Sentiment Analysis Benchmark
Supplementary materials show qualitative feature-importance examples for Twitter sentiment analysis, including semantically meaningful word-level assignments.
- Qualitative Twitter examples: CXPlain (MLP) assigns high importance to words with positive or negative connotations in short Twitter messages.The explanations were subjectively judged to appear semantically meaningful.
- Supplementary configurations: The supplementary material also includes hyperparameter-range tables for CXPlain models across the MNIST, ImageNet, and sentiment benchmarks.
- Cross-method comparisons: Additional visual comparisons for MNIST and ImageNet show source images alongside masks or attribution scores from multiple explanation methods.
- Qualitative Twitter examples: Table S5 presents examples of short messages and the importances assigned by CXPlain (MLP).
- Table S5 encoding: Table S5 uses deeper colors to indicate higher importance scores.
- Table S5 encoding: All messages shown in Table S5 are labeled as positive in sentiment.