Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Rewarding Doubt: A Reinforcement Learning Approach to Calibrated Confidence Expression of Large Language Models

About

A safe and trustworthy use of Large Language Models (LLMs) requires an accurate expression of confidence in their answers. We propose a novel Reinforcement Learning approach that allows to directly fine-tune LLMs to express calibrated confidence estimates alongside their answers to factual questions. Our method optimizes a reward based on the logarithmic scoring rule, explicitly penalizing both over- and under-confidence. This encourages the model to align its confidence estimates with the actual predictive accuracy. The optimal policy under our reward design would result in perfectly calibrated confidence expressions. Unlike prior approaches that decouple confidence estimation from response generation, our method integrates confidence calibration seamlessly into the generative process of the LLM. Empirically, we demonstrate that models trained with our approach exhibit substantially improved calibration and generalize to unseen tasks without further fine-tuning, suggesting the emergence of general confidence awareness.

David Bani-Harouni, Chantal Pellegrini, Paul Stangel, Ege \"Ozsoy, Kamilia Zaripova, Nassir Navab, Matthias Keicher• 2025

Related benchmarks

TaskDatasetResultRank
Multiple-Answer Question AnsweringQAMPARI--
16
Question AnsweringTriviaQA Out-of-distribution (test)
Accuracy71.1
15
Question AnsweringHotpotQA In-distribution (test)
Accuracy29.4
15
Question AnsweringMSMARCO Out-of-distribution (test)
Accuracy40.9
15
Question AnsweringNQ-Open Out-of-distribution (test)
Accuracy42.4
15
Question AnsweringHotpotQA, TriviaQA, MSMARCO, NQ-Open (macro-average)
Tokens193.3
15
Question AnsweringTriviaQA Single-Answer
ECE2.26
11
Short-form Question AnsweringTrivia QA
AUROC85.9
7
Question AnsweringMedQA
ECE0.1145
3
Showing 9 of 9 rows

Other info

Follow for update