atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

Energy-based Automated Model Evaluation

ICLR 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Group-robust classification · 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 Meta-Distribution Energy (MDE), a training-free AutoEval measure that normalizes per-sample energies over an unlabeled test set and linearly regresses accuracy from it, outperforming prior AutoEval baselines on vision and NLI benchmarks.

Problem

Estimating a classifier's accuracy on unlabeled out-of-distribution test sets (automated model evaluation) without the overconfidence, storage and compute overhead of prior methods.

Approach

Treat the classifier as an energy-based model with per-sample free energy from logits, then re-normalize the energies over the unlabeled dataset into a meta-distribution statistic (MDE) and fit a linear regressor from MDE to accuracy on synthetic shifted sets.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use MDE instead of NuclearNorm for accuracy predictionAutoEval · CIFAR-10, CIFAR-100, TinyImageNet, MNLI, and ImageNet synthetic shiftsR²: 0.935 -> 0.973 (CIFAR-10); 0.950 -> 0.966 (CIFAR-100); 0.969 -> 0.977 (TinyImageNet); 0.496 -> 0.680 (MNLI); 0.955 -> 0.Mixed
Use MDE instead of AvgEnergyEnergy-based AutoEval · CIFAR-10, CIFAR-100, TinyImageNet, MNLI, and ImageNet synthetic shiftsR²: 0.957 -> 0.973 (CIFAR-10); 0.963 -> 0.966 (CIFAR-100); 0.964 -> 0.977 (TinyImageNet); 0.595 -> 0.680 (MNLI); 0.953 -> 0.Helped
Use MDE instead of NuclearNorm for natural-shift accuracy predictionAutoEval · CIFAR-10, TinyImageNet, MNLI, and ImageNet natural shiftsMAE: 3.42 -> 1.78 (CIFAR-10); 3.84 -> 2.15 (TinyImageNet); 8.44 -> 5.76 (MNLI); 8.63 -> 5.48 (ImageNet)Helped
Use MDE instead of NuclearNormWILDS accuracy prediction · Camelyon17, RxRx1, and FMoW natural shiftsMAE: 4.26 -> 2.93 (Camelyon17); 3.67 -> 1.62 (RxRx1); 4.49 -> 2.24 (FMoW)Helped
Increase the MDE temperature above T=1 (ablation)MDE · CIFAR-10 with VGG-11correlation and MAE performanceHurt
Vary the training random seed (ablation)MDE AutoEval · CIFAR-10 with VGG-11performanceNo clear effect
Apply new transformations to naturally shifted test sets (ablation)MDE · CIFAR-10.1-A/B, CIFAR-10.2-A/B, CINIC-10-A/B, and STL-10.1-A/BMAE: MAE<5.92Mixed
Create long-tail test sets with imbalance ratio r≥0.4 (ablation)MDE · CIFAR-10-C, severity level 2accuracy prediction performanceHelped
Create severely imbalanced test sets with r≤0.2 (ablation)MDE · CIFAR-10-C, severity level 2accuracy prediction performanceHurt
Compute MDE from shallow-layer features instead of classification-head features (ablation)MDE · CIFAR-10 ID and OOD test setsdiscriminabilityHurt

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 CIFAR-100 TinyImageNet ImageNet-1K MNLI WILDS (Camelyon17, RxRx1, FMoW) CIFAR-10-C CIFAR-100-C ImageNet-C TinyImageNet-C

Benchmarks: CIFAR-10-C CIFAR-100-C ImageNet-C TinyImageNet-C CIFAR-10.1 CIFAR-10.2 CINIC-10 STL-10 ImageNet-V2 ImageNet-Sketch ImageNet-Rendition ImageNet-Adversarial

Metrics: R^2 Pearson r Spearman rho mean absolute error

Baselines: ConfScore Entropy Frechet AgreeScore ATC COT AvgEnergy ProjNorm NuclearNorm

Builds on: energy-based model (JEM) Average Thresholded Confidence Nuclear norm AutoEval ProjNorm

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