用注意力头的范数投票,显著减少大模型幻觉。
NoVo: Norm Voting off Hallucinations with Attention Heads in Large Language Models
- 通过筛选与真相相关的注意力头范数,实现无需训练的零样本推理。
- 在TruthfulQA MC1上比当前最佳方法高出至少19个百分点。
- 通用性强,20个数据集中有超90%获得提升,适合高可靠性场景。
大语言模型中的幻觉仍是高风险应用中事实准确性的一大障碍。尽管表示编辑和阅读方法在减少幻觉方面取得进展,但其高度依赖专用工具且需领域内样本训练,难以扩展并易过拟合,限制了准确率提升和泛化能力。本文提出一种轻量级方法Norm Voting(NoVo),利用注意力头范数的未被挖掘潜力,在零样本多项选择题中显著提升事实准确性。NoVo采用仅需30个随机样本的高效推理算法,自动选取与真相相关联的头范数,实现对多样化数据集的无缝扩展。随后,将选中的头范数用于简单投票机制,大幅提高预测准确率。在TruthfulQA MC1上,NoVo超越当前最先进方法及所有先前方法,至少提升19个百分点。在20个不同数据集上表现出色,超过90%的数据集有显著增益,远超现有表示编辑与阅读方法。此外,NoVo在微调策略和文本对抗防御中也展现出良好前景。该方法通过头范数的有效利用,为大模型可解释性、鲁棒性和可靠性开辟新方向。
原文摘要 · Abstract (English)
Hallucinations in Large Language Models (LLMs) remain a major obstacle, particularly in high-stakes applications where factual accuracy is critical. While representation editing and reading methods have made strides in reducing hallucinations, their heavy reliance on specialised tools and training on in-domain samples, makes them difficult to scale and prone to overfitting. This limits their accuracy gains and generalizability to diverse datasets. This paper presents a lightweight method, Norm Voting (NoVo), which harnesses the untapped potential of attention head norms to dramatically enhance factual accuracy in zero-shot multiple-choice questions (MCQs). NoVo begins by automatically selecting truth-correlated head norms with an efficient, inference-only algorithm using only 30 random samples, allowing NoVo to effortlessly scale to diverse datasets. Afterwards, selected head norms are employed in a simple voting algorithm, which yields significant gains in prediction accuracy. On TruthfulQA MC1, NoVo surpasses the current state-of-the-art and all previous methods by an astounding margin -- at least 19 accuracy points. NoVo demonstrates exceptional generalization to 20 diverse datasets, with significant gains in over 90\% of them, far exceeding all current representation editing and reading methods. NoVo also reveals promising gains to finetuning strategies and building textual adversarial defence. NoVo's effectiveness with head norms opens new frontiers in LLM interpretability, robustness and reliability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。