Correcting Negative Bias in Large Language Models through Negative Attention Score Alignment

TL;DR

通过负注意力分数对齐方法,减少大型语言模型中的负偏差,提升准确率和召回率。

cs.CL 🔴 高级 2024-08-01 4 次浏览
Sangwon Yu Jongyoon Song Bongkyu Hwang Hoyoung Kang Sooah Cho Junhwa Choi Seongho Joe Taehee Lee Youngjune L. Gwon Sungroh Yoon
语言模型 负偏差 注意力机制 参数高效微调 复杂推理任务

核心发现

方法论

研究提出了负注意力分数(NAS)和负注意力分数对齐(NASA)方法,通过识别和微调负偏差注意力头来减少模型的负偏差。NASA是一种参数高效的微调技术,能够在不损害模型泛化能力的情况下显著改善精确度和召回率。

关键结果

  • NASA方法在多领域推理任务中显著减少了精确度和召回率之间的差距,同时保持或提升了模型的准确性和F1分数。例如,在StrategyQA数据集上,精确度提高了6%,召回率提高了8%。
  • 在GSM8K-Rephrased数据集上,NASA方法将召回率从33.6%提升到83.8%,显著改善了模型的负偏差。
  • 实验结果表明,NASA方法在不同提示格式下表现稳定,包括少样本设置。

研究意义

该研究通过提出负注意力分数对齐方法,解决了大型语言模型在二元决策任务中的负偏差问题,提高了模型的可靠性和推理能力。这一方法在学术界和工业界具有广泛的应用潜力,能够显著改善模型在复杂推理任务中的表现。

技术贡献

技术贡献包括提出了一种新的负注意力分数(NAS)框架,用于系统性探测模型中的负偏差注意力头,并通过NASA方法进行参数高效微调。这一方法与现有的SOTA方法相比,提供了新的理论保证和工程可能性。

新颖性

该研究首次系统性地识别和微调负偏差注意力头,通过负注意力分数对齐方法显著减少了模型的负偏差。与相关工作相比,该方法提供了更精确的负偏差检测和更高效的微调策略。

局限性

  • 在某些复杂推理任务中,NASA方法可能无法完全消除负偏差,尤其是在数据集不平衡的情况下。
  • 该方法依赖于现有数据集的质量和多样性,可能在低质量数据集上表现不佳。

未来方向

未来工作可以探索NASA方法在更多类型的推理任务中的应用,并研究如何进一步优化微调过程以提高模型的泛化能力。

AI 总览摘要

大型语言模型在处理复杂推理任务时,常常表现出负偏差,导致模型在二元决策任务中精确度高但召回率低。现有解决方案未能有效解决这一问题。

本文提出了一种新的负注意力分数对齐(NASA)方法,通过识别和微调负偏差注意力头来减少模型的负偏差。NASA是一种参数高效的微调技术,能够在不损害模型泛化能力的情况下显著改善精确度和召回率。

实验结果表明,NASA方法在多个领域推理任务中显著减少了精确度和召回率之间的差距,同时保持或提升了模型的准确性和F1分数。这一方法在学术界和工业界具有广泛的应用潜力,能够显著改善模型在复杂推理任务中的表现。

深度解读

原文摘要

A binary decision task, like yes-no questions or answer verification, reflects a significant real-world scenario such as where users look for confirmation about the correctness of their decisions on specific issues. In this work, we observe that language models exhibit a negative bias in the binary decisions of complex reasoning tasks. Based on our observations and the rationale about attention-based model dynamics, we propose a negative attention score (NAS) to systematically and quantitatively formulate negative bias. Based on NAS, we identify attention heads that attend to negative tokens provided in the instructions as answer candidate of binary decisions, regardless of the question in the prompt, and validate their association with the negative bias. Additionally, we propose the negative attention score alignment (NASA) method, which is a parameter-efficient fine-tuning technique to address the extracted negatively biased attention heads. Experimental results from various domains of reasoning tasks and large model search space demonstrate that NASA significantly reduces the gap between precision and recall caused by negative bias while preserving their generalization abilities.

cs.CL cs.AI