Improving Semantic Uncertainty Quantification in Language Model Question-Answering via Token-Level Temperature Scaling

Tom A. Lamb, Desi R. Ivanova, Philip H. S. Torr, Tim G. J. Rudner
4/8/2026
cs.LG

No official code found. Best implementation: tomalamb/semantic-calibration (3 stars, last commit 2026-04-08).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

Code repository page for our paper ``Improving Semantic Uncertainty Quantification in Language Models via Token-Level Temperature Scaling''.

30PythonNov 3, 20256 months ago

Similar Papers

Cite this paper

@article{lamb2026improving,
  title  = {Improving Semantic Uncertainty Quantification in Language Model Question-Answering via Token-Level Temperature Scaling},
  author = {Tom A. Lamb and Desi R. Ivanova and Philip H. S. Torr and Tim G. J. Rudner},
  year   = {2026},
  eprint = {2604.07172},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2604.07172v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2604.07172.svg)](https://codeofpaper.com/paper/2604.07172)

Discussion