arXiv:2510.02239cs.LGmath.OC2025-10被引 7

只更新部分层,让模型训练更快更稳

Drop-Muon: Update Less, Converge Faster

  • 按随机计划只更新部分网络层,结合分层非欧优化
  • 相同精度下比全层更新快1.4倍,且有严格收敛保障
  • 适合追求高效训练的大规模模型开发者

深度学习优化中普遍认为应每步更新所有层,包括最新先进优化器Muon。本文挑战这一假设,证明全网更新在理论上和实践中均可能非最优。提出非欧随机渐进训练框架Drop-Muon:每步仅按随机策略更新部分层,融合渐进训练效率与分层非欧更新,实现顶尖性能。在层平滑性和层(L^0, L^1)平滑性条件下,首次为随机梯度设置下的渐进训练提供严谨收敛保证。成本分析显示,除非层平滑常数满足特定关系,否则全网更新并非最优。控制实验表明,Drop-Muon在保持相同精度前提下,相比全网更新的Muon,壁钟时间提速最高达1.4倍。

原文摘要 · Abstract (English)

Conventional wisdom in deep learning optimization dictates updating all layers at every step-a principle followed by all recent state-of-the-art optimizers such as Muon. In this work, we challenge this assumption, showing that full-network updates can be fundamentally suboptimal, both in theory and in practice. We introduce a non-Euclidean Randomized Progressive Training method-Drop-Muon-a simple yet powerful framework that updates only a subset of layers per step according to a randomized schedule, combining the efficiency of progressive training with layer-specific non-Euclidean updates for top-tier performance. We provide rigorous convergence guarantees under both layer-wise smoothness and layer-wise $(L^0, L^1)$-smoothness, covering deterministic and stochastic gradient settings, marking the first such results for progressive training in the stochastic and non-smooth regime. Our cost analysis further reveals that full-network updates are not optimal unless a very specific relationship between layer smoothness constants holds. Through controlled CNN experiments, we empirically demonstrate that Drop-Muon consistently outperforms full-network Muon, achieving the same accuracy up to $1.4\times$ faster in wall-clock time. Together, our results suggest a shift in how large-scale models can be efficiently trained, challenging the status quo and offering a highly efficient, theoretically grounded alternative to full-network updates.

优化器训练加速深度学习

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