解读大型语言模型的偏见-电子发烧友网

在一项新的研究中，研究人员发现了 LLM 中某种偏见的根本原因，为更准确、更可靠的 AI 系统铺平了道路。

研究表明，大型语言模型（LLM）往往倾向于过分强调文档或对话开头和结尾的信息，而相对忽略中间部分。这种“位置偏差”意味着，若律师使用LLM支持的虚拟助手检索30页宣誓书中的某个短语，该短语位于第一页或最后一页时被找到的可能性会大得多。

麻省理工学院的研究人员揭示了这种现象背后的机制。他们构建了一个理论框架，用以探究信息在构成LLM基础的机器学习架构中的流动方式。研究发现，某些控制模型处理输入数据方式的设计选择可能会引发位置偏差。实验表明，模型架构，特别是那些影响信息在模型内词元间传播方式的部分，可能会导致或加剧位置偏差，而训练数据同样可能是造成该问题的一个因素。

除了查明位置偏差的根源外，他们的框架还可用于在未来模型设计中诊断和纠正这一问题。这有望使聊天机器人在长时间对话中更好地保持主题连贯性，让医疗AI系统在处理大量患者数据时推理更加公平，也能使代码助手更均衡地关注程序的所有部分。

“这些模型如同黑匣子，因此作为LLM用户，您或许不会意识到位置偏差可能导致模型表现不一致。您可能仅按照任意顺序向模型提供文档，并期望其正常运行。然而，通过深入理解这些黑匣子模型的潜在机制，我们能够针对性地解决其局限性，从而改进模型性能。”麻省理工学院数据、系统和社会研究所（IDSS）和信息与决策系统实验室（LIDS）的研究生、该研究论文的第一作者Xinyi Wu说道。

1、分析注意力

Claude、Llama 和 GPT - 4 等 LLM（大型语言模型）由一种名为 transformer 的神经网络架构提供支持。Transformer 架构主要用于处理顺序数据，它能够将句子分解为称为词元的多个小块，随后通过学习这些词元之间的关系来预测下一个单词。由于其配备了注意力机制，这些模型在这一任务上表现得十分出色。注意力机制由互连的数据处理节点层构成，它允许词元有选择地关注或聚焦于相关的词元，从而更好地理解上下文。

然而，问题在于，若每个词元都要处理 30 页文档中的所有其他词元，那么计算成本将会迅速攀升，变得难以承受。鉴于此，工程师在构建 transformer 模型时，通常会运用注意力掩蔽技术来限定一个词元可以关注的词元范围。

以因果掩码为例，它仅允许一个词元关注其前面的词元。除此之外，工程师还会采用位置编码技术，以此帮助模型准确把握句子中每个词元所处的位置，进而提升模型的性能。

研究人员构建了一个基于图的理论框架，借此深入探究这些建模选择（包括注意力掩码和位置编码）对位置偏差所产生的影响。研究分析发现，因果掩码致使模型对输入内容的开头部分存在天然的偏爱，即便数据本身并无此倾向。即使前面的词元对于句子的整体含义作用不大，因果掩码仍会使 transformer 模型更加关注句子的开头部分。而且，随着模型规模的扩大以及注意力层数的增加，这种位置偏差还会进一步加剧，因为输入的早期部分在模型的推理过程中被更为频繁地使用。

不过，位置编码在这方面也发挥了一定的 “积极作用”，它能够将一个词元与附近的词元更紧密地联系起来，从而在一定程度上缓解位置偏差问题，遗憾的是，这种缓解效果在多层模型中往往会遭到稀释。

随着模型的不断增长，注意力机制的额外层级所引入的偏差会被逐渐放大，原因在于输入的早期部分在模型的推理过程中被更为频繁地使用。此外，位置编码技术通过将单词与其附近的单词紧密关联，能够将模型的注意力重新引导至正确的位置，进而减轻位置偏差，但这一效果在拥有众多注意力层的模型中可能会被削弱。需要指出的是，这些设计选择仅仅是导致位置偏差的原因之一，部分位置偏差或许还源自模型用于学习如何确定序列中单词优先级的训练数据。

“假如你知道你的数据在某种程度上存在偏差，那么除了对建模选择进行调整之外，你还应当对模型进行微调。”Wu 表示。

2、实验验证：U形模式下的“迷失在中间”

在建立理论框架后，研究人员展开了实验。实验中，他们系统性地改变正确答案在文本序列中的位置，以完成信息检索任务。

实验揭示了一种 “迷失在中间” 现象，即检索精度呈现出 U 形模式。当正确答案位于序列开头时，模型性能最佳；越接近序列中间，性能下降得越显著；而当正确答案接近序列末尾时，性能又会有所回升。

总体而言，他们的研究指出，采用不同的掩码技术、减少注意力机制的额外层级或有针对性地运用位置编码，能够降低位置偏差，进而提升模型的准确性。

通过理论分析和实验探究，研究人员展现了模型设计选择带来的影响，凸显了在高风险应用里把握模型有效性的重要性，并且他们打算继续深入研究位置编码及相关策略，进一步挖掘位置偏差的利用潜力。斯坦福大学教授 Amin Saberi 对团队在理解 Transformer 注意力机制上取得的成果予以高度评价，认为该项工作不仅数学推导清晰严谨，而且对实际问题有着深刻洞察。

3、如何减少LLM在处理信息时的位置偏差？