ADPTNet提出非线性SSM,兼顾长程依赖与GPU并行

论文AI 评分 75/100arXiv cs.AI待复核
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

为替代高耗能Transformer,ADPTNet被提出以同时实现数据自适应、长程依赖捕获、GPU并行及非线性循环四项特性。其基于局部拓扑共轭构建,结合线性注意力与黎曼优化,实现非线性且可预测的长期行为,并在参数控制上提供理论保证。测试显示,其在选择性复制任务优于Hawk,状态追踪优于Mamba;在序列CIFAR-10上以更少参数匹配线性SSM精度并超越Transformer。其脉冲版本SpikingADPTNet在Spiking Speech Commands数据集创下83.56%的新SOTA。此外,其恒定时间尺度支持无雅可比矩阵的Conv DEER与Forward DEER扩展,Conv DEER无额外计算开销并首次实现非线性RNN并行化。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

四条性质同时满足

这项工作把目标定在同时具备四种能力:数据自适应、能捕捉长距离依赖、可在 GPU 上并行、并且是非线性循环结构。作者认为,只有这四点齐备,才谈得上替代以 Transformer 为代表的序列建模方案。ADPTNet 的核心构造是局部拓扑共轭,做法是把线性注意力与黎曼优化组合起来,作用在静态的全局动力学之上。这样得到的系统虽然是非线性的,长期行为却仍然可以预测。论文借助动力系统理论的证明,为网络时间尺度的参数化控制提供了理论保证,作者用李雅普诺夫谱来描述这一时间尺度。

与已有方案的对比

在选择性复制任务上,ADPTNet 的成绩优于 Hawk,而 Hawk 是此前在长程记忆与自适应之间取得平衡的方案。在状态跟踪方面,它比 Mamba 这类线性状态空间模型表现更好。在 sequential CIFAR-10 上,ADPTNet 达到了与线性状态空间模型相当的准确率,并且在使用更少参数的情况下超过了包括 Transformer 在内的已有选择性模型。作者还给出了一个神经形态版本 SpikingADPTNet,它在 Spiking Speech Commands 数据集上取得了新的最优准确率。

并行仿真的两条扩展

由于 ADPTNet 的时间尺度是恒定的,作者在 DEER 并行仿真算法上做出了两种无需计算雅可比矩阵的扩展,分别称为 Conv DEER 与 Forward DEER,二者保持了相同的平均收敛性。其中 Conv DEER 除了网络本身的前向传播之外不增加额外计算开销,并且首次让非线性循环神经网络可以通过迭代卷积实现并行化。论文同时归入神经与进化计算、人工智能、分布式并行与集群计算等方向,作者为 Matei-Ioan Stan 与 Oliver Rhodes,提交时间为 2026 年 9 月 27 日。

为什么值得看

首次实现非线性RNN无额外开销并行化,且多指标超越Mamba与Transformer,为替代高耗能架构提供新解。

GPU数据集

信源1 家

  1. [1]arXiv cs.AI一手信源ADPTNet: Adaptive with Prescriptive Timescales Non-Linear SSM for Sequence Modelling

关键事实

  • ADPTNet同时实现数据自适应、长程依赖、GPU并行与非线性循环四项特性[1]

  • 选择性复制优于Hawk,状态追踪优于Mamba,序列CIFAR-10以更少参数超Transformer[1]

  • SpikingADPTNet在Spiking Speech Commands数据集达83.56%新SOTA[1]

  • Conv DEER无额外计算开销,首次通过迭代卷积实现非线性RNN并行化[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。