跳到正文
arXiv cs.LG· El Mahdi Chayti, Martin Jaggi·· 7 小时前AI 评分22

随机动量传输:用廉价预测器降低 STORM 类优化器的 oracle 调用成本

Randomized Momentum Transport with Cheap Predictors

AI 导读

研究者提出随机动量传输方法,以概率 q 计算动量修正、其余时候用廉价预测替代,并通过重要性加权保持条件均值,代价是方差增加 (q⁻¹-1) 倍的预测损失。在传输偏差、二阶矩和残差控制假设下,包括 Muon 谱方向在内的 LMO 更新对任意固定 q>0 保持 O(T⁻¹/3) 的稳定性界。

来源:arXiv cs.LG · arxiv.org