arXiv:2512.13751cs.LGcs.AI2025-12

用记忆层替代冗余模块,让模型扩容更高效

MIDUS: Memory-Infused Depth Up-Scaling

  • 用记忆层替代重复的前馈网络,实现轻量级深度扩展
  • 在多个基准上提升性能,同时降低计算开销
  • 适合追求高效扩容的模型优化场景

扩大预训练语言模型可通过复制Transformer块并插入骨干网络实现。但现有方法复制了以前馈网络(FFN)为主的块,导致参数和计算成本上升,且容量增加依赖于密集的残差分支。研究表明,新增容量无需绑定于密集的FFN结构,而注意力头常扮演异质角色,提示应采用更高效的头级残差修正。本文提出记忆注入式深度扩展(MIDUS),将复制的FFN块替换为记忆层,使新增深度转化为基于检索的轻量级残差容量。引入头级记忆层(HML),结合多头乘积键记忆与头级隐式值扩展(HIVE)。HML为每个注意力头分配独立键空间,而HIVE通过紧凑投影从共享潜在库中实现头特定值。实证结果表明,该方法在性能与效率上均有提升;头重要性分析与固定检索结构分析进一步揭示,HML+HIVE是一种结构上独立、头条件化的残差扩展替代方案。

原文摘要 · Abstract (English)

Expanding pre-trained language models offers a practical way to increase capacity without training larger models from scratch. Depth Up-Scaling (DUS) does so by duplicating Transformer blocks and inserting them into a pre-trained backbone. This process also duplicates FFN-heavy blocks, increasing parameter and compute cost while adding capacity through a block-level dense residual branch. Yet prior work suggests that added capacity need not remain tied to dense FFN branches, while attention heads often play heterogeneous roles, motivating more efficient head-level residual corrections. We propose Memory-Infused Depth Up-Scaling (MIDUS), which replaces the duplicated FFN branches with memory layers and turns added depth into lightweight retrieval-based residual capacity. We introduce a Head-wise Memory Layer (HML), which combines multi-head product-key memory with Head-wise Implicit Value Expansion (HIVE). HML assigns each head a distinct key space, while HIVE realizes head-specific values from a shared latent bank through compact projections. Alongside empirical improvements in performance and efficiency, our head-importance and fixed-retrieval structural analyses characterize HML with HIVE as a structurally distinct, head-conditioned alternative to FFN-based residual expansion.

模型扩容记忆机制注意力机制高效架构

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