Improving Semantic Uncertainty Quantification in Language Model Question-Answering via Token-Level Temperature Scaling
Not on the map. Why?
Abstract (arXiv)
Calibration is central to reliable semantic uncertainty quantification, yet prior work has largely focused on discrimination, neglecting calibration. As calibration and discrimination capture distinct aspects of uncertainty, focusing on discrimination alone yields an incomplete picture. We address this gap by systematically evaluating both aspects across a broad set of confidence measures. We show that current approaches, particularly fixed-temperature heuristics, produce systematically miscalibrated and poorly discriminative semantic confidence distributions. We demonstrate that optimising a single scalar temperature, which, we argue, provides a suitable inductive bias, is a surprisingly simple yet effective solution. Our exhaustive evaluation confirms that temperature scaling consistently improves semantic calibration, discrimination, and downstream entropy, outperforming both heuristic baselines and more expressive token-level recalibration methods on question-answering tasks.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 25 of its 64 references that are in the map. First posted on arXiv 8 Apr 2026.
Builds on
25 of 64 references.
References · 25
- 2017Attention Is All You Needcited by 59,932
- 2023GPT-4 Technical Reportcited by 18,132
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,665
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,157
- 2022Language Models (Mostly) Know What They Knowcited by 1,783
- 2023Semantic Uncertainty: Linguistic Invariances for Uncertainty Estimation in Natural Language Generationcited by 795
13 more
- 2019Measuring Calibration in Deep Learningcited by 443
- 2020Simple and Principled Uncertainty Estimation with Deterministic Deep Learning via Distance Awarenesscited by 389
- 2020Calibration of Pre-trained Transformerscited by 307
- 2023Generating with Confidence: Uncertainty Quantification for Black-box Large Language Modelscited by 286
- 2021Deep Deterministic Uncertainty: A Simple Baselinecited by 200
- 2024Kernel Language Entropy: Fine-grained Uncertainty Quantification for LLMs from Semantic Similaritiescited by 156
- 2024Long-form factuality in large language modelscited by 146
- 2018Correcting Length Bias in Neural Machine Translationcited by 119
- 2023Bayesian Low-rank Adaptation for Large Language Modelscited by 112
- 2024Calibrating Language Models with Adaptive Temperature Scalingcited by 57
- 2024Improving Uncertainty Estimation through Semantically Diverse Language Generationcited by 34
- 2020Evaluating probabilistic classifiers: Reliability diagrams and score decompositions revisitedcited by 25
- 2025Revisiting Uncertainty Quantification Evaluation in Language Models: Spurious Interactions with Response Length Bias Resultscited by 21
Built on by
Citing papers per year: 2026: 2.
2 papers.