arXiv:2601.19089cs.LGcs.CV2026-01中稿 · the 39th Canadian …

通过渐进共享激活,提升Transformer训练与推理效率。

EPAS: Efficient Training with Progressive Activation Sharing

  • 训练中逐步从深层向浅层扩展激活共享区域,减少计算量。
  • 训练吞吐率最高提升11.1%,推理吞吐率最高提升29%。
  • 适用于资源受限场景,尤其适合持续预训练任务。

我们提出一种高效训练方法——渐进激活共享(EPAS),将渐进式训练范式与Transformer深层中冗余的QK(或KV)激活现象相结合。EPAS在训练过程中逐步扩大共享区域,将解码器层由浅入深地切换至激活共享模式,从而降低计算开销并提升吞吐率。共享区域从模型深层开始,逐步向浅层扩展。经EPAS训练的模型可在推理时根据计算预算灵活调整激活共享长度。在参数量为125M至7B的LLaMA模型上,基于QK激活共享的实证评估显示,训练吞吐率最高提升11.1%,推理吞吐率最高提升29%,同时保持与基线模型相近的损失曲线。此外,在持续预训练中应用EPAS将TinyLLaMA转为注意力共享模型,平均准确率相比现有方法最高提升10%,凸显了渐进训练在跨层激活共享中的重要性。

原文摘要 · Abstract (English)

We present a novel method for Efficient training with Progressive Activation Sharing (EPAS). This method bridges progressive training paradigm with the phenomenon of redundant QK (or KV ) activations across deeper layers of transformers. EPAS gradually grows a sharing region during training by switching decoder layers to activation sharing mode. This results in throughput increase due to reduced compute. To utilize deeper layer redundancy, the sharing region starts from the deep end of the model and grows towards the shallow end. The EPAS trained models allow for variable region lengths of activation sharing for different compute budgets during inference. Empirical evaluations with QK activation sharing in LLaMA models ranging from 125M to 7B parameters show up to an 11.1% improvement in training throughput and up to a 29% improvement in inference throughput while maintaining similar loss curve to the baseline models. Furthermore, applying EPAS in continual pretraining to transform TinyLLaMA into an attention-sharing model yields up to a 10% improvement in average accuracy over state-of-the-art methods, emphasizing the significance of progressive training in cross layer activation sharing models.

Transformer高效训练激活共享推理优化

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