HelloAI ← 可视化画廊
📐 数学/优化 · 10 分钟

Loss 3D 曲面:训练在什么样的地形上走

把 loss 函数想象成一个山谷地形—— 优化器要找最低点。但真实地形有鞍点平台峡谷…… 不同地形让 SGD / Momentum / Adam 表现差异巨大。

SGD(无动量) Momentum Adam ★ 目标
拖动画布旋转视角 · 滚动缩放
控制
0.020
当前 step
0
优化器位置(loss)
SGD
Momentum
Adam
🥣 简单碗(凸函数)
最理想情况:loss 函数是凸的,从任何起点出发都能找到全局最小值。 三种优化器都收敛,但 Adam 通常最快。

⚡ 五种地形的训练含义

🥣 碗 → 容易
线性回归、逻辑回归——全部凸。任何优化器都行。
🐎 鞍点 → SGD 卡住
高维空间里鞍点比局部最小还多。Adam 用二阶信息能脱困。
🏞️ 峡谷 → 来回震荡
深度学习常见。Momentum 沿谷底加速,SGD 来回震。
🏜️ 平台 → 梯度消失
深网络早期常遇。残差连接 + LR warmup 才能爬出。
🗻 多峰 → 局部最小陷阱
不同初始化收敛到不同解。SGD 的噪声有时反而是优势。
📊 真实 LLM?
高维 loss 大部分是鞍点 + 平台,几乎没有"全局最优"概念——只是"足够好"。
配套
L2-09 优化器深度解析
SGD / Momentum / Adam / AdamW
相关
2D 梯度下降登山者
同样优化器但 2D 视角