AdamW: Difference between revisions

From Cornell University Computational Optimization Open Textbook - Optimization Wiki
Jump to navigation Jump to search
VisualWikitext
No edit summary
No edit summary
Line 177: Line 177:
AdamW is commonly used to optimize large-scale deep learning models in areas such as natural language processing (NLP), computer vision, reinforcement learning, and generative modeling (Devlin et al., 2019; Brown et al., 2020; Dosovitskiy et al., 2021).
AdamW is commonly used to optimize large-scale deep learning models in areas such as natural language processing (NLP), computer vision, reinforcement learning, and generative modeling (Devlin et al., 2019; Brown et al., 2020; Dosovitskiy et al., 2021).


==== '''Natural Language Processing (NLP)''' ====
==== Natural Language Processing (NLP) ====
AdamW has been effectively employed in training large-scale transformer models like BERT and GPT. For BERT, improved downstream performance on NLP benchmarks has been reported compared to earlier optimizers (Devlin et al., 2019). Similarly, GPT-3’s training benefited from AdamW-like optimization for stable and efficient training (Brown et al., 2020).
AdamW has been effectively employed in training large-scale transformer models like BERT and GPT. For BERT, improved downstream performance on NLP benchmarks has been reported compared to earlier optimizers (Devlin et al., 2019). Similarly, GPT-3’s training benefited from AdamW-like optimization for stable and efficient training (Brown et al., 2020).
==== '''Computer Vision''' ====
Vision Transformers (ViT) utilize AdamW to achieve state-of-the-art results in image classification tasks. Training with AdamW improved top-1 accuracy on ImageNet compared to traditional optimizers, contributing to the success of ViT models (Dosovitskiy et al., 2021).
==== '''Reinforcement Learning''' ====
AdamW has been used in reinforcement learning scenarios where stable policy convergence is important. Empirical findings have demonstrated that AdamW leads to more predictable and stable training dynamics than standard Adam (Loshchilov & Hutter, 2017).
==== '''Generative Models''' ====
Generative models, including variants of GANs and VAEs, benefit from AdamW’s improved regularization properties. Evaluations have indicated that AdamW can result in more stable training and improved generative quality (Loshchilov & Hutter, 2017).
==== '''Time-Series Forecasting and Finance''' ====
Financial applications, such as stock price prediction, have employed AdamW to enhance training stability and predictive performance of deep learning models. Empirical studies have reported lower validation errors and reduced overfitting when using AdamW compared to standard Adam (Chen et al., 2021).
=== '''Advantages over Other Approaches''' ===
Quantitative studies have supported the superiority of AdamW over traditional Adam and other optimizers. The original AdamW paper demonstrated improved test accuracy and more stable validation losses (Loshchilov & Hutter, 2017). Devlin et al. (2019) reported that AdamW contributed to BERT’s superior performance on the GLUE benchmark, and Dosovitskiy et al. (2021) showed that ViT models trained with AdamW achieved higher accuracy than models trained with classical optimizers like SGD with momentum.


== Conclusion ==
== Conclusion ==

Revision as of 16:42, 12 December 2024

Author: Yufeng Hao (yh2295), Zhengdao Tang (zt278), Yixiao Tian (yt669), Yijie Zhang (yz3384), Zheng Zhou (zz875) (ChemE 6800 Fall 2024)

Stewards: Nathan Preuss, Wei-Han Chen, Tianqi Xiao, Guoqing Hu

Introduction

AdamW is an influential optimization algorithm in deep learning, developed as a modification to the Adam optimizer to decouple weight decay from gradient-based updates (Loshchilov & Hutter, 2017). This decoupling was introduced to address overfitting issues that often arise when using standard Adam, especially for large-scale neural network models.

By applying weight decay separately from the adaptive updates of parameters, AdamW achieves more effective regularization while retaining Adam’s strengths, such as adaptive learning rates and computational efficiency. This characteristic enables AdamW to achieve superior convergence and generalization compared to its predecessor, making it particularly advantageous for complex tasks involving large transformer-based architectures like BERT and GPT (Devlin et al., 2019; Brown et al., 2020).

As deep learning models grow in scale and complexity, AdamW has become a preferred optimizer due to its robust and stable convergence properties. Research has shown that AdamW can yield improved validation accuracy, faster convergence, and better generalization compared to both standard Adam and stochastic gradient descent (SGD) with momentum, especially in large-scale applications (Loshchilov & Hutter, 2017; Devlin et al., 2019; Dosovitskiy et al., 2021).

Algorithm Discussion

The standard Adam optimizer integrates weight decay by adding a term proportional to the parameters directly to the gradient, effectively acting as an L2 regularization term. This approach can interfere with Adam’s adaptive learning rates, leading to suboptimal convergence characteristics (Loshchilov & Hutter, 2017).

AdamW addresses this shortcoming by decoupling the weight decay step from the gradient-based parameter updates. Weight decay is applied after the parameter update is performed, preserving the integrity of the adaptive learning rate mechanism while maintaining effective regularization. This decoupling leads to more stable and predictable training dynamics, which is critical for large-scale models prone to overfitting (Loshchilov & Hutter, 2017).

Algorithm Steps

Given the parameters , a learning rate Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \alpha} , and weight decay Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \lambda} , AdamW follows these steps:

  • Initialize:
    • Initialize parameters Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_0} , the first-moment estimate Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle m_0 = 0} , and the second-moment estimate Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle v_0 = 0} .
    • Set hyperparameters:
      • Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \alpha} : learning rate
      • Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \beta_1} : exponential decay rate for the first moment
      • Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \beta_2} : exponential decay rate for the second moment
      • Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \epsilon} : small constant to avoid division by zero
  • For each time step Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle t} :
    • Compute Gradient:
      • Calculate the gradient of the objective function:
     Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g_t = \nabla_{\theta_t} f(\theta_t)}
    • Update First Moment Estimate:
      • Update the exponentially decaying average of past gradients:
     Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle m_t = \beta_1 m_{t-1} + (1 - \beta_1) g_t}
    • Update Second Moment Estimate:
      • Update the exponentially decaying average of squared gradients (element-wise square):
     Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle v_t = \beta_2 v_{t-1} + (1 - \beta_2) (g_t \odot g_t)}
 
     # where Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g_t \odot g_t}
 denotes element-wise multiplication of Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g_t}
 with itself.
    • Bias Correction:
      • Compute bias-corrected first and second moment estimates:
     Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \hat{m}_t = \frac{m_t}{1 - \beta_1^t},}

     Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \hat{v}_t = \frac{v_t}{1 - \beta_2^t}}
    • Parameter Update with Weight Decay:
      • Update parameters Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_t} with weight decay applied separately from the gradient step:
     Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{t+1} = \theta_t - \alpha \left( \frac{\hat{m}_t}{\sqrt{\hat{v}_t} + \epsilon} + \lambda \theta_t \right)}
      • This form highlights that weight decay Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \lambda \theta_t} is applied as a separate additive term to the parameter update, reinforcing the decoupling concept.

Pseudocode for AdamW

Initialize Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_0} , Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle m_0 = 0} , Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle v_0 = 0}

Set hyperparameters: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \alpha} , Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \beta_1} , Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \beta_2} , Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \lambda} , Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \epsilon}

For Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle t = 1} to Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle T} : 

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g_t = \nabla_\theta f(\theta_t)}
  # Compute gradient
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle m_t = \beta_1 m_{t-1} + (1 - \beta_1) g_t}

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle v_t = \beta_2 v_{t-1} + (1 - \beta_2) (g_t \odot g_t)}

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \hat{m}_t = \frac{m_t}{1 - \beta_1^t}}

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \hat{v}_t = \frac{v_t}{1 - \beta_2^t}}

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_t' = \theta_t - \alpha \frac{\hat{m}_t}{\sqrt{\hat{v}_t} + \epsilon}}
  # Update parameters without weight decay
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{t+1} = \theta_t' - \alpha \lambda \theta_t}
  # Apply decoupled weight decay

Numerical Examples

To demonstrate the functionality of the AdamW algorithm, a straightforward numerical example is presented. This example utilizes small dimensions and simplified values to clearly illustrate the key calculations and steps involved in the algorithm.

Example Setup

Consider the following:

  • Initial parameter: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_0 = 10}
  • Learning rate: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \alpha = 0.1}
  • Weight decay: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \lambda = 0.01}
  • First-moment decay rate: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \beta_1 = 0.9}
  • Second-moment decay rate: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \beta_2 = 0.999}
  • Small constant: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \epsilon = 10^{-8}}
  • Objective function gradient: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g_t}

For this example, assume we have a simple quadratic function: 

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle f(\theta) = \theta^2}

The gradient of this function is: 

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g_t = 2 \theta_t}

Step-by-Step Calculation

Initialization

  • First moment estimate: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle m_0 = 0}
  • Second moment estimate: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle v_0 = 0}
  • Initial parameter: Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_0 = 10}

Iteration 1

  • Step 1: Compute Gradient:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g_1 = 2 \times \theta_0 = 2 \times 10 = 20}
  • Step 2: Update First Moment Estimate:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle m_1 = \beta_1 m_0 + (1 - \beta_1) g_1 = 0.9 \times 0 + 0.1 \times 20 = 2}
  • Step 3: Update Second Moment Estimate:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle v_1 = \beta_2 v_0 + (1 - \beta_2) g_1^2 = 0.999 \times 0 + 0.001 \times 20^2 = 0 + 0.4 = 0.4}
  • Step 4: Bias Correction for First Moment:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \hat{m}_1 = \frac{m_1}{1 - \beta_1^1} = \frac{2}{1 - 0.9} = \frac{2}{0.1} = 20}
  • Step 5: Bias Correction for Second Moment:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \hat{v}_1 = \frac{v_1}{1 - \beta_2^1} = \frac{0.4}{1 - 0.999} = \frac{0.4}{0.001} = 400}
  • Step 6: Parameter Update with Weight Decay:
    • Gradient Update:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{1} = \theta_{0} - \alpha \times \frac{\hat{m}_1}{\sqrt{\hat{v}_1} + \epsilon} = 10 - 0.1 \times \frac{20}{\sqrt{400} + 10^{-8}}}
    • Simplify the denominator:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \sqrt{\hat{v}_1} + \epsilon = \sqrt{400} + 10^{-8} = 20 + 10^{-8}}
    • Compute the update:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{1} = 10 - 0.1 \times \frac{20}{20 + 10^{-8}} = 10 - 0.1 \times 1 = 9.9}
    • Weight Decay:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{1} = \theta_{1} - \alpha \times \lambda \times \theta_{0} = 9.9 - 0.1 \times 0.01 \times 10 = 9.9 - 0.01 = 9.89}
    • Updated Parameter:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{1} = 9.89}

Iteration 2

  • Step 1: Compute Gradient:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g_1 = 2 \times \theta_0 = 2 \times 10 = 20}
  • Step 2: Update First Moment Estimate:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle m_1 = \beta_1 m_0 + (1 - \beta_1) g_1 = 0.9 \times 0 + 0.1 \times 20 = 2}
  • Step 3: Update Second Moment Estimate:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle v_1 = \beta_2 v_0 + (1 - \beta_2) g_1^2 = 0.999 \times 0 + 0.001 \times 20^2 = 0 + 0.4 = 0.4}
  • Step 4: Bias Correction for First Moment:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \hat{m}_1 = \frac{m_1}{1 - \beta_1^1} = \frac{2}{1 - 0.9} = \frac{2}{0.1} = 20}
  • Step 5: Bias Correction for Second Moment:
  Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \hat{v}_1 = \frac{v_1}{1 - \beta_2^1} = \frac{0.4}{1 - 0.999} = \frac{0.4}{0.001} = 400}
  • Step 6: Parameter Update with Weight Decay:
    • Gradient Update:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{1} = \theta_{0} - \alpha \times \frac{\hat{m}_1}{\sqrt{\hat{v}_1} + \epsilon} = 10 - 0.1 \times \frac{20}{\sqrt{400} + 10^{-8}}}
    • Simplify the denominator:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \sqrt{\hat{v}_1} + \epsilon = \sqrt{400} + 10^{-8} = 20 + 10^{-8}}
    • Compute the update:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{1} = 10 - 0.1 \times \frac{20}{20 + 10^{-8}} = 10 - 0.1 \times 1 = 9.9}
    • Weight Decay:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{1} = \theta_{1} - \alpha \times \lambda \times \theta_{0} = 9.9 - 0.1 \times 0.01 \times 10 = 9.9 - 0.01 = 9.89}
    • Updated Parameter:
    Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_{1} = 9.89}

Explanations for Each Step

Step 1: The gradient is calculated based on the current parameter value. For the function Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle f(\theta) = \theta^2} , the gradient Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g_t = 2 \theta_t} represents the slope of the function at Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta_t} .

Steps 2 and 3: The first and second moment estimates (Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle m_t} and Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle v_t} ) are updated using exponentially decaying averages of past gradients and squared gradients, respectively. These updates help the optimizer adjust the learning rate dynamically for each parameter, improving efficiency.

Steps 4 and 5: Bias correction is applied to the moment estimates to address their initial bias toward zero. This correction is particularly important during the early stages of optimization, ensuring more accurate estimates.

Step 6: The parameter is updated in two key parts:

  • Gradient Update: The parameter is adjusted in the opposite direction of the gradient. This adjustment is scaled by the learning rate and adapted using the corrected moment estimates.
  • Weight Decay: A regularization term is applied by reducing the parameter's value slightly. This encourages smaller parameter values, which helps to prevent overfitting.

By repeatedly performing these steps, the AdamW optimizer effectively moves the parameters closer to the function's minimum while controlling overfitting through the use of decoupled weight decay.

Application

Areas of Application

AdamW is commonly used to optimize large-scale deep learning models in areas such as natural language processing (NLP), computer vision, reinforcement learning, and generative modeling (Devlin et al., 2019; Brown et al., 2020; Dosovitskiy et al., 2021).

Natural Language Processing (NLP)

AdamW has been effectively employed in training large-scale transformer models like BERT and GPT. For BERT, improved downstream performance on NLP benchmarks has been reported compared to earlier optimizers (Devlin et al., 2019). Similarly, GPT-3’s training benefited from AdamW-like optimization for stable and efficient training (Brown et al., 2020).

Conclusion

AdamW is a highly effective optimization algorithm for training large-scale deep learning models. Its key innovation—decoupling weight decay from gradient-based parameter updates—preserves the adaptive learning rate mechanism, leading to improved generalization and stable convergence (Loshchilov & Hutter, 2017). These properties make AdamW well-suited for modern architectures, including transformer-based models in NLP and computer vision, as well as for applications in reinforcement learning, generative modeling, and time-series forecasting (Devlin et al., 2019; Dosovitskiy et al., 2021; Chen et al., 2021).

As deep learning continues to evolve, AdamW is likely to remain a critical tool. Future work may involve integrating AdamW with learning rate schedules, second-order optimization techniques, or further algorithmic refinements to improve efficiency and robustness under varied and challenging training conditions.

Reference

  • Brown, T. B., Mann, B., Ryder, N., Subbiah, M., et al. (2020). Language Models are Few-Shot Learners. Advances in Neural Information Processing Systems (NeurIPS). https://arxiv.org/abs/2005.14165.
  • Chen, X., Zhan, Y., Wu, W., Yang, Y., & Yang, Y. (2021). Improving Stock Movement Prediction with Adversarial Training and AdamW. IEEE Access, 9, 25842–25850. https://doi.org/10.1109/ACCESS.2021.3057083.
  • Devlin, J., Chang, M.-W., Lee, K., & Toutanova, K. (2019). BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding. Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics (NAACL). https://arxiv.org/abs/1810.04805.
  • Dosovitskiy, A., Beyer, L., Kolesnikov, A., et al. (2021). An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale. International Conference on Learning Representations (ICLR). https://arxiv.org/abs/2010.11929.
  • Loshchilov, I., & Hutter, F. (2017). Decoupled Weight Decay Regularization. arXiv preprint arXiv:1711.05101. https://arxiv.org/abs/1711.05101.
Retrieved from "https://optimization.cbe.cornell.edu/index.php?title=AdamW&oldid=7069"