atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

SPADE: Sparsity-Guided Debugging for Deep Neural Networks

ICML 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Sparse training from scratch · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SPADE, a sample-specific one-shot pruning preprocessing step that sparsifies a dense pretrained network per input so that existing saliency and neuron-visualization methods give more accurate, human-useful explanations of the original model.

Problem

Interpretability methods (saliency maps, neuron visualizations) are confounded by polysemantic neurons and superimposed features; existing sparsity-based fixes require retraining or alter the model's behavior, so their explanations no longer reflect the original dense model.

Approach

Treat sparsification as a per-sample interpretation preprocessing step: augment the input, then layer-by-layer one-shot prune the pretrained network (OBC or SparseGPT solvers) so each layer's sparse output matches its dense output on that sample, and run any interpretability method on the resulting trace, without retraining or changing inference on the sample.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add sample-specific SPADE pruning before saliency interpretationResNet50 saliency methods · ImageNet-1K Trojan patchesAverage AUC (%): 91.01 -> 96.55Helped
Use FastSPADE sample-specific pruning before saliency interpretationResNet50 saliency methods · ImageNet-1K Trojan patchesAverage AUC (%): 91.01 -> 94.34Helped
Add FastSPADE preprocessingResNet50 saliency methods · ImageNet-1K clean inputsInsertion AUC (%): 44.18 -> 47.20Helped
Add FastSPADE preprocessingResNet50 saliency methods · ImageNet-1K clean inputsDeletion AUC (%): 10.62 -> 9.82Helped
Preprocess the model with SPADE before neuron visualizationResNet50 feature visualizations · ImageNet-1K human evaluationOverall success: 56.7% -> 69.8%Helped
Use a linear sparsity schedule instead of searched layer sparsities (ablation)SPADE · ImageNet ResNet50 Trojan patchesAverage AUC: 95.78 -> 93.51Hurt
Tune sparsity ratios using 30 examples instead of 100 (ablation)FastSPADE · ImageNet ResNet50 Trojan patchesAverage AUC: 94.34 -> 93.97Hurt
Tune SPADE sparsity ratios using FastSPADE (ablation)SPADE · ImageNet ResNet50 Trojan patchesAverage AUC: 96.55 -> 94.58Hurt
Use sample-specific SPADE pruningResNet50 saliency methods · CelebA Trojan patchesAverage AUC: 80.71 -> 88.80Mixed
Use sample-specific SPADE pruningResNet50 saliency methods · Food-101 Trojan patchesAverage AUC: 83.54 -> 95.33Helped
Use OBC-based SPADE instead of FastSPADEMobileNet saliency methods · ImageNetAverage AUC improvement: +2.90%Helped
Use sample-specific SPADE pruningBERT Layer Integrated Gradients · SST-2DFFOT: 0.1835 -> 0.1766Hurt

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-1K CelebA Food-101 SST-2 AG news

Benchmarks: ImageNet-1K Trojan patch AUC CelebA Trojan patch AUC Food-101 Trojan patch AUC Insertion/Deletion AUC on clean ImageNet

Metrics: AUROC insertion AUC deletion AUC human agreement rate with Score-CAM

Baselines: Dense model Sparse FC Correlation-aware pruning (CAP)

Builds on: OBC SparseGPT Sparse FC Captum

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy