Uncertainty in Language Models: Assessment through Rank-Calibration

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Huang, Xinmeng, Li, Shuo, Yu, Mengxin, Sesia, Matteo, Hassani, Hamed, Lee, Insup, Bastani, Osbert, Dobriban, Edgar
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914949067964416
author Huang, Xinmeng
Li, Shuo
Yu, Mengxin
Sesia, Matteo
Hassani, Hamed
Lee, Insup
Bastani, Osbert
Dobriban, Edgar
author_facet Huang, Xinmeng
Li, Shuo
Yu, Mengxin
Sesia, Matteo
Hassani, Hamed
Lee, Insup
Bastani, Osbert
Dobriban, Edgar
contents Language Models (LMs) have shown promising performance in natural language generation. However, as LMs often generate incorrect or hallucinated responses, it is crucial to correctly quantify their uncertainty in responding to given inputs. In addition to verbalized confidence elicited via prompting, many uncertainty measures ($e.g.$, semantic entropy and affinity-graph-based measures) have been proposed. However, these measures can differ greatly, and it is unclear how to compare them, partly because they take values over different ranges ($e.g.$, $[0,\infty)$ or $[0,1]$). In this work, we address this issue by developing a novel and practical framework, termed $Rank$-$Calibration$, to assess uncertainty and confidence measures for LMs. Our key tenet is that higher uncertainty (or lower confidence) should imply lower generation quality, on average. Rank-calibration quantifies deviations from this ideal relationship in a principled manner, without requiring ad hoc binary thresholding of the correctness score ($e.g.$, ROUGE or METEOR). The broad applicability and the granular interpretability of our methods are demonstrated empirically.
format Preprint
id arxiv_https___arxiv_org_abs_2404_03163
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Uncertainty in Language Models: Assessment through Rank-Calibration
Huang, Xinmeng
Li, Shuo
Yu, Mengxin
Sesia, Matteo
Hassani, Hamed
Lee, Insup
Bastani, Osbert
Dobriban, Edgar
Computation and Language
Artificial Intelligence
Machine Learning
Language Models (LMs) have shown promising performance in natural language generation. However, as LMs often generate incorrect or hallucinated responses, it is crucial to correctly quantify their uncertainty in responding to given inputs. In addition to verbalized confidence elicited via prompting, many uncertainty measures ($e.g.$, semantic entropy and affinity-graph-based measures) have been proposed. However, these measures can differ greatly, and it is unclear how to compare them, partly because they take values over different ranges ($e.g.$, $[0,\infty)$ or $[0,1]$). In this work, we address this issue by developing a novel and practical framework, termed $Rank$-$Calibration$, to assess uncertainty and confidence measures for LMs. Our key tenet is that higher uncertainty (or lower confidence) should imply lower generation quality, on average. Rank-calibration quantifies deviations from this ideal relationship in a principled manner, without requiring ad hoc binary thresholding of the correctness score ($e.g.$, ROUGE or METEOR). The broad applicability and the granular interpretability of our methods are demonstrated empirically.
title Uncertainty in Language Models: Assessment through Rank-Calibration
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2404.03163