arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2602.21798cs.LGcs.AI

激发:专家的动量

Excitation: Momentum For Experts

Aleph Alpha Research(Aleph Alpha 研究所)

Sagi Shaier

AI总结:

Excitation是一种无需额外参数的优化框架,通过动态调节专家更新提升稀疏架构模型的训练效率和性能。

摘要:

我们提出了Excitation,一种新的优化框架,旨在加速稀疏架构(如混合专家MoEs)中的学习。与传统优化器将所有参数统一处理不同,Excitation利用批次层面的专家利用率动态调节更新。它引入了竞争更新动态,放大对高利用率专家的更新,并可选择性抑制低利用率的专家,从而有效提升路由的专业性。值得注意的是,我们发现了深度MoEs中的

英文摘要:

We propose Excitation, a novel optimization framework designed to accelerate learning in sparse architectures such as Mixture-of-Experts (MoEs). Unlike traditional optimizers that treat all parameters uniformly, Excitation dynamically modulates updates using batch-level expert utilization. It introduces a competitive update dynamic that amplifies updates to highly-utilized experts and can selectively suppress low-utilization ones, effectively sharpening routing specialization. Notably, we identify a phenomenon of "structural confusion" in deep MoEs, where standard optimizers fail to establish functional signal paths; Excitation acts as a specialization catalyst, "rescuing" these models and enabling stable training where baselines remain trapped. Excitation is optimizer-, domain-, and model-agnostic, requires minimal integration effort, and introduces neither additional per-parameter optimizer state nor learnable parameters, making it highly viable for memory-constrained settings. Across language and vision tasks, Excitation consistently improves convergence speed and final performance in MoE models, indicating that active update modulation is a key mechanism for effective conditional computation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑