Uncertainty Quantification of Large Language Models through Multi-Dimensional Responses

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Chen, Tiejin, Liu, Xiaoou, Da, Longchao, Chen, Jia, Papalexakis, Vagelis, Wei, Hua
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915171095543808
author Chen, Tiejin
Liu, Xiaoou
Da, Longchao
Chen, Jia
Papalexakis, Vagelis
Wei, Hua
author_facet Chen, Tiejin
Liu, Xiaoou
Da, Longchao
Chen, Jia
Papalexakis, Vagelis
Wei, Hua
contents Large Language Models (LLMs) have demonstrated remarkable capabilities across various tasks due to large training datasets and powerful transformer architecture. However, the reliability of responses from LLMs remains a question. Uncertainty quantification (UQ) of LLMs is crucial for ensuring their reliability, especially in areas such as healthcare, finance, and decision-making. Existing UQ methods primarily focus on semantic similarity, overlooking the deeper knowledge dimensions embedded in responses. We introduce a multi-dimensional UQ framework that integrates semantic and knowledge-aware similarity analysis. By generating multiple responses and leveraging auxiliary LLMs to extract implicit knowledge, we construct separate similarity matrices and apply tensor decomposition to derive a comprehensive uncertainty representation. This approach disentangles overlapping information from both semantic and knowledge dimensions, capturing both semantic variations and factual consistency, leading to more accurate UQ. Our empirical evaluations demonstrate that our method outperforms existing techniques in identifying uncertain responses, offering a more robust framework for enhancing LLM reliability in high-stakes applications.
format Preprint
id arxiv_https___arxiv_org_abs_2502_16820
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Uncertainty Quantification of Large Language Models through Multi-Dimensional Responses
Chen, Tiejin
Liu, Xiaoou
Da, Longchao
Chen, Jia
Papalexakis, Vagelis
Wei, Hua
Computation and Language
Artificial Intelligence
Large Language Models (LLMs) have demonstrated remarkable capabilities across various tasks due to large training datasets and powerful transformer architecture. However, the reliability of responses from LLMs remains a question. Uncertainty quantification (UQ) of LLMs is crucial for ensuring their reliability, especially in areas such as healthcare, finance, and decision-making. Existing UQ methods primarily focus on semantic similarity, overlooking the deeper knowledge dimensions embedded in responses. We introduce a multi-dimensional UQ framework that integrates semantic and knowledge-aware similarity analysis. By generating multiple responses and leveraging auxiliary LLMs to extract implicit knowledge, we construct separate similarity matrices and apply tensor decomposition to derive a comprehensive uncertainty representation. This approach disentangles overlapping information from both semantic and knowledge dimensions, capturing both semantic variations and factual consistency, leading to more accurate UQ. Our empirical evaluations demonstrate that our method outperforms existing techniques in identifying uncertain responses, offering a more robust framework for enhancing LLM reliability in high-stakes applications.
title Uncertainty Quantification of Large Language Models through Multi-Dimensional Responses
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2502.16820