arXiv:2504.12585cs.CLcs.AI2025-04被引 2

发现大模型错误源于先验分布,通过轻量微调可显著提升确定性任务准确率。

Identifying and Mitigating the Influence of the Prior Distribution in Large Language Models

  • 通过提示避免依赖先验知识,显著改善模型表现。
  • 定位到影响决策的网络层,轻量微调后在未见数据上达到高准确率。
  • 适用于需减少幻觉、提升确定性推理的场景,如计数与缩写生成。

大型语言模型在处理计数或生成缩写等确定性任务时,常因隐含的词元序列先验分布而产生错误响应。本文表明,模型实际上已计算出正确信息,但受先验干扰未能输出。通过提示模型忽略先验知识,性能显著提升;利用机制可解释性技术定位先验相关的网络层,并对这些层进行轻量微调(基于简单提示),可在未见答案上实现高准确率。结果表明,正确答案所需信息存在于模型对问题的表征中。微调后误差不再与先验相关,且在先验主导任务中效果尤为明显。这暗示可通过干预先验依赖程度,有效提升模型在因先验导致幻觉场景下的表现。

原文摘要 · Abstract (English)

Large language models (LLMs) sometimes fail to respond appropriately to deterministic tasks -- such as counting or forming acronyms -- because the implicit prior distribution they have learned over sequences of tokens influences their responses. In this work, we show that, in at least some cases, LLMs actually compute the information needed to perform these tasks correctly, and we identify some interventions that can allow them to access this information to improve their performance. First, we show that simply prompting the language model to not rely on its prior knowledge leads to dramatic improvements in prior-dominated tasks. We then use mechanistic interpretability techniques to localize the prior within the LLM and manipulate the extent to which that prior influences its responses. Specifically, we show that it is possible to identify layers of the underlying neural network that correlate with the prior probability of a response and that lightweight finetuning of these layers with basic prompts on prior-dominated tasks achieves high performance on held-out answers. These results suggest that the information required to produce a correct response is contained within the representations of the problems formed by the models. Furthermore, we show that this finetuning is significantly more effective for prior-dominated tasks, and that the error after finetuning is no longer correlated with the prior. Our results suggest that it may be possible to define effective methods for manipulating the extent to which LLMs rely upon their priors in solving problems, potentially increasing their performance in settings where LLMs hallucinate for reasons related to the prior probability of token sequences.

大模型先验分布推理优化微调

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