训练小型模型学模 113 加法,Grokking 后自发用三角恒等式

tetsuoai · x · 2026-09-26

作者训练了一个小型 transformer 做 mod 113 的加法,每 20 步保存一次 embedding,观察模型从记忆转向泛化的 Grokking 过程。

可视化显示,模型 Grokking 后把数字排到了圆环上,并用三角学里的角度加法公式来实现模加法——与此前关于 Grokking 涌现傅里叶结构的研究结论一致,直观展示了模型如何自发学出数学算法。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →