CLUE: Concept-Level Uncertainty Estimation for Large Language Models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Wang, Yu-Hsiang, Bai, Andrew, Tsai, Che-Ping, Hsieh, Cho-Jui
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917769118744576
author Wang, Yu-Hsiang
Bai, Andrew
Tsai, Che-Ping
Hsieh, Cho-Jui
author_facet Wang, Yu-Hsiang
Bai, Andrew
Tsai, Che-Ping
Hsieh, Cho-Jui
contents Large Language Models (LLMs) have demonstrated remarkable proficiency in various natural language generation (NLG) tasks. Previous studies suggest that LLMs' generation process involves uncertainty. However, existing approaches to uncertainty estimation mainly focus on sequence-level uncertainty, overlooking individual pieces of information within sequences. These methods fall short in separately assessing the uncertainty of each component in a sequence. In response, we propose a novel framework for Concept-Level Uncertainty Estimation (CLUE) for LLMs. We leverage LLMs to convert output sequences into concept-level representations, breaking down sequences into individual concepts and measuring the uncertainty of each concept separately. We conduct experiments to demonstrate that CLUE can provide more interpretable uncertainty estimation results compared with sentence-level uncertainty, and could be a useful tool for various tasks such as hallucination detection and story generation.
format Preprint
id arxiv_https___arxiv_org_abs_2409_03021
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle CLUE: Concept-Level Uncertainty Estimation for Large Language Models
Wang, Yu-Hsiang
Bai, Andrew
Tsai, Che-Ping
Hsieh, Cho-Jui
Computation and Language
Machine Learning
Large Language Models (LLMs) have demonstrated remarkable proficiency in various natural language generation (NLG) tasks. Previous studies suggest that LLMs' generation process involves uncertainty. However, existing approaches to uncertainty estimation mainly focus on sequence-level uncertainty, overlooking individual pieces of information within sequences. These methods fall short in separately assessing the uncertainty of each component in a sequence. In response, we propose a novel framework for Concept-Level Uncertainty Estimation (CLUE) for LLMs. We leverage LLMs to convert output sequences into concept-level representations, breaking down sequences into individual concepts and measuring the uncertainty of each concept separately. We conduct experiments to demonstrate that CLUE can provide more interpretable uncertainty estimation results compared with sentence-level uncertainty, and could be a useful tool for various tasks such as hallucination detection and story generation.
title CLUE: Concept-Level Uncertainty Estimation for Large Language Models
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2409.03021