arXiv:2609.05275cs.AI2026-09

优化层丢弃策略,让大模型训练更快更省算力,推理也更高效。

Don't Drop Dropout: Optimizing Layer Sparsity for Efficient LLM Training and Inference

论文配图:Don't Drop Dropout: Optimizing Layer Sparsity for Efficient LLM Training and Inference
图 1 · 摘自论文原文
  • 设计最优层丢弃分布与调度,提升训练效率
  • 相同训练步数下降低损失,节省最多25%计算量
  • 支持推理阶段加速,提速1.5倍且几乎不降精度

层丢弃(layer dropout)已被证明能加速训练、提升准确率并增强对零样本层剪枝的鲁棒性。然而随着模型和数据规模扩大,层丢弃在大语言模型(LLM)预训练中已基本消失。尽管有研究指出丢弃可能降低准确率,但尚无系统研究量化或缓解此问题。本研究证实,层丢弃应被纳入最先进的LLM训练流程,建立了训练与后训练的最优实践及可扩展性分析。具体而言,在最优层分布、时间调度与优化器超参下,相同训练浮点运算量(FLOPs)时,使用层丢弃可实现更低损失;相同训练步数下,验证损失更低或相当,同时节省高达25%的训练FLOPs。此外,层丢弃显著促进后训练优化,如提前退出、中间层跳过和自推测解码,带来最高1.5倍的推理加速,精度损失可忽略。基于2400余次实验,涵盖271M至8.2B参数模型及最大160B token数据集,结果在大规模训练中一致可靠。所有预训练实验均在Cerebras CS-3系统上完成。

原文摘要 · Abstract (English)

Layer dropout (a.k.a. stochastic depth) has been shown to enable faster training, higher accuracy, and robustness to zero-shot layer pruning in both language and vision transformers. However, as models and datasets have scaled, dropout - particularly layer dropout - has largely disappeared from large language models (LLMs) pre-training recipes. While some prior work has reported that dropout can degrade accuracy, no comprehensive study has quantified, let alone mitigated, this effect. In this study, we show that layer dropout should be used in state-of-the-art LLM training, establishing best practices and scaling analysis for both training and post-training benefits. Concretely, with optimal layer distribution, time schedule, and optimizer hyperparameters, we observe that at the same training FLOPs layer dropout leads to lower loss. For a given number of training steps, LLMs can achieve lower or similar validation loss while saving upto 25% of training FLOPs. Moreover, layer dropout enables significant post-training optimizations, such as early exit, intermediate-layer skipping, and self-speculative decoding, yielding up to 1.5x inference speedup with negligible accuracy loss. Across more than 2400 training experiments, spanning models from 271M to 8.2B parameters and datasets up to 160B tokens, we demonstrate that these findings extend reliably to large-scale training regimes. All pre-training experiments were run on Cerebras CS-3 systems.

大模型训练层丢弃推理加速

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