arXiv:2509.26307cs.LG2025-09被引 1

用可解释性引导生成,让大模型更听话、更靠谱。

Attribution-Guided Decoding

  • 根据用户指定区域的贡献度选择最优输出词
  • 指令遵循成功率从66.0%提升至79.1%
  • 适合需要高可靠性和准确性的场景

大型语言模型(LLM)在遵循复杂指令和生成事实准确文本方面的能力,对其实际应用至关重要。然而,标准解码方法往往无法稳健满足这些要求,而现有的控制技术常导致输出质量下降。本文提出一种基于可解释性的解码策略——属性引导解码(AGD)。AGD不直接操作模型激活,而是考察一组高概率输出候选词,选择对用户定义的感兴趣区域(ROI)贡献度最高的一个。该ROI可灵活定义在模型输入或内部组件的不同部分,从而引导生成向多种理想行为靠拢。我们在三个挑战性任务中验证了AGD的有效性:在指令遵循任务中,其显著提升了遵循率(如在Llama 3.1上总体成功率从66.0%提升至79.1%);在知识密集型任务中,引导模型利用内部知识组件或上下文来源,可减少幻觉,在闭卷与开卷设置下均提高事实准确性。此外,我们提出一种基于熵的自适应变体,仅在模型不确定时施加引导,缓解质量下降并降低计算开销。本工作提供了一种更灵活、可解释且高效的现代大模型可靠性增强方法。

原文摘要 · Abstract (English)

The capacity of Large Language Models (LLMs) to follow complex instructions and generate factually accurate text is critical for their real-world application. However, standard decoding methods often fail to robustly satisfy these requirements, while existing control techniques frequently degrade general output quality. In this work, we introduce Attribution-Guided Decoding (AGD), an interpretability-based decoding strategy. Instead of directly manipulating model activations, AGD considers a set of high-probability output token candidates and selects the one that exhibits the highest attribution to a user-defined Region of Interest (ROI). This ROI can be flexibly defined over different parts of the model's input or internal components, allowing AGD to steer generation towards various desirable behaviors. We demonstrate AGD's efficacy across three challenging domains. For instruction following, we show that AGD significantly boosts adherence (e.g., improving the overall success rate on Llama 3.1 from 66.0% to 79.1%). For knowledge-intensive tasks, we show that guiding generation towards usage of internal knowledge components or contextual sources can reduce hallucinations and improve factual accuracy in both closed-book and open-book settings. Furthermore, we propose an adaptive, entropy-based variant of AGD that mitigates quality degradation and reduces computational overhead by applying guidance only when the model is uncertain. Our work presents a versatile, more interpretable, and effective method for enhancing the reliability of modern LLMs.

大模型解码策略可解释性事实准确

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