SimpleHC 将 Hyper-Connections 训练开销从约 59–100% 降到 2.4%

SonglinYang4 · x · 2026-11-09

新 arXiv 论文系统研究 Hyper-Connections(HC)架构设计,指出现有实现(如 DeepSeek mHC)在 PyTorch 下会带来 59–100% 的额外训练开销,根源在于对扩展残差状态的重复访问。

论文发现两个有效交互组:不带混合的可学习读写门控、或固定门控下的无约束混合,任取其一即可达到与 mHC 相当的性能。据此提出 SimpleHC:去掉残差混合,通过共享内存访问简化执行——各层门控共享组入口状态,支持批量读取与延迟写入,标量修正精确保留前序子层输出。

结果:在最大测试宽度下,SimpleHC 相对 vanilla residual 仅增加 2.4% 端到端训练时间;从 15B 扩展到 30B MoE 模型仍保持与 mHC 相当的性能。Songlin Yang 补充指出,该做法本质上等价于 chunkwise 线性注意力——块内可并行计算 ((q^T k)v),无需显式物化扩展隐状态。

原文链接 →

「模型」频道最新

更多「模型」频道 AI 资讯 →