AI Terms Dictionary

A comprehensive multilingual AI terminology dictionary

Definition

Adam (Adaptive Moment Estimation) is a popular first-order gradient-based optimization algorithm used in training deep neural networks. It combines the advantages of two other extensions of stochastic gradient descent: AdaGrad, which works well with sparse gradients, and RMSProp, which works well in online and non-stationary settings. Adam maintains exponential moving averages of both the gradient and the squared gradient to adapt the learning rate for each weight individually.

Summary

An optimization algorithm that computes adaptive learning rates for each parameter.

Key Concepts

Use Cases

Code Example

1
2
import torch.optim as optim
optimizer = optim.Adam(model.parameters(), lr=0.001)