Test-time Calibration Learning for Large Language Model Reasoning
cs.LG, cs.AI
Submitted: 2026-10-02
Updated: 2026-10-02
Comments: 34 pages
Code: https://github.com/tmlr-group/TTCL
License: http://creativecommons.org/licenses/by/4.0/
The gist: Reliable large language models (LLMs) must not only produce accurate answers but also express confidence that faithfully reflects their probability of being correct.
Terminology
Abstract
Reliable large language models (LLMs) must not only produce accurate answers but also express confidence that faithfully reflects their probability of being correct. Such calibration is essential for identifying uncertain predictions and supporting reliable decision-making in real-world deployment. Recent studies incorporate calibration learning into reinforcement learning (RL), jointly optimizing answer correctness and verbalized confidence using ground-truth correctness supervision. However, their reliance on labeled data limits their applicability in practical test-time settings, where ground-truth labels are unavailable and calibration may need to adapt to newly encountered target tasks. To address this challenge, we propose Test-Time Calibration Learning (TTCL), a label-free framework that jointly adapts reasoning accuracy and verbalized confidence directly on unlabeled target-task data. Specifically, TTCL derives self-supervision signals for both correctness and calibration from multiple model-generated responses, enabling calibration learning at test time without ground-truth labels. Theoretical analysis further establishes TTCL as a bounded surrogate for the ideal calibration objective. Extensive experiments on mathematical reasoning and factual question answering demonstrate that TTCL consistently improves both accuracy and calibration across diverse models and tasks. On base models, TTCL achieves an average relative accuracy improvement of +40.13% and an ECE reduction of +70.80% across eight benchmarks. Moreover, TTCL can further improve both accuracy and calibration for already calibrated models under domain shift, particularly when source-domain calibration transfers poorly to target tasks. In the math-to-factQA setting, TTCL achieves an average relative accuracy gain of +20.35% and reduces ECE by +53.83%. The source code is released at https://github.com/tmlr-group/TTCL.
Sources
- Uncalibrated Reasoning: GRPO Induces Overconfidence for Stochastic Outcomes
- Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challenge
- CALIBER: Calibrating Confidence Before and After Reasoning in Language Models
- The Llama 3 Herd of Models
- SimpleQA Verified: A Reliable Factuality Benchmark to Measure Parametric Knowledge
- Language Models (Mostly) Know What They Know
- Why Language Models Hallucinate
- Understanding R1-Zero-Like Training: A Critical Perspective
- Uncertainty Estimation in Autoregressive Structured Prediction
- Maximizing Confidence Alone Improves Reasoning
- GPQA: A Graduate-Level Google-Proof Q&A Benchmark
- Can Large Reasoning Models Self-Train?
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Rethinking How to Remember: Beyond Atomic Facts in Lifelong LLM Agent Memory
- Kimi K3: Open Frontier Intelligence
- Measuring short-form factuality in large language models
- Mitigating LLM Hallucination via Behaviorally Calibrated Reinforcement Learning
- MiMo-V2-Flash Technical Report
- Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvement
- Qwen3 Technical Report
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks