Dynamic Short Convolutions Improve Transformers
动态短卷积:用输入依赖的局部混合增强 Transformer
🧭 动态短卷积增强 Transformer 架构 ★ 深度解读
Massachusetts Institute of Technology · MIT-IBM Watson AI Lab
提出输入依赖的短卷积替代静态卷积,在 QKV 或全部线性层上带来 1.33–1.60× 计算优势,并保持低训练开销。
📌 工作参考价值(4 点)
- 在推荐、Agent 序列建模中,可将动态短卷积作为一种局部上下文交互层,放置在 QKV 或所有线性层后,增强模型对内容依赖的局部组合能力。
- 参数效率方面,使用低秩分解(rank R)或 head‑wise 共享动态卷积分量,在参数量仅小幅增加的前提下获得明显性能提升,适合资源敏感的业务场景。
- 工程实现上,可借鉴 Triton 内核融合(将动态权重生成与卷积操作熔合,避免中间张量写回 HBM)以减少内存带宽压力,使训练吞吐仅下降约 8%,适合实际部署。
- 动态卷积可与现有序列混合器(如 Attention、Mamba、DeltaNet)组合使用,替换其中的静态卷积,无需改动整体架构设计即能获得额外收益。