AMUSE: Anytime Muon with Stable Gradient Evaluation
AMUSE: 任何时刻的Muon with Stable Gradient Evaluation
机构 * KAIST(韩国科学技术院) ; KRAFTON(KRAFTON公司) ; Seoul National University(首尔国立大学)
AI总结 本文研究了Muon算法的机制,提出了一种名为AMUSE的算法,通过结合Muon的快速批量进步和Schedule-Free平均的稳定效果,实现了无需学习率调度的任何时刻训练,并在视觉任务和大语言模型预训练中提升了性能-迭代帕累托前沿。
Comments 44 pages, 27 figures