arXiv 论文比较 Transformer 与状态空间模型的长上下文位置偏差

论文AI 评分 62/100arXiv cs.AI待复核
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

一篇投稿至 NeurIPS 2026 PriGM 工作坊的论文,比较了 Transformer、状态空间与混合 Mamba 三类架构在长上下文问答中的证据位置偏差。实验把一份含答案文档在九个干扰文档间移动:所评测的 Transformer 表现出开头优势,状态空间模型没有;两类模型都偏好靠近末尾的证据。纯 Mamba 与混合 Mamba 的对照方向一致,但统计上不确定。该差异只出现在能完成任务的模型上。在合成检索任务中,Transformer 的开头优势再次出现,而状态空间模型因准确率触顶无法比较。同一状态空间架构更换语料会改变总体准确率,但曲线形状没有可检测变化;提示词改动可以制造或几乎消除开头优势。位置信息在各架构中均可解码,所测 Transformer 的初始 token 注意力集中程度与开头敏感性同向变化。作者称这些为探索性结果,未把序列混合器认定为唯一原因。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

实验怎么做的

研究团队设计了一个受控设置:把唯一包含答案的文档放进固定的一组九个干扰文档中,只改变它的位置,然后观察模型能否取用这份证据。比较对象覆盖三类序列混合方式——Transformer、状态空间模型以及混合 Mamba。这样做的目的是把「证据在提示中的位置」单独拎出来,看不同架构对位置的敏感程度是否不同。作者强调这是探索性研究,结论限定在所评测的模型范围内。

两类架构的偏差差异

在所评测的 Transformer 上,证据靠近提示开头时更容易被用上,形成开头优势;所评测的状态空间模型没有这一现象。但两类模型都更偏好靠近末尾的证据。纯 Mamba 与混合 Mamba 的对照指向同一方向,只是统计上还不够确定。作者指出,这种家族差异只在能够完成任务的模型之间才显现;在合成检索任务里,状态空间模型准确率触顶,无法进行家族间比较,而 Transformer 的开头优势再次出现。

语料与提示词的影响

在同一个状态空间架构内更换语料,总体准确率会变化,但位置-准确率曲线的形状没有可检测的改变。提示词的变化则能制造出开头优势,也能几乎把它消除。这说明有效上下文利用不只由序列混合器决定,还和任务、提示形式以及模型能力有关。作者据此没有把架构认定为唯一原因。

可解码性与注意力线索

论文还报告,位置信息在不同架构中都可以被解码出来,即便各架构的响应曲线并不相同。在所测试的 Transformer 家族里,初始 token 附近的注意力集中程度与开头敏感性呈同向变化,为理解开头优势提供了一个可观察的内部线索。作者将整项工作定位为探索性结果,代码已公开。

为什么值得看

长上下文模型「开头记得住、中间易丢失」的偏差是否由架构决定,这篇对比给出了可复现的实验线索。

论文

信源1 家

  1. [1]arXiv cs.AI一手信源Mixing Matters: Evidence Position Bias Across Sequence Mixers in Long-Context Question Answering

关键事实

  • 论文将一份含答案文档在固定九个干扰文档间移动,比较三类架构的证据位置偏差[1]

  • 所评测 Transformer 有开头优势,状态空间模型没有,但两者都偏好靠近末尾的证据[1]

  • 纯 Mamba 与混合 Mamba 的对照方向一致,统计上仍不确定[1]

  • 同一状态空间架构更换语料改变总体准确率,曲线形状无可检测变化[1]

  • 论文被 NeurIPS 2026 第二届 PriGM 工作坊接收,共 9 页[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。