研究人员开发了一种名为双轴内存初始化(DAMI)的新方法,以改进门控线性注意力模型中的长上下文检索。DAMI在训练前配置关键参数,以确保更好的记忆保持和跨不同模型头和层的不同检索距离。当应用于GDN、KDA、GDN2、Mamba-2和RWKV-7等模型时,DAMI与标准的随机初始化相比,检索准确率显著提高了高达16.71个百分点,同时在语言建模和推理任务中保持了可比的性能。 AI
影响 这项新的初始化技术有望带来更强大的语言模型,并改善长上下文理解能力,从而惠及需要广泛文本处理的应用。
排序理由 该集群包含一篇学术论文,详细介绍了一种提高AI模型性能的新方法。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →