Daily Tech Briefing
AI 科技速览
每天 5 分钟内学习 AI。获取最新的人工智能新闻,理解其重要性,并学习如何将其应用于您的工作。
arXiv Machine Learning · 2026/7/31 17:54:46
Sign compression for Muon: SignMuon, MuonSign, and the Limits of Error Feedback
AI 中文解读
Muon优化器迎来了一项大胆的“减法”实验:把每个参数的更新信息压缩成仅1比特的正负号,大幅降低AI训练时的通信成本。这篇论文最吸引人的地方在于,理论证明“不该收敛”的方法,在真实测试中却是最强表现,让严谨的科学结论和实际效果罕见地唱起了反调。
简单说,AI学习就像在黑暗中摸索下山的路,优化器负责判断下一步该往哪走。这次的新方法把“走多远”的信息全丢掉,只保留“往哪个方向”,消息自然精简到极致。研究人员用数学证明了某几种压缩方式在某些情况下会迷路甚至爬坡,但把压缩放在不同位置,情况又不同。最有趣的是,在一系列真实任务(如图片识别、联邦学习和语言模型训练)中,那个数学上“会迷路”的启发式做法,跑出的成绩反而碾压那些有坚实理论保障的版本。
这次发现提醒我们,理论最优未必是实战最优。未来训练超大模型时,每步只传1比特的技术有望显著降低数据中心和手机间的通信成本,让联邦学习更快、更省钱,普通人也会用上更聪明又更省电的AI应用。
SignMuon compresses the Muon update to one bit per parameter by taking its elementwise sign, providing the most direct way to run a matrix-aware optimizer under an extremely low communication budget. It outperforms SignSGD in practice, yet it can ascend even on a linear function. Signing the gradient before the Linear Minimization Oracle (LMO), rather than after, does not repair this: we construct a small explicit instance on which sign-before (MuonUSign) and sign-on-both-sides (MuonSign) ascend as well, so no placement of the sign around the oracle descends in general. Error feedback, the standard remedy for a biased compressor, does not rescue SignMuon: when applied to Muon's output, error feedback can fail for every smoothness constant, step size, and momentum. Applied to the gradient, error feedback does work, and EF21-MuonUSign and EF21-MuonSign attain the standard $\mathcal{O}(T^{-1/2})$ rate for the squared gradient norm on smooth nonconvex problems, the latter at one bit in each direction. Experiments then reverse the ordering: across centralized CIFAR-10, federated CIFAR-10, and the nanoGPT speedrun, the strongest compressed method is consistently sign-after-the-LMO, precisely the placement we prove divergent, with the provably convergent variants trailing it. Compressing after the LMO, a heuristic, matters more at these scales than the guarantee does.
分享
阅读原文 ↗