atlas

Research map / Architectures and efficiency / RNNs and normalization

Measuring Per-Unit Interpretability at Scale Without Humans

NeurIPS 2024 · Architectures and efficiency › RNNs and normalization · approach: Representation geometry and collapse

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 the Machine Interpretability Score (MIS), a fully automated per-unit interpretability metric for vision DNNs that replaces human 2-AFC psychophysics with DreamSim similarity, and uses it to analyze 835 models and training dynamics.

Problem

Measuring per-unit interpretability of vision DNNs currently requires costly human psychophysics experiments, which does not scale and limits interpretability research.

Approach

Replace human participants in the 2-AFC psychophysics task with a machine classifier that matches query images to positive/negative visual explanations using perceptual similarity (DreamSim), yielding a human-free score (MIS) with a single calibration parameter alpha.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use DreamSim to compute image similarityMIS on IMI per-unit annotations · IMIcorrelation with HIS: 0.80 Pearson; 0.80 SpearmanHelped
replace DreamSim with LPIPS (ablation)MIS · IMI per-unitcorrelation with HIS: 0.80 -> 0.70 Pearson; 0.80 -> 0.70 SpearmanHurt
replace DreamSim with DISTS (ablation)MIS · IMI per-unitcorrelation with HIS: 0.80 -> 0.95 Pearson; 0.80 -> 0.70 SpearmanMixed
use feature visualizations instead of dataset examples (ablation)MIS · IMI model-levelcorrelation with HIS: 0.95 Pearson; 0.88 SpearmanNo clear effect
use the MIS to select easiest and hardest units before human testingGoogLeNet and ResNet-50 · newly tested unitshuman interpretability: easiest MIS 0.99; hardest MIS 0.63 and 0.59Helped
increase classification performancevision models · ImageNet-2012average per-unit MIS: Pearson’s r = −0.5; Spearman’s r = −0.55Hurt
use LPIPS or DISTS instead of DreamSim at the model level (ablation)MIS · IMI model-levelcorrelation with HIS: LPIPS: ρs = 0.95, ρp = 0.98; DISTS: ρs = 0.96, ρp = 0.99No clear effect
replace a layer with a standard SAE or Gated SAE (ablation)GoogLeNet mixed4b_3x3 · SAE interpretabilityMIS: 91.21 %No clear effect
increase SAE sparsity weight across the tested settings (ablation)ResNet-50 layer2_2_conv2 SAE · SAE hyperparameter sensitivityMIS: 0.892; 0.908; 0.916; 0.915; 0.919; 0.918; 0.922; 0.918Mixed
increase number of MIS tasks (ablation)MIS · GoogLeNetabsolute difference from MIS using 20 tasksHelped
train ResNet-50 beyond its first epochResNet-50 · ImageNet-2012 trainingMIS: MIS increases to around 0.93 after the first epoch, before it decaysHurt
increase relative layer depthConv/Linear, BatchNorm, and LayerNorm layers · 835 vision modelsaverage per-unit MIS: Conv/Linear ρs = 0.85, ρp = 0.85; BatchNorm ρs = 0.77, ρp = 0.77; LayerNorm ρs = 0.85, ρp = 0.86Mixed

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-2012 IMI (ImageNet Mechanistic Interpretability) dataset

Benchmarks: IMI human interpretability annotations ImageNet-2012 top-1 accuracy

Metrics: Pearson correlation Spearman correlation average per-unit MIS psychophysics accuracy

Baselines: LPIPS DISTS Human interpretability scores (IMI)

Builds on: Borowski et al. 2-AFC psychophysics task IMI dataset (Zimmermann et al. 2023) DreamSim

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