Smooth MPRT (sMPRT)
- sMPRT is an adaptation of MPRT that denoises gradient attributions by averaging multiple input perturbations to reduce high-frequency noise.
- It employs progressive model parameter randomisation with similarity metrics like SSIM to reliably assess explanation sensitivity.
- Empirical results demonstrate that sMPRT stabilizes explanation rankings and reduces variance, enhancing robust offline benchmarking.
Smooth Model Parameter Randomisation Test (sMPRT) is an adaptation of the Model Parameter Randomisation Test (MPRT) for the evaluation of local explanation methods in deep learning, designed to mitigate the confounding impact of “shattering” noise in gradient-based attributions and to enable more reliable, variance-reduced assessment of explanation faithfulness. sMPRT achieves this by averaging attributions over multiple i.i.d. input perturbations before calculating similarity under progressive model parameter randomisation, thereby restoring the intended sensitivity-to-parameters principle while sharply reducing noise artifacts (Hedström et al., 2024, Hedström et al., 2024).
1. Motivation and Problem Statement
The original MPRT operates by progressively randomising the parameters of a model and measuring, via a similarity metric , how much the corresponding explanation function changes. However, gradient-based explanations (e.g., Saliency, Integrated Gradients) commonly exhibit high-frequency, spatially uncorrelated noise, known as “gradient shattering.” This noise can artificially reduce the measured similarity between explanations of the original and randomised models, thereby confounding the interpretability of the MPRT and inflating/reducing differences between explanation methods, especially when similarity metrics such as SSIM or Spearman rank are highly sensitive to uncorrelated input perturbations. sMPRT is introduced to denoise the explanation maps before similarity calculation, leveraging the statistical variance reduction of averaging explanations over a set of Gaussian-perturbed inputs, in the spirit of SmoothGrad.
2. Formal Definition and Mathematical Formulation
Let denote a differentiable classifier with parameters , and let denote a local explanation (typically an attribution or saliency map) for input and target , possibly with method-specific hyperparameters . Let denote the model with parameters randomised up to layer 0 (bottom-up randomisation), and 1 the number of input perturbation samples.
For each 2: 3 Let 4 and 5. The denoised explanations are: 6 The sMPRT score at layer 7 is defined as: 8 Alternatively, one may aggregate similarity before smoothing: 9 0 is typically the Structural Similarity Index (SSIM), but may also be Spearman correlation or other similarity metrics. Per-feature normalisation by the root-mean-square of the attributions is recommended.
3. Algorithmic Implementation
The sMPRT procedure for a given input–target pair 1, model 2, and randomisation levels 3 is as follows:
7 sMPRT is available as a two-stage “sample-and-average” procedure and implemented in the Quantus XAI evaluation library (Hedström et al., 2024).
4. Theoretical Underpinnings and Noise Reduction
Averaging 4 i.i.d. noisy attributions reduces variance by 5, as per the Central Limit Theorem. The consequence is that high-frequency, layerwise-uncorrelated “shattering noise” – prevalent in single-pass gradient methods – is suppressed, and the similarity measurement across model randomisations more faithfully reflects true model parameter sensitivity rather than coincidental noise. This denoising effect is mathematically identical to SmoothGrad, which adds i.i.d. input noise to reduce explanation variance. sMPRT thereby provides lower-variance, more stable metric values, enabling robust ranking of explanation methods even when similarity metrics are highly sensitive to uncorrelated perturbations (Hedström et al., 2024, Hedström et al., 2024).
A plausible implication is that variance-induced instability can be transformed into reliable performance differences under sMPRT, meaning methods previously seen as “noise-robust” due to shattering insensitivity may be detected as artificially over-performing.
5. Empirical Results and Performance Analysis
Experiments in (Hedström et al., 2024) and (Hedström et al., 2024) evaluated sMPRT against MPRT and eMPRT across datasets and explanation methods:
Datasets and Setup
- ImageNet with ResNet-18, VGG-16 (pretrained; 300 samples)
- MNIST, fMNIST with LeNet (1000 samples)
- Explanation methods include Gradient, Saliency, Input×Gradient, GradCAM, GradientSHAP, SmoothGrad, IntegratedGradients, GuidedBackprop, LRP-ε, LRP-z+, and random baseline
- Similarity metric: SSIM with feature-wise RMS normalisation
Principal Observations
- Layer-wise SSIM curves: With MPRT (6), similarity can be artificially decreased for gradient-based methods by high-frequency noise. Under sMPRT (7), these methods (e.g. Saliency, SmoothGrad) show lower similarity, revealing previously hidden sensitivity to parameter randomisation. For noise-invariant methods (LRP-8, Guided Backprop), the difference is diminished, and scores converge across explanation types.
- Area Under Curve (AUC) in similarity-versus-layer plots stabilises around 9; increasing 0 further provides minimal gain in stability.
- Meta-evaluation (MetaQuantus protocol): Across four tasks and method subsets, sMPRT shows higher meta-consistency scores (1MC 2–3 above MPRT; exceeds one standard deviation of MPRT variability in many cases).
- sMPRT reduces the range of SSIM across methods (~0.38 to ~0.34 for VGG; ~0.50 to ~0.46 for ResNet), tightening method ranking distribution (Hedström et al., 2024).
| Variant | Sensitivity Principle Retained? | Hyperparameters | Cost (per input) | Direct Explanation Similarity? |
|---|---|---|---|---|
| MPRT | Yes | – | 1× | Yes |
| sMPRT | Yes (denoised) | 4, 5 | 6× | Yes |
| eMPRT | Complexity-drift only | – | ≤2× | No |
6. Practical Recommendations and Limitations
- Number of Samples (7): 8 recommended as balance between variance reduction and compute time; 9 provides partial improvement; 0 is smoother but computationally expensive. For vision, 1 is empirically validated.
- Noise Scale (2): Set to 3 as in SmoothGrad. In low-dynamic-range or highly structured domains (e.g. medical imaging, climate data), domain-specific tuning may be required.
- Similarity Metric (4): SSIM is standard; alternatives (correlation, Spearman-5) require uniform normalisation of attributions (preferably RMS).
- Compute Cost: sMPRT is approximately 6 times slower than MPRT. For sampled-explanation methods (e.g. SmoothGrad), marginal sMPRT benefit is modest; eMPRT may be more suitable.
- When Recommended: Best suited for offline benchmarking of noisy, gradient-based explanations where MPRT appears overly optimistic, and available compute is non-limiting. Useful to discern whether apparent noise-invariance is method-intrinsic or an artifact.
Trade-offs include increased computational burden and the introduction of two additional hyperparameters. Potential bias may be induced when perturbed samples leave the data manifold, especially for highly non-Gaussian or non-Euclidean input domains.
7. Comparison to Related Tests
- Original MPRT: sMPRT retains the model parameter sensitivity principle but corrects for shattering/noise artifacts, reducing variance and providing robust, interpretable rankings at a linear compute overhead. MPRT is faster but subject to high similarity-score variability.
- Efficient MPRT (eMPRT): eMPRT measures the rise in complexity (e.g. attribution histogram entropy) after randomisation, eschewing direct explanation similarity. eMPRT is efficient and hyperparameter-free, but the interpretability of its score is less direct with respect to explanation collapse.
In conclusion, sMPRT constitutes a minimal but mathematically justified extension to MPRT, providing a denoised, faithful assessment of explanation method sensitivity to model parameters while significantly improving score consistency and robustness by controlling for the obfuscating effect of high-frequency gradient noise (Hedström et al., 2024, Hedström et al., 2024).