LinDragon 筆記圖書館2026年10月
Journal of CNN

卷積神經網路 · 優化器 · 學習率調度

Research Note

SGD(Stochastic Gradient Descent)隨機梯度下降

SGD(Stochastic Gradient Descent)隨機梯度下降

最常用的優化器之一,用來一步一步調整權重,讓 loss 變小。

核心概念:每次只看「一小批(mini‑batch)」資料算梯度,而不是整個訓練集,來更新參數。

為什麼叫「Stochastic(隨機)」?

特點:最基本的優化器,簡單但有效

optimizer = optim.SGD(
    model.parameters(),
    lr=0.01,           # 學習率(通常需要較大值)
    momentum=0.9,      # 動量(加速收斂,減少震盪)
    weight_decay=1e-4, # 權重衰減
    nesterov=False     # 是否使用 Nesterov 動量
)

參數說明:

參數說明推薦值
lr學習率0.01 ~ 0.1
momentum動量係數0.9
weight_decayL2 正則化1e-4
nesterovNesterov 加速True(可選)

優點:

缺點:

適用場景:


相關筆記