paper-with-me

홈 › Papers

Calibrating Large Language Models Using Their Generations Only

2024-03-09 · Dennis Ulmer, Martin Gubri, Hwaran Lee, Sangdoo Yun, Seong Joon Oh

As large language models (LLMs) are increasingly deployed in user-facing applications, building trust and maintaining safety by accurately quantifying a model's confidence in its prediction becomes even more important. However, finding effective ways to calibrate LLMs - especially when the only interface to the models is their generated text - remains a challenge. We propose APRICOT (auxiliary prediction of confidence targets): A method to set confidence targets and train an additional model that predicts an LLM's confidence based on its textual input and output alone. This approach has several advantages: It is conceptually simple, does not require access to the target model beyond its output, does not interfere with the language generation, and has a multitude of potential usages, for instance by verbalizing the predicted confidence or adjusting the given answer based on the confidence. We show how our approach performs competitively in terms of calibration error for white-box and black-box LLMs on closed-book question-answering to detect incorrect LLM answers.

📄 PDF Abstract BibTeX arXiv:2403.05973

Code (1)

parameterlab/apricot 공식 구현 pytorch

Tasks

Question AnsweringText Generation

Methods 이 논문이 사용한 방법론

SET Dynamic Sparse Training method where weight mask is updated randomly periodically

Similar Papers 제목 키워드 기반

A Survey of Confidence Estimation and Calibration in Large Language Models

2023-11-14 · Jiahui Geng, Fengyu Cai, Yuxia Wang, Heinz Koeppl 외

Large language models (LLMs) have demonstrated remarkable capabilities across a wide range of tasks in various domains. Despite their impressive performance, they can be unreliable due to factual errors in their generati…

Language Modelling

Calibrating Large Language Models with Sample Consistency

2024-02-21 · Qing Lyu, Kumar Shridhar, Chaitanya Malaviya, Li Zhang 외

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 technique…

LitCab: Lightweight Language Model Calibration over Short- and Long-form Responses

2023-10-30 · Xin Liu, Muhammad Khalifa, Lu Wang

A model is considered well-calibrated when its probability estimate aligns with the actual likelihood of the output being correct. Calibrating language models (LMs) is crucial, as it plays a vital role in detecting and m…

FormLanguage ModelingLanguage ModellingText Generation

Calibrating Long-form Generations from Large Language Models

2024-02-09 · Yukun Huang, Yixin Liu, Raghuveer Thirukovalluru, Arman Cohan 외

To enhance Large Language Models' (LLMs) reliability, calibration is essential -- the model's assessed confidence scores should align with the actual likelihood of its responses being correct. However, current confidence…

Form

Long Horizon Temperature Scaling

2023-02-07 · Andy Shih, Dorsa Sadigh, Stefano Ermon

Temperature scaling is a popular technique for tuning the sharpness of a model distribution. It is used extensively for sampling likely generations and calibrating model uncertainty, and even features as a controllable p…

Multiple-choice