arXiv · cs.CL· Seiya Tokui·· 4 天前AI 评分33
头维度影响语言模型的长上下文检索能力
Head Dimension Matters in Language Models for Long-Context Retrieval
来源与原文
arXiv · cs.CL
作者：Seiya Tokui
发布时间：（北京时间）
阅读原始来源 · arxiv.org ↗完整链接与内容说明
https://arxiv.org/abs/2610.32213v1
本站目前仅有 AI 导读与摘要，未展示原文正文；技术细节、代码和完整结论请查看原始来源。
AI 导读
一项 arXiv 研究发现，在固定模型规模下，更宽的注意力头维度能提升长上下文检索性能，而预训练损失和短上下文非检索任务几乎不受影响。更宽的头拥有更高比例的检索头，将更多参数用于检索。即使冻结注意力模块的投影权重，继续预训练中检索性能仍随头宽提升，表明宽头还促进了嵌入向量和 MLP 等组件的有效适配。
来源：arXiv · cs.CL · arxiv.org