atlas

Research map / Large language models / Language, safety and interpretability

Post-Hoc Reversal: Are We Selecting Models Prematurely?

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: Uncertainty quantification for LLMs · 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.

Shows that post-hoc transforms (TS, SWA, ensembling) can reverse performance trends across epochs, model sizes and hyperparameters on noisy data, and proposes post-hoc selection that picks models using post-transform validation metrics.

Problem

Standard practice selects base models by base-model performance before applying post-hoc transforms such as temperature scaling, SWA and ensembling, which can be suboptimal, especially on noisy data where base models overfit.

Approach

Identifies "post-hoc reversal", where post-hoc transforms reverse base performance trends, and proposes post-hoc selection: choose epochs, model sizes and hyperparameters using validation metrics computed after SWA/ensembling/TS rather than base metrics.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Select checkpoints using post-hoc metrics rather than base-model metricsSWA+TS and SWA+Ens+TS · CIFAR-N and FMoWtest loss and test error: > 2× improvement from post-hoc selection as compared to naive selectionHelped
Select checkpoints using post-hoc metrics rather than base-model metricsSWA+Ens+TS on LLaMA-2-7B · Guanaco test setMMLU accuracy: > 1.5 ×Helped
Use a 0.7-epoch checkpointing interval instead of higher or lower intervals (ablation)SWA on LLaMA-2-7B · Guanaco instruction tuningMMLU accuracyHelped
Use post-hoc selection with CE, SOP, or ELR training objectives (ablation)CIFAR-N models · CIFAR-Ntest loss and test errorHelped
Replace the CE training objective with SOP or ELR (ablation)CIFAR-N models · CIFAR-Ntest loss and test errorNo clear effect
Use constant rather than exponentially decaying or cosine learning rates (ablation)DenseNet121 with post-hoc transforms · FMoWpost-hoc performanceHelped
Use SWA and ensembling instead of the base model aloneResNet18 · CIFAR-10-N noisy label setstest loss and test errorHelped
Use SWA, ensembling, or their compositionsDenseNet121 · FMoWtest error: approx. 10-point improvementHelped
Use SWA instead of ensemblingDenseNet121 · FMoWtest errorHurt
Ensemble SWA models instead of ensembling models aloneDenseNet121 · FMoWtest errorNo clear effect
Apply TS to align loss with errorBase models · CIFAR-N and FMoWloss-error mismatchHelped
Use post-hoc selection on OOD rather than ID validation/test splitsSWA+TS and SWA+Ens+TS · FMoW OODtest loss and test error: > 5 pointsHelped

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10-N CIFAR-100-N FMoW Guanaco Yelp Folktables Reddit-5k Reddit-12k Collab MMLU

Benchmarks: CIFAR-10-N CIFAR-100-N FMoW Guanaco MMLU Yelp Folktables Reddit-5k Reddit-12k Collab

Metrics: test cross-entropy loss test classification error MMLU accuracy perplexity causal language modeling error

Baselines: naive selection

Builds on: stochastic weight averaging temperature scaling double descent effective model complexity

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