跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

Toeplitz MLP混合器:低复杂度、高信息量的序列模型

文章摘要

一种名为Toeplitz MLP混合器(TMM)的新架构用三角掩蔽Toeplitz矩阵乘法替代注意力机制,实现了亚二次复杂度(训练时间O(dn log n),空间O(dn)),同时保留更多输入信息,提升了复制能力、信息检索和上下文学习准确率。

来源arXiv Machine Learning作者: Benjamin L. Badger, Ethan Roland
Toeplitz MLP混合器:低复杂度、高信息量的序列模型
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

Transformer架构在大型语言模型中占据主导地位,但其注意力机制的二次时间与空间复杂度(相对于序列长度)严重限制了模型的扩展能力。随着序列长度的增加,计算和内存需求呈平方级增长,这成为处理长文本或高分辨率数据的主要瓶颈。为了突破这一限制,研究人员提出了多种亚二次复杂度架构,但大多数都引入了复杂的输入调制或状态维护机制。

在这篇发表于2026年4月的论文中,Benjamin L. Badger和Ethan Roland提出了Toeplitz MLP混合器(TMM),一种新颖的Transformer替代方案。TMM的核心创新在于用三角掩蔽的Toeplitz矩阵乘法取代了标准的注意力机制。Toeplitz矩阵具有特殊的结构,其每条对角线上的元素相同,这使得矩阵乘法可以通过快速傅里叶变换(FFT)高效实现,从而将复杂度从二次降低到近线性。具体而言,TMM在训练时的时间复杂度为O(dn log n),空间复杂度为O(dn),其中d是隐藏维度,n是序列长度。在推理预填充阶段,其时间和空间复杂度均为O(dn)。这种高效的复杂度使得TMM在处理长序列时具有显著优势。

令人惊讶的是,尽管TMM缺乏其他亚二次架构中的复杂输入调制或状态维护,但它在训练效率上表现更优——在相同的计算量和内存消耗下,TMM能够实现更低的损失。作者认为,这是因为TMM没有引入多余的架构偏差,从而得以保留更多的输入信息。实验表明,TMM在复制任务上表现显著提升,这直接归因于其更高的信息保留能力。此外,在信息检索和上下文学习基准测试中,TMM的准确率也优于其他可比较的架构。这些结果一致表明,TMM是一种信息丰富且高效的序列模型。

最后,作者从算子指数理论的角度进行了深入分析,并得出一个反直觉的结论:在因果(causal)非可逆模型中,训练后的Toeplitz层反而比那些实际可逆模型中的层更可能是可逆的或近乎可逆的。这一理论发现为理解TMM的优越性能提供了全新的视角。总之,TMM为开发低复杂度、高信息保持力的序列模型开辟了新的途径,有望成为下一代高效Transformer的有力候选者。

展开要点与分析

文章情报

工程师进阶

要点

  • TMM使用Toeplitz矩阵替代注意力,降低复杂度且保持性能。
  • 保留更多输入信息,改善复制和上下文学习。
  • 每个计算和内存单位获得更高训练效率。
  • 算子指数理论分析表明训练后的Toeplitz层几乎可逆。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。