arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2610.13630cs.CLcs.CV
收藏

多流注意力残差

Multi-Stream Attention Residuals

School of AI, Shanghai Jiao Tong University(上海交通大学人工智能学院);Shanghai Innovation Institute(上海创新研究院);Dots Studio, Xiaohongshu Inc.(小红书股份有限公司Dots工作室);School of CS, Peking University(北京大学计算机学院)

Xiaohan Qin, Xiangdong Zhang, Tuo Dai, Sunan Zou, Huaijin Wu, Yebin Yang, Zhuo Xia, Lin Yao, Yuliang Liu, Junchi Yan

AI总结:

该研究提出多流注意力残差(MSAR),将注意力残差扩展为多头深度注意力,在18B、28B MoE规模的12个下游基准中性能优于现有残差变体,等损失计算加速达1.58倍,是大规模预训练的有前景残差架构。

摘要:

注意力残差(AttnRes)支持基于内容的网络深度检索,可视为沿深度轴的单流softmax注意力。受序列轴多头注意力成功的启发,我们引入多流注意力残差(MSAR),它是AttnRes的扩展,将该机制推广到多头深度注意力,以极小的额外开销实现了显著的性能提升。MSAR通过多尺度因果卷积从序列上下文获取流多样性,使不同头能通过不同上下文视图检索历史表示,同时保留隐藏表示的完整性。检索到的流通过轻量自适应混合组合,生成单流层输入。通过保留单流源并按需扩展,MSAR支持多流计算,无需持久的多流存储或通信。在18B和28B MoE规模的全部12个下游基准测试中,MSAR始终优于现有残差变体,包括标准残差连接、流形约束超连接(mHC)和AttnRes。缩放律实验显示,与标准残差基线相比,MSAR实现了1.58倍的等损失计算加速,而AttnRes为1.35倍,mHC为1.30倍。这些结果使MSAR成为大规模预训练的有前景的残差架构。

英文摘要:

Attention Residuals (AttnRes) enables content-dependent retrieval over network depth, which can be viewed as single-stream softmax attention along the depth axis. Inspired by the success of multi-head attention along the sequence axis, we introduce Multi-Stream Attention Residuals (MSAR), an extension of AttnRes that generalizes this mechanism to multi-head depth attention, yielding substantial performance gains with minimal additional overhead. MSAR draws stream diversity from sequence context through multi-scale causal convolutions, allowing different heads to retrieve historical representations through distinct contextual views while preserving the integrity of the hidden representation. The retrieved streams are combined through lightweight adaptive mixing to generate the single-stream layer input. By retaining single-stream sources and expanding them on demand, MSAR supports multi-stream computation without requiring persistent multi-stream storage or communication. Across all 12 downstream benchmarks at the 18B and 28B MoE scales, MSAR consistently outperforms prior residual variants, including standard residual connections, manifold-constrained Hyper-Connections (mHC), and AttnRes. Scaling-law experiments yield an iso-loss compute speedup of $1.58\times$ over the standard residual baseline, compared with $1.35\times$ for AttnRes and $1.30\times$ for mHC. These results make MSAR a promising residual architecture for large-scale pre-training.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑