AIMEApr 23

Unbiased Prevalence Estimation with Multicalibrated LLMs

arXiv:2604.2154967.3h-index: 19
Predicted impact top 54% in AI · last 90 daysOriginality Incremental advance
AI Analysis

It provides a theoretically grounded solution to a fundamental measurement problem across disciplines, with practical bias reduction in prevalence estimation using LLMs or any classifier.

The paper shows that multicalibration, which enforces calibration conditional on input features, enables unbiased prevalence estimation under covariate shift, while standard calibration and quantification methods fail. Simulations and empirical applications (e.g., employment prevalence across U.S. states, political text classification across countries) demonstrate that multicalibration substantially reduces bias.

Estimating the prevalence of a category in a population using imperfect measurement devices (diagnostic tests, classifiers, or large language models) is fundamental to science, public health, and online trust and safety. Standard approaches correct for known device error rates but assume these rates remain stable across populations. We show this assumption fails under covariate shift and that multicalibration, which enforces calibration conditional on the input features rather than just on average, is sufficient for unbiased prevalence estimation under such shift. Standard calibration and quantification methods fail to provide this guarantee. Our work connects recent theoretical work on fairness to a longstanding measurement problem spanning nearly all academic disciplines. A simulation confirms that standard methods exhibit bias growing with shift magnitude, while a multicalibrated estimator maintains near-zero bias. While we focus the discussion mostly on LLMs, our theoretical results apply to any classification model. Two empirical applications -- estimating employment prevalence across U.S. states using the American Community Survey, and classifying political texts across four countries using an LLM -- demonstrate that multicalibration substantially reduces bias in practice, while highlighting that calibration data should cover the key feature dimensions along which target populations may differ.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes