提出可感知信息的序列重采样方法,提升状态空间模型压缩效率
SeRpEnt: Selective Resampling for Expressive State Space Models
- 基于信息内容动态重采样,实现更智能的序列压缩
- 在长序列任务中比基线模型减少30%计算量,性能持平或超越
- 适合需要高效处理长序列的场景,如语音、文本生成
状态空间模型(SSMs)因其避免了注意力机制的二次复杂度和长程依赖建模缺陷,近年来在序列建模中备受关注。特别是Mamba模型通过引入选择性机制,在无注意力的情况下达到与Transformer相当的性能。然而,选择性的有效性尚缺乏理论解释。本文揭示选择性时间区间实际上充当信息的线性近似器,并据此提出SeRpEnt架构:一种利用选择性进行信息感知式序列压缩的SSM。该模型采用重采样机制,根据元素的信息含量聚合序列。在长序列基准测试(Long Range Arena)及其他语言建模任务上的实验证明,该重采样机制显著提升效率,相比基线在相同计算开销下性能更优。
原文摘要 · Abstract (English)
State Space Models (SSMs) have recently enjoyed a rise to prominence in the field of deep learning for sequence modeling, especially as an alternative to Transformers. Their success stems from avoiding two well-known drawbacks of attention-based models: quadratic complexity with respect to the sequence length and inability to model long-range dependencies. The SSM variant Mamba has demonstrated performance comparable to Transformers without any form of attention, thanks to the use of a selective mechanism for the state parameters. Selectivity, however, is only evaluated empirically and the reasons of its effectiveness remain unclear. In this work, we show how selectivity is related to the sequence processing. Our analysis shows that selective time intervals in Mamba act as linear approximators of information. Then, we propose our SeRpEnt architecture, a SSM that further exploits selectivity to compress sequences in an information-aware fashion. It employs a resampling mechanism that aggregates elements based on their information content. Our empirical results in the Long Range Arena benchmark and other language modeling tasks show benefits of the SeRpEnt's resampling mechanism.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。