arXiv:2602.04577cs.CL2026-02被引 4

用轻量模型蒸馏语言模型的不确定性,提升可靠性判断效率

Semantic Self-Distillation for Language Model Uncertainty

  • 用学生模型蒸馏语言模型的语义分布,实现快速不确定性估计
  • 在TriviaQA和MMLU上,预测幻觉效果媲美传统采样方法
  • 适合需要快速可靠性评估的场景,如问答系统和域外检测

大型语言模型在可解释的不确定性量化方面面临挑战,部分源于其复杂性和输出多样性。语义分散(即采样答案意义的差异)被用作不确定性代理指标,但计算成本高,难以用于低延迟场景。本文提出将采样语义分布蒸馏为轻量级学生模型,该模型在语言模型生成答案前预测提示条件下的语义密度。学生模型输出可能答案的语义分布,其熵值提供提示级别的不确定性信号,概率密度则支持答案级别的可靠性评估。在TriviaQA和MMLU上的实验表明,该方法在幻觉预测任务中性能与基于采样的语义分散基线相当,同时具备额外的域外检测和多选题答案选择能力。该技术称为语义自蒸馏(SSD),可作为复杂输出空间中预测不确定性的通用框架。

原文摘要 · Abstract (English)

Large language models present challenges for principled uncertainty quantification, in part due to their complexity and the diversity of their outputs. Semantic dispersion, or the variance in the meaning of sampled answers, has been proposed as a useful proxy for model uncertainty, but the associated computational cost prohibits its use in latency-critical applications. We show that sampled semantic distributions can be distilled into lightweight student models which estimate a prompt-conditioned density before the language model generates an answer token. The student model predicts a semantic distribution over possible answers; the entropy of this distribution provides a prompt-level uncertainty signal, and the probability density allows answer-level reliability evaluation. Across experiments on TriviaQA and MMLU, we find our student models perform competitively relative to sampling-based semantic dispersion baselines on a hallucination prediction task, whilst offering additional uncertainty primitives for out-of-domain detection and multiple-choice answer selection. We term this technique Semantic Self-Distillation (SSD), which can serve as a general framework for distilling predictive uncertainty in complex output spaces beyond language.

不确定性语言模型知识蒸馏可靠性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。