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

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

Research Note

AdaGrad

AdaGrad

常被更新的參數:步伐越來越小
很少被更新的參數:步伐相對比較大

特點:自動調整學習率,適合稀疏資料

optimizer = optim.Adagrad(
    model.parameters(),
    lr=1e-2,           # 初始學習率(可以設大一點)
    lr_decay=0,        # 學習率衰減
    weight_decay=0,    # 權重衰減
    eps=1e-10          # 數值穩定性
)

也就是說,它會根據「過去累積的梯度大小」來調整每個參數的 learning rate:

AdaGrad 有個很大的問題:

優點:

缺點:

適用場景:


相關筆記