arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.20161cs.CLcs.AIcs.LG

语义标记聚类用于大型语言模型中的高效不确定性量化

Semantic Token Clustering for Efficient Uncertainty Quantification in Large Language Models

The University of Tokyo(东京大学)

Qi Cao, Andrew Gambardella, Takeshi Kojima, Yutaka Matsuo, Yusuke Iwasawa

AI总结:

本文提出语义标记聚类方法,通过语义信息高效量化大型语言模型的不确定性,无需辅助模型且计算开销低。

摘要:

大型语言模型(LLMs)在多样任务中展现了显著能力,但其输出的真实性无法保证,且过度自信倾向进一步限制了可靠性。不确定性量化提供了一种识别潜在不可靠输出的有前途的方法,但现有方法大多依赖重复采样或辅助模型,引入了显著的计算开销。为了解决这些限制,我们提出了语义标记聚类(STC),一种高效的不确定性量化方法,利用LLMs内固有的语义信息。具体而言,我们通过嵌入聚类和前缀匹配将标记分为语义一致的簇,并基于相应语义簇上的概率质量量化不确定性。我们的方法仅需一次生成,且不依赖辅助模型。实验结果表明,STC在性能上与最先进的基线相当,同时显著减少了计算开销。

英文摘要:

Large language models (LLMs) have demonstrated remarkable capabilities across diverse tasks. However, the truthfulness of their outputs is not guaranteed, and their tendency toward overconfidence further limits reliability. Uncertainty quantification offers a promising way to identify potentially unreliable outputs, but most existing methods rely on repeated sampling or auxiliary models, introducing substantial computational overhead. To address these limitations, we propose Semantic Token Clustering (STC), an efficient uncertainty quantification method that leverages the semantic information inherently encoded in LLMs. Specifically, we group tokens into semantically consistent clusters using embedding clustering and prefix matching, and quantify uncertainty based on the probability mass aggregated over the corresponding semantic cluster. Our approach requires only a single generation and does not depend on auxiliary models. Experimental results show that STC achieves performance comparable to state-of-the-art baselines while substantially reducing computational overhead.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑