用轻量代理模型精准感知大模型是否真的知道答案。
Estimating the Black-box LLM Uncertainty with Distribution-Aligned Adversarial Distillation

- 构建生成-判别架构,让小模型学习大模型输出的高质量区域。
- 仅需原模型1%大小的代理模型即可实现可靠不确定性估计。
- 适合需要实时评估黑盒大模型可信度的部署场景。
大型语言模型在复杂推理和问答任务中进展迅速,但幻觉问题仍是实际部署的核心瓶颈,尤其针对仅通过API访问的商业黑盒模型。现有不确定性量化方法通常依赖计算开销大的多重采样或内部参数,难以实现实时估计,且无法捕捉黑盒推理过程中的隐含信息。为此,我们提出分布对齐对抗蒸馏(DisAAD),采用生成-判别架构,引导轻量级代理模型学习目标黑盒模型输出分布的高质量区域,从而有效赋予其判断黑盒模型是否真正掌握知识的能力。随后,利用该代理模型复现黑盒模型的具体回答,并基于证据学习进行不确定性估计。大量实验验证了该方法的有效性与潜力,表明即使代理模型规模仅为目标模型的1%,也能实现可靠的不确定性量化。
原文摘要 · Abstract (English)
Large language models (LLMs) have progressed rapidly in complex reasoning and question answering, yet LLM hallucination remains a central bottleneck that hinders practical deployment, especially for commercial black-box LLMs accessible only via APIs. Existing uncertainty quantification methods typically depend on computationally expensive multiple sampling or internal parameters, which prevents real-time estimation and fails to capture information implicit in the black-box reasoning process. To address this issue, we propose Distribution-Aligned Adversarial Distillation (DisAAD), which introduces a generation-discrimination architecture to guide a lightweight proxy model to learn the high-quality regions of the output distribution of the black-box LLM, thus effectively endowing it with the ability to know whether the black-box LLM knows or not. Subsequently, we use the proxy model to reproduce the specific responses of the black-box LLM and estimate the corresponding uncertainty based on evidence learning. Extensive experiments have verified the effectiveness and promise of our proposed method, indicating that a proxy model even one that only accounts for 1\% of the target LLM's size can achieve reliable uncertainty quantification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。