大模型何时承认错误?关键在于它是否‘相信’自己错了。
When Do LLMs Admit Their Mistakes? Understanding The Role Of Model Belief In Retraction
- 用内部表征的线性探测器衡量模型的即时信念,预测其是否纠错。
- 模型仅在自我判断错误时才纠错,且这种信念常与知识库答案不符。
- 训练可提升模型信念准确性,从而增强纠错能力,适合可信AI研究者阅读。
我们研究了大语言模型在何时选择收回错误答案——即自发且立即承认先前错误陈述的行为。通过特定测试平台发现,尽管模型具备纠错能力,但实际纠错频率极低,即使在独立交互中能识别错误。我们发现一个可靠的纠错预测因子:基于模型内部表示的线性探测器所测得的瞬时信念。该探测器在无关数据集上训练以预测答案正确性,再应用于需纠错的情境。模型仅在生成时“相信”答案错误时才会收回;此类信念常与参数化知识(通过事实问答测得)不一致。引导实验表明,信念对纠错具有因果影响:当模型认为答案错误时,会主动尝试验证,并改变注意力分布以促进收回。最后,我们发现监督微调利用了现有信念-纠错机制,主要通过帮助模型建立更准确的内部信念来提升纠错表现。
原文摘要 · Abstract (English)
We study the internal mechanisms that govern when LLMs choose to retract wrong answers, i.e., spontaneously and immediately acknowledge errors in their previously generated false assertions. Using model-specific testbeds, we find that while LLMs are capable of retraction, they do so only rarely, even when they can recognize their mistakes when asked in a separate interaction. We identify a reliable predictor of retraction: the model's momentary belief, as measured by a linear probe on its internal representation. The probe is trained to predict the correctness of answers on external datasets unrelated to retraction, then applied to settings where models should retract. A model retracts only when it "believes" its answers to be incorrect during generation; these beliefs frequently diverge from models' parametric knowledge as measured by factoid questions. Steering experiments further demonstrate that model belief causally drives retraction. In particular, when the model believes its answer to be incorrect, this not only encourages the model to attempt further verification, but also alters attention dynamics to promote retraction. Finally, we show that supervised fine-tuning re-uses this existing mechanism linking belief with retraction, and primarily improves retraction performance by helping the model learn more accurate internal beliefs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。