From Interpretability to Performance: Optimizing Retrieval Heads for Long-Context Language Models
从可解释性到性能:优化长上下文语言模型的检索头
机构 * Department of Computer Science, Institute of Science Tokyo(东京科学研究所计算机科学系)
专题命中 知识编辑与模型理解 :language model(title);LLM(abstract_cn);分类 cs.CL
AI总结 本文研究检索头如何提升长上下文语言模型性能,提出RetMask方法通过对比正常输出与屏蔽检索头的输出生成训练信号,显著提升生成和重排序性能,验证了检索头的功能作用。
Comments Findings of ACL 2026; Source code available at https://github.com/YoumiMa/RetMask