跳到正文
arXiv cs.CL· Peiyu Zhang, Heng Ping, Shinibali Bhattacharyya, Anwesan Pal, Nikos Kanakaris, Kevin Cheang, Shixuan Li, Wei Yang, Paul Bogdan·· 10 小时前AI 评分34

QGPM:用查询引导的 RoPE 调制缓解长上下文 LLM 的“中间信息丢失”

Query-Guided RoPE Modulation for Lost-in-the-Middle Mitigation in Long-Context LLMs

AI 导读

研究者提出免训练方法 QGPM(Query-Guided Positional Modulation),利用模型自身注意力头得到的查询相关性重塑 RoPE 位置,缓解长上下文 LLM 的“lost-in-the-middle”问题。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:长上下文语言模型常常忽略输入中间部分的信息,这一现象被称为“中间迷失”。这一差距的部分原因可能在于旋转位置嵌入(RoPE)的远程衰减,它削弱了远距离 token 之间的位置信号。现有的输入压缩方法会删除低重要性 token,而注意力重加权只在 RoPE 已削弱该信号之后才起作用。RoPE 层面的方法直接修改位置,但对每个查询都应用相同的重映射。为解决这些局限,我们提出 QGPM(Query-Guided Positional Modulation,查询引导的位置调制),这是一种无需训练的方法,利用模型自身注意力头中的查询相关性来重塑 RoPE 位置。QGPM 为低层构建宽泛的相关性图,为高层构建锐化的相关性图。低相关性 token 在快速旋转平面上前进不足一个位置,从而将相关证据拉向查询。慢速旋转平面保持原始布局,每个 token 都留在原位。在多种上下文长度下于 NoLiMa 和 RULER 上的实验表明,QGPM 优于基础模型和强基线,在检索任务上提升最大,并且可跨多个骨干模型泛化。我们的代码可在 此 https URL 获取。
评论: 13 页,3 幅图,7 张表。代码:此 https URL
主题: 计算与语言(cs.CL);人工智能(cs.AI);机器学习(cs.LG)
引用为: arXiv:2611.09554 [cs.CL]
  (或 arXiv:2611.09554v1 [cs.CL] 用于此版本)
  https://doi.org/10.48550/arXiv.2611.09554

由 DataCite 颁发的 arXiv DOI(待注册)

提交历史

来自:Peiyu Zhang [查看邮箱]
[v1] 2026 年 11 月 6 日星期五 15:01:03 UTC(372 KB)

来源:arXiv cs.CL · arxiv.org