atlas

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

When is Multicalibration Post-Processing Necessary?

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

Large empirical study of when multicalibration post-processing (HKRR, HJZ) helps versus ERM and simple calibration (Platt, isotonic, temperature scaling) across tabular, image, and language datasets.

Problem

Whether multicalibration is a natural consequence of ERM in practice, and when multicalibration post-processing is needed and effective, given finite data split between training and post-processing over finite subgroup collections.

Approach

Systematic large-scale empirical evaluation comparing ERM, classical recalibration, and multicalibration post-processing (HKRR and HJZ) across model families and calibration-set fractions, measuring max group-wise smooth ECE on held-out test data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
apply HKRR multicalibration post-processingDistilBERT · Civil Commentsworst-group smECE: 0.06 to 0.043Helped
apply HKRR multicalibration post-processingViT · Camelyon17worst-group smECE: 0.07 to 0.02Helped
apply HKRR multicalibration post-processingResNet-56 · Amazon Polarityworst-group smECE: 0.09 to 0.05Helped
apply multicalibration post-processing to neural-network ERMMLP · tabular datasetsworst-group calibration errorNo clear effect
apply HKRR or HJZ post-processingSVM, decision tree, or naive Bayes · tabular datasetsmax smECE: 0.05 or moreHelped
apply multicalibration post-processingMLP, random forest, or logistic regression · HMDAworst-group calibration error: 0.03-0.07Helped
use isotonic regression instead of no post-processingnaive Bayes · tabular datasetsmax smECE: up to 0.01 errorHelped
use multicalibration post-processing at a calibration fraction that lowers worst-group errorMLP · HMDAaccuracy: about 2%Hurt
select the post-processing method using worst-group ECE rather than worst-group smECEdecision tree · MEPSworst-group calibration error: 0.156Mixed
use HKRR rather than HJZimage and language models · image and language datasetsworst-group calibration errorHelped
reuse model training data for multicalibration post-processing instead of holding out calibration datamulticalibration post-processing · tabular datasetsworst-group smECE: 0.025; 0.05-0.1Mixed
use isotonic regression rather than ERMlarge image and language models · image and language datasetsworst-group calibration error: 4/6Helped

Surprising findings

Limitations

Experimental setup

Datasets: ACS Income UCI Bank Marketing UCI Credit Default HMDA MEPS CelebA Camelyon17 CivilComments Amazon Polarity

Benchmarks: ACS Income UCI Bank Marketing UCI Credit Default HMDA MEPS CelebA Camelyon17 CivilComments Amazon Polarity

Metrics: max group-wise smooth ECE max group-wise binned ECE test accuracy

Baselines: ERM Platt scaling isotonic regression temperature scaling HKRR HJZ

Builds on: HKRR multicalibration algorithm HJZ multicalibration algorithm family Platt scaling isotonic regression temperature scaling

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