Target-Aware Calibration Data Selection for Preserving Uncertainty in Quantized Language Models
Not on the map. Why?
Abstract (arXiv)
Quantization is widely used to deploy large language models, but its effect on uncertainty behavior, such as confidence, margins, and abstention, is rarely treated as a primary objective. We frame calibration-data selection for quantization as a target-dependent uncertainty-preservation problem. Different deployments emphasize different regions of the input distribution, yet prior work mainly optimizes accuracy-oriented compression metrics or adjusts scores after quantization. We formalize this goal with distributional and boundary preservation risks, and provide a simple mixture-mismatch argument explaining why no single calibration recipe should be expected to fit all targets. We introduce Doubt-Preserving Quantization (DPQ), a lightweight pre-quantization recipe family that uses full-precision predictions to construct target-aligned calibration mixtures of high-doubt examples and generic anchors. Across 8 language models, 9 NLP benchmarks, and 22 comparison methods, the leading fixed recipe changes with the preservation target: DPQ-r75 leads on SQuAD2 answerability-boundary preservation, while milder or single-signal variants, including DPQ-r50, confidence-only, and entropy-only, better preserve broad multiple-choice QA behavior. These results show that calibration data should be selected for the specific full-precision score behavior a deployment needs to preserve, rather than treated as a fixed quantization detail.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 33 of its 56 references that are in the map. First posted on arXiv 21 Aug 2026.
Builds on
33 of 56 references.
References · 33
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
- 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
- 2021Calibrate Before Use: Improving Few-Shot Performance of Language Modelscited by 1,392
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,298
21 more
- 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,287
- 2019How Can We Know What Language Models Know?cited by 959
- 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scalecited by 906
- 2017Selective Classification for Deep Neural Networkscited by 852
- 2023Large Language Modelscited by 780
- 2024QuaRot: Outlier-Free 4-Bit Inference in Rotated LLMscited by 634
- 2019Deep Batch Active Learning by Diverse, Uncertain Gradient Lower Boundscited by 610
- 2022ZeroQuant: Efficient and Affordable Post-Training Quantization for Large-Scale Transformerscited by 526
- 2019Measuring Calibration in Deep Learningcited by 446
- 2021Revisiting the Calibration of Modern Neural Networkscited by 390
- 2024QuIP#: Even Better LLM Quantization with Hadamard Incoherence and Lattice Codebookscited by 378
- 2023SpQR: A Sparse-Quantized Representation for Near-Lossless LLM Weight Compressioncited by 367
- 2019Beyond temperature scaling: Obtaining well-calibrated multiclass probabilities with Dirichlet calibrationcited by 364
- 2020Calibration of Pre-trained Transformerscited by 315
- 2019Verified Uncertainty Calibrationcited by 309
- 2024Extreme Compression of Large Language Models via Additive Quantizationcited by 232
- 2022Uncertainty Quantification with Pre-trained Language Models: A Large-Scale Empirical Analysiscited by 104
- 2023On the Impact of Calibration Data in Post-training Quantization and Pruningcited by 58
- 2024Self-calibration for Language Model Quantization and Pruningcited by 13
- 2026FlashSVD v1.5: Making Low-Rank Transformers Inference Actually Fastcited by 3
- 2026BudgetDraft: Acceptance-Aware Multi-View Training for Sparse-KV Speculative Decodingcited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Reliability Scaling Laws for Quantized Large Language Models
- 2026Calibrating Beyond English: Language Diversity for Better Quantized Multilingual LLM
- 2026Does Compression Preserve Uncertainty? A Unified Benchmark for Quantized and Sparse LLMs via Conformal Prediction
- 2024When Quantization Affects Confidence of Large Language Models?
- 2025Revisiting Uncertainty Estimation and Calibration of Large Language Models
- 2026FAQ: Mitigating Quantization Error via Regenerating Calibration Data with Family-Aware Quantization
- 2026MixQuant: Adaptive Mixed-Precision Quantization for Large Language Models
- 2026Calibration vs Decision Making: Revisiting the Reliability Paradox in Unlearned Language Models