用池化与递归结构替代注意力,提升长序列建模效率
Poolformer: Recurrent Networks with Pooling for Long-Sequence Modeling
- 用池化+递归块递归压缩序列,取代耗时的自注意力
- 训练速度更快,FID和IS指标更优,有效防过拟合
- 适合处理音频、视频等长序列数据,可拓展至多模态
序列建模模型在人工智能中占据核心地位,尤其在引入Transformer架构后。尽管最初用于自然语言处理,这类模型已广泛应用于计算机视觉等领域。传统模型依赖循环或自注意力机制交换时间维度信息,但自注意力随序列长度呈平方增长,限制了其在超长序列中的应用。本文提出Poolformer,一种以递归层结合池化操作替代自注意力的序列建模模型。该模型通过跳过块(SkipBlock)实现递归设计,包含残差块、下采样池化层、嵌套跳过块、上采样池化层及额外残差块。大量实验验证了架构选择的有效性:池化显著加速训练,提升感知质量(FID和IS),并缓解过拟合。实验还表明,长程依赖由深层处理,浅层负责短期特征。在原始音频任务上,Poolformer优于SaShiMi和Mamba等先进模型。未来方向包括文本、视觉及多模态应用,如基于Poolformer的大型语言模型可高效处理图像与视频的密集表示。
原文摘要 · Abstract (English)
Sequence-to-sequence models have become central in Artificial Intelligence, particularly following the introduction of the transformer architecture. While initially developed for Natural Language Processing, these models have demonstrated utility across domains, including Computer Vision. Such models require mechanisms to exchange information along the time dimension, typically using recurrent or self-attention layers. However, self-attention scales quadratically with sequence length, limiting its practicality for very long sequences. We introduce Poolformer, a sequence-to-sequence model that replaces self-attention with recurrent layers and incorporates pooling operations to reduce sequence length. Poolformer is defined recursively using SkipBlocks, which contain residual blocks, a down-pooling layer, a nested SkipBlock, an up-pooling layer, and additional residual blocks. We conduct extensive experiments to support our architectural choices. Our results show that pooling greatly accelerates training, improves perceptual metrics (FID and IS), and prevents overfitting. Our experiments also suggest that long-range dependencies are handled by deep layers, while shallow layers take care of short-term features. Evaluated on raw audio, which naturally features long sequence lengths, Poolformer outperforms state-of-the-art models such as SaShiMi and Mamba. Future directions include applications to text and vision, as well as multi-modal scenarios, where a Poolformer-based LLM could effectively process dense representations of images and videos.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。