CALIBER: Calibrating Confidence Before and After Reasoning in Language Models

15d ago · Global · primary source: export.arxiv.org

A new protocol called CALIBER improves how reasoning language models estimate their own confidence by supervising predictions both before and after the model generates an answer, according to research published on arXiv [1]. The method, whose name stands for Calibration Before and After Reasoning, addresses a limitation in existing systems that typically elicit confidence only once — either before the model begins its chain-of-thought or after it finishes [1]. The authors argue that confidence is state-dependent: a pre-reasoning estimate should reflect the probability of eventually solving the prompt, while a post-reasoning estimate should predict whether the specific answer just produced is correct [2]. CALIBER supervises each estimate with a target matched to its information state [1]. On the BigMathDigits benchmark, CALIBER reduced Expected Calibration Error by 52.5% compared to the strongest single-confidence baseline when tested with a 7-billion-parameter model [1]. It also achieved the best Brier score and AUROC on that dataset while remaining within 2.1 points of the top accuracy figure [1]. When scaled to a 30-billion-parameter model, the protocol again posted the best ECE on BigMathDigits and stayed competitive on Brier score and AUROC [2]. Out-of-distribution testing showed further gains. CALIBER recorded the best ECE and Brier score on GPQA and TriviaQA, two datasets that differ from the training distribution, and remained competitive on SimpleQA [1]. Ablation studies indicated that aligning the supervision target with the model’s position in the reasoning process is most valuable under distribution shift, where it consistently lowered calibration error across all out-of-distribution benchmarks [2]. The work arrives as language models are increasingly deployed in settings where users need reliable confidence scores alongside answers [1]. The researchers note that without position-aware supervision, models may conflate prompt-level and answer-level uncertainty, degrading the usefulness of the confidence signal [2].

research-papersafety-research

Background sources we checked (7)
  • arxiv.org ↗ Reasoning language models are increasingly asked not only to answer difficult questions, but also to estimate their likelihood of success. Existing methods typically elicit confidence only once: either before thinking or after answering. We argue that confidence in reasoning mode…
  • en.wikipedia.org ↗ On the American late-night live television sketch comedy and variety show Saturday Night Live (SNL), a commercial advertisement parody is commonly shown after the host's opening monologue. Many of the parodies were produced by James Signorelli. The industries, products, and ad f…
  • arxiv.org ↗ CatalyzeX Code Finder for Papers (What is CatalyzeX?) ... DagsHub Toggle ... DagsHub (What is DagsHub?)…
  • arxiv.org ↗ With the creation of new datasets, the question arises of whether the data in them is complementary to other datasets for training ML models (see recent reviews for a perspective of catalysts informatics22, 23, 24). This is especially important when consolidating data with a vari…
  • arxiv.org ↗ CatalyzeX Code Finder for Papers (What is CatalyzeX?) ... DagsHub Toggle ... DagsHub (What is DagsHub?)…
  • en.wikipedia.org ↗ Sustainable Development Goals (abbr. SDGs) were adopted in 2015 by all United Nations (UN) members for the 2030 Agenda for Sustainable Development. The aim of the 17 global goals is "peace and prosperity for people and the planet", tackling climate change, and working to preserv…
  • en.wikipedia.org ↗ In molecular biology, a transcription factor (TF) (or sequence-specific DNA-binding factor) is a protein that controls the rate of transcription of genetic information from DNA to messenger RNA, by binding to DNA sequences. Specificity can be due to sequence motifs, or epigenetic…

Sources

Spot something wrong? Report an issue