Calibrating Large Language Models with Sample Consistency

Qing Lyu, Kumar Shridhar, Chaitanya Malaviya, Li Zhang, Yanai Elazar +4 more
2/21/2024
cs.CL

No official code found. Best implementation: veronica320/Calibrating-LLMs-with-Consistency (3 stars, MIT, last commit 2024-12-19).

Abstract

Accurately gauging the confidence level of Large Language Models' (LLMs) predictions is pivotal for their reliable application. However, LLMs are often uncalibrated inherently and elude conventional calibration techniques due to their proprietary nature and massive scale. In this work, we explore the potential of deriving confidence from the distribution of multiple randomly sampled model generations, via three measures of consistency. We perform an extensive evaluation across various open and closed-source models on nine reasoning datasets. Results show that consistency-based calibration methods outperform existing post-hoc approaches. Meanwhile, we find that factors such as intermediate explanations, model scaling, and larger sample sizes enhance calibration, while instruction-tuning makes calibration more difficult. Moreover, confidence scores obtained from consistency have the potential to enhance model performance. Finally, we offer practical guidance on choosing suitable consistency metrics for calibration, tailored to the characteristics of various LMs.

View on arXivView PDF

Code Implementations(1)

MIT

Code and data accompanying our paper "Calibrating Large Language Models with Sample Consistency" on arxiv.

30PDDL, ShellJul 26, 20241 years agoMIT

Similar Papers

Cite this paper

@article{lyu2024calibrating,
  title  = {Calibrating Large Language Models with Sample Consistency},
  author = {Qing Lyu and Kumar Shridhar and Chaitanya Malaviya and Li Zhang and Yanai Elazar and Niket Tandon and Marianna Apidianaki and Mrinmaya Sachan and Chris Callison-Burch},
  year   = {2024},
  eprint = {2402.13904},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2402.13904v1}
}

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/2402.13904.svg)](https://codeofpaper.com/paper/2402.13904)

Discussion