全部文章0

科学空间苏剑林··访问 1

为什么官方版Muon比MuP版多出一个max(1, ⋅)?

原网页
原始内容

在文章《Muon优化器指南:快速上手与关键细节》中,我们罗列了Muon的几个版本,它们的区别是学习率的矩阵形状相关的缩放因子不同,其中“官方版(KellerJordan版)”只比“MuP版”多出了一个$\max(1,\cdot)$的截断操作。本文就来专门讨论一下,这个截断操作是怎么来的呢?

几个版本

Muon的更新规则,可以统一地写成
\begin{equation}\begin{aligned}\newcommand{msign}{\mathop{\text{msign}}}
\boldsymbol{M}_t =&\, \beta \boldsymbol{M}_{t-1} + \boldsymbol{G}_t \\[5pt]
\boldsymbol{W}_t =&\, \boldsymbol{W}_{t-1} - \eta_t (\alpha \msign(\boldsymbol{M}_t) + \lambda \boldsymbol{W}_{t-1})
\end{aligned}\end{equation}

[...]