arXiv:2606.03825cs.LGcs.CL2026-06

动态短卷积提升Transformer,让模型更高效更强大。

Dynamic Short Convolutions Improve Transformers

  • 用输入相关滤波器的动态短卷积替代静态卷积,增强局部性与表达力。
  • 在150M到2B参数模型中均优于标准Transformer,计算效率提升1.6倍。
  • 适用于Transformer、Mamba-2和专家混合模型,适合追求性能的开发者。

Transformer已成为大语言模型的主流架构,得益于注意力机制、前馈层、残差连接和归一化的可扩展性与灵活性。本文引入动态短卷积作为改进Transformer的新神经网络组件。与静态短卷积不同,动态卷积采用输入依赖的滤波器,在保持卷积局部性偏置的同时提升表达能力。实验表明,将动态短卷积应用于键、查询和值表示,可显著提升复杂关联回忆任务的表现,优于静态卷积变体。在参数规模从150M到2B的语言建模实验中,动态卷积持续超越标准Transformer及加入静态短卷积的变体。拟合缩放定律显示:当动态卷积作用于键、查询、值向量时,相比计算量相当的Transformer有1.33倍的计算优势;若在每个线性层后添加动态卷积,则优势达1.60倍。该方法还提升了线性RNN(如Mamba-2/Gated DeltaNet)和混合专家架构的性能。通过自定义Triton内核,实现了高效训练,端到端延迟可控。结果表明,动态短卷积是一种可扩展、硬件友好且表达力强的先进神经原语。

原文摘要 · Abstract (English)

Transformers have become the dominant architecture for large language models, largely due to the scalability and flexibility of attention, feed-forward layers, residual connections, and normalization. This paper introduces dynamic short convolutions as an additional neural network primitive for improving Transformers. Unlike static short convolutions, dynamic convolutions use input-dependent filters, which preserves the locality bias of convolution while increasing expressivity. Motivating experiments show that applying dynamic short convolutions to key, query, and value representations improves performance on challenging associative recall tasks compared with static convolutional variants. Across language-modeling experiments ranging from 150M to 2B parameters, dynamic convolutions consistently outperform standard Transformers and Transformers augmented with static short convolutions. Fitting scaling laws indicates a 1.33$\times$ compute advantage over compute-matched Transformers when dynamic convolutions are applied to the key, query, and value vectors, and a 1.60$\times$ advantage when adding dynamic convolutions after every linear layer. Dynamic convolutions also offer improvements on linear RNNs (Mamba-2/Gated DeltaNet) and mixture-of-experts architectures. We make these gains practical with custom Triton kernels that enable efficient training with a manageable end-to-end slowdown. These results suggest that dynamic short convolutions are a scalable, hardware-efficient, and expressive primitive for advancing Transformer-based language models.

Transformer动态卷积模型优化高效计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。