ResonatorLM:用因果共振场混合实现高效长上下文语言建模
ResonatorLM是一种新型机制,用物理启发的因果共振函数替代注意力机制,将token序列视为一维潜场。在6M参数模型上,训练和预填充速度随序列长度增加,解码速度在32K token时达到标准Transformer的6.47倍,WikiText准确率从55.32%提升至61.31%。该论文已被ICANN 2026接收。
随着大语言模型的发展,Transformer架构因其自注意力机制而成为主流,实现了高效的并行训练。然而,在处理长上下文时,Transformer及其变体如RNN和CNN的效率会显著下降。这是因为自注意力机制的计算复杂度与序列长度呈二次关系,导致长序列下的训练和推理变得非常昂贵。针对这一问题,研究者提出了一种名为ResonatorLM的新机制,它用物理推导的因果共振场混合替代了传统的注意力计算,为长上下文建模提供了一种全新的思路。
ResonatorLM的核心思想是将token序列视为一个单一的、受驱动的一维潜场,并用阻尼谐振子的因果函数来替代注意力点积。这种设计巧妙地避免了注意力机制中的二次复杂度,使得模型能够以线性或接近线性的复杂度处理长序列。具体来说,ResonatorLM通过模拟物理系统中的共振现象,让每个token对其后续token产生因果性的影响,从而捕获序列中的依赖关系。这种物理启发的方法不仅理论上优雅,而且在实际实验中表现出了优异的性能。
研究团队在传统的网络架构上实现了ResonatorLM,并在标准的长上下文建模任务上进行了评估。实验在6M参数的小规模设置下进行,结果显示,ResonatorLM的训练和预填充速度随着序列长度的增加而显著提升。在32K token的序列长度上,解码速度达到了标准优化Transformer的6.47倍,这意味着在处理超长文档时,ResonatorLM可以大幅降低推理延迟。同时,在WikiText数据集上,ResonatorLM的准确率达到61.31%,相比基准Transformer的55.32%有显著提升,证明其不仅能提高效率,还能保持或改进模型质量。
该论文已被ICANN 2026会议接收,标志着长上下文建模领域的一项有前景的创新。ResonatorLM的提出不仅为克服Transformer的局限性提供了新的方向,还展示了物理启发方法在机器学习中的巨大潜力。未来,这项工作可能会影响模型架构的设计选择,推动更高效的长上下文应用,如文档理解、代码生成和多轮对话等。