Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Benchmarking Uncertainty Quantification Methods for Large Language Models with LM-Polygraph

About

The rapid proliferation of large language models (LLMs) has stimulated researchers to seek effective and efficient approaches to deal with LLM hallucinations and low-quality outputs. Uncertainty quantification (UQ) is a key element of machine learning applications in dealing with such challenges. However, research to date on UQ for LLMs has been fragmented in terms of techniques and evaluation methodologies. In this work, we address this issue by introducing a novel benchmark that implements a collection of state-of-the-art UQ baselines and offers an environment for controllable and consistent evaluation of novel UQ techniques over various text generation tasks. Our benchmark also supports the assessment of confidence normalization methods in terms of their ability to provide interpretable scores. Using our benchmark, we conduct a large-scale empirical investigation of UQ and normalization techniques across eleven tasks, identifying the most effective approaches. Code: https://github.com/IINemo/lm-polygraph Benchmark: https://huggingface.co/LM-Polygraph

Roman Vashurin, Ekaterina Fadeeva, Artem Vazhentsev, Lyudmila Rvanova, Akim Tsvigun, Daniil Vasilev, Rui Xing, Abdelrahman Boda Sadallah, Kirill Grishchenkov, Sergey Petrakov, Alexander Panchenko, Timothy Baldwin, Preslav Nakov, Maxim Panov, Artem Shelmanov• 2024

Related benchmarks

TaskDatasetResultRank
Question AnsweringTriviaQA
EM55.7
116
Multi-answer Question AnsweringMAQA-ΔK−1
KL Divergence0.43
48
Uncertainty EstimationTriviaQA--
37
Question AnsweringTriviaQA
Exact Match AUC80.3
28
Multi-answer Question AnsweringMAQA
Hamming Distance0.254
28
SummarizationCNN/DailyMail
Hamming Score-0.217
28
Uncertainty QuantificationCNN/DailyMail
Hamming AUC0.605
28
Uncertainty QuantificationMAQA-ΔK−1
KL Divergence AUC0.69
28
Uncertainty QuantificationWMT 19
COMET AUC0.577
28
Machine TranslationWMT19
COMET Score0.218
28
Showing 10 of 11 rows

Other info

Follow for update