提出全新概率框架,用输入扰动量化大模型不确定性
Inv-Entropy: A Fully Probabilistic Framework for Uncertainty Quantification in Language Models
- 构建双随机游走模型,将输入输出视为马尔可夫链
- 引入逆熵(Inv-Entropy)度量,比现有方法更准确捕捉语义不确定性
- 支持多种嵌入与扰动策略,适合需要可信推理的场景
大型语言模型在自然语言处理中取得突破,但可靠部署需有效不确定性量化(UQ)。现有方法多为启发式且缺乏概率解释。本文从理论出发,论证扰动在LLM UQ中的作用,提出双随机游走视角,将输入输出对建模为由语义相似性定义转移概率的两个马尔可夫链。基于此,构建基于逆模型的全概率框架,通过系统扰动评估给定输出下输入空间的多样性来量化不确定性。在此框架中定义新度量Inv-Entropy。其优势在于灵活性:支持多种不确定性度量、嵌入方式、扰动策略及相似性度量。提出基于遗传算法的扰动算法GAAP,提升采样输入多样性。另引入新评估指标温度敏感性不确定性(TSU),直接评估不确定性而无需依赖正确性作为代理。大量实验表明,Inv-Entropy优于现有语义UQ方法。代码见https://github.com/UMDataScienceLab/Uncertainty-Quantification-for-LLMs。
原文摘要 · Abstract (English)
Large language models (LLMs) have transformed natural language processing, but their reliable deployment requires effective uncertainty quantification (UQ). Existing UQ methods are often heuristic and lack a probabilistic interpretation. This paper begins by providing a theoretical justification for the role of perturbations in UQ for LLMs. We then introduce a dual random walk perspective, modeling input-output pairs as two Markov chains with transition probabilities defined by semantic similarity. Building on this, we propose a fully probabilistic framework based on an inverse model, which quantifies uncertainty by evaluating the diversity of the input space conditioned on a given output through systematic perturbations. Within this framework, we define a new uncertainty measure, Inv-Entropy. A key strength of our framework is its flexibility: it supports various definitions of uncertainty measures, embeddings, perturbation strategies, and similarity metrics. We also propose GAAP, a perturbation algorithm based on genetic algorithms, which enhances the diversity of sampled inputs. In addition, we introduce a new evaluation metric, Temperature Sensitivity of Uncertainty (TSU), which directly assesses uncertainty without relying on correctness as a proxy. Extensive experiments demonstrate that Inv-Entropy outperforms existing semantic UQ methods. The code to reproduce the results can be found at https://github.com/UMDataScienceLab/Uncertainty-Quantification-for-LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。