跳到正文
OpenReview · State space models· Liliang Ren; Yang Liu; Shuohang Wang; Yichong Xu; Chenguang Zhu; ChengXiang Zhai·· 2023-09-22精选AI 评分79

用于高效序列建模的稀疏模块激活

Sparse Modular Activation for Efficient Sequence Modeling

AI 导读

基于摘要分析。本文研究通用序列建模中 SSM 与自注意力混合模型的质量—效率权衡,提出 Sparse Modular Activation(SMA),以可微方式为不同序列元素动态、稀疏地激活子模块,而非对所有元素统一执行注意力计算。 核心机制是让序列元素跳过未激活的子模块。作者据此构建 SeqBoat:利用 SSM 学到的状态表示,选择性激活 Gated Attention Unit(GAU),并将局部注意力限制在被激活的输入上。作者称,这种设计可在训练和推理阶段减少计算及内存消耗,并使 SeqBoat 兼具线性推理复杂度与理论上无限的注意力跨度;其质量—效率权衡优于基于分块的模型。复杂度结论的具体假设,以及注意力跨度如何实现,需查阅正文核对。 作者报告,在长序列建模、语音分类和语言建模实验中,SeqBoat 在具有线性复杂度的混合模型范围内取得新的最佳结果,学习到的稀疏激活模式还能反映各任务所需的注意力使用量。摘要未提供具体数据集、划分、指标、性能数值或硬件条件,实验协议、消融及统计信息尚未验证。摘要提供了公开代码仓库,但实现与复现条件尚未核验。 平台推测(待验证):其与 EEG 的潜在对应是利用 SSM 表征连续时间依赖,仅对部分时间位置执行较昂贵的注意力计算,而非已验证的 BCI 方法。可复用部分是动态模块选择思路;输入编码、时间粒度与跨通道信息融合需针对 EEG 改造。假设稀疏选择能够保留任务相关片段,则可能降低长时 EEG 序列的计算开销;但低信噪比、跨被试差异及小数据训练可能导致选择机制偏向伪迹或遗漏弱信号。一个可检验的小实验是在固定 EEG 数据划分、预处理和训练预算下,对比纯 SSM、静态注意力混合模型与 SMA 变体,同时记录任务指标、激活比例、时延和峰值内存。已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对其基于 SSM 状态动态选择注意力计算的机制,以及稀疏激活模式与任务性能、训练和推理开销之间的关系;摘要中的效率与性能结论仍需结合全文协议验证。

来源:OpenReview · State space models · openreview.net