pytorch-optimizer 2.11.0__tar.gz → 2.11.2__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (90) hide show
  1. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/PKG-INFO +155 -146
  2. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/README.rst +152 -143
  3. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pyproject.toml +14 -11
  4. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/__init__.py +12 -2
  5. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/focal.py +20 -0
  6. pytorch_optimizer-2.11.2/pytorch_optimizer/loss/lovasz.py +50 -0
  7. pytorch_optimizer-2.11.2/pytorch_optimizer/loss/tversky.py +31 -0
  8. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adabelief.py +1 -1
  9. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adafactor.py +1 -1
  10. pytorch_optimizer-2.11.2/pytorch_optimizer/optimizer/came.py +220 -0
  11. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/dadapt.py +129 -0
  12. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/fp16.py +1 -1
  13. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lomo.py +1 -1
  14. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lookahead.py +15 -3
  15. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/rotograd.py +2 -2
  16. pytorch_optimizer-2.11.2/pytorch_optimizer/optimizer/tiger.py +90 -0
  17. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/LICENSE +0 -0
  18. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/__init__.py +0 -0
  19. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/exception.py +0 -0
  20. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/optimizer.py +0 -0
  21. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/scheduler.py +0 -0
  22. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/types.py +0 -0
  23. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/__init__.py +0 -0
  24. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/bi_tempered.py +0 -0
  25. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/cross_entropy.py +0 -0
  26. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/dice.py +0 -0
  27. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/f1.py +0 -0
  28. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/jaccard.py +0 -0
  29. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/ldam.py +0 -0
  30. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/__init__.py +0 -0
  31. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/chebyshev.py +0 -0
  32. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/cosine_anealing.py +0 -0
  33. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/experimental/__init__.py +0 -0
  34. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/experimental/deberta_v3_lr_scheduler.py +0 -0
  35. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/linear_warmup.py +0 -0
  36. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/proportion.py +0 -0
  37. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/__init__.py +0 -0
  38. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/a2grad.py +0 -0
  39. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adabound.py +0 -0
  40. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adadelta.py +0 -0
  41. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adahessian.py +0 -0
  42. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adai.py +0 -0
  43. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adamax.py +0 -0
  44. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adamod.py +0 -0
  45. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adamp.py +0 -0
  46. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adams.py +0 -0
  47. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adan.py +0 -0
  48. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adanorm.py +0 -0
  49. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adapnm.py +0 -0
  50. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adashift.py +0 -0
  51. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adasmooth.py +0 -0
  52. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/agc.py +0 -0
  53. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/aggmo.py +0 -0
  54. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/alig.py +0 -0
  55. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/amos.py +0 -0
  56. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/apollo.py +0 -0
  57. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/avagrad.py +0 -0
  58. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/diffgrad.py +0 -0
  59. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/fromage.py +0 -0
  60. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/gc.py +0 -0
  61. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/gravity.py +0 -0
  62. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/gsam.py +0 -0
  63. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lamb.py +0 -0
  64. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lars.py +0 -0
  65. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lion.py +0 -0
  66. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/madgrad.py +0 -0
  67. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/msvag.py +0 -0
  68. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/nero.py +0 -0
  69. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/novograd.py +0 -0
  70. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/padam.py +0 -0
  71. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/pcgrad.py +0 -0
  72. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/pid.py +0 -0
  73. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/pnm.py +0 -0
  74. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/prodigy.py +0 -0
  75. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/qhadam.py +0 -0
  76. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/qhm.py +0 -0
  77. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/radam.py +0 -0
  78. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/ranger.py +0 -0
  79. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/ranger21.py +0 -0
  80. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sam.py +0 -0
  81. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sgd.py +0 -0
  82. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sgdp.py +0 -0
  83. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/shampoo.py +0 -0
  84. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/shampoo_utils.py +0 -0
  85. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sm3.py +0 -0
  86. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sophia.py +0 -0
  87. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/srmm.py +0 -0
  88. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/swats.py +0 -0
  89. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/utils.py +0 -0
  90. {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/yogi.py +0 -0
@@ -1,10 +1,10 @@
1
1
  Metadata-Version: 2.1
2
- Name: pytorch-optimizer
3
- Version: 2.11.0
2
+ Name: pytorch_optimizer
3
+ Version: 2.11.2
4
4
  Summary: optimizer & lr scheduler & objective function collections in PyTorch
5
5
  Home-page: https://github.com/kozistr/pytorch_optimizer
6
6
  License: Apache-2.0
7
- Keywords: pytorch,deep-learning,optimizer,lr scheduler,A2Grad,ASGD,AccSGD,AdaBelief,AdaBound,AdaDelta,AdaFactor,AdaMax,AdaMod,AdaNorm,AdaPNM,AdaSmooth,AdaHessian,Adai,AdamP,AdamS,Adan,AggMo,AliG,Amos,Apollo,AvaGrad,DAdaptAdaGrad,DAdaptAdam,DAdaptAdan,DAdaptSGD,DiffGrad,Fromage,Gravity,GSAM,LARS,Lamb,Lion,LOMO,Lookahead,MADGRAD,MSVAG,Nero,NovoGrad,PAdam,PCGrad,PID,PNM,Prodigy,QHAdam,QHM,RAdam,Ranger,Ranger21,RotoGrad,SAM,SGDP,SGDW,SignSGD,SM3,SopihaH,SRMM,SWATS,ScalableShampoo,Shampoo,Yogi,BCE,BCEFocal,Focal,FocalCosine,SoftF1,Dice,LDAM,Jaccard,Bi-Tempered
7
+ Keywords: pytorch,deep-learning,optimizer,lr scheduler,A2Grad,ASGD,AccSGD,AdaBelief,AdaBound,AdaDelta,AdaFactor,AdaMax,AdaMod,AdaNorm,AdaPNM,AdaSmooth,AdaHessian,Adai,AdamP,AdamS,Adan,AggMo,AliG,Amos,Apollo,AvaGrad,CAME,DAdaptAdaGrad,DAdaptAdam,DAdaptAdan,DAdaptSGD,DAdaptLion,DiffGrad,Fromage,Gravity,GSAM,LARS,Lamb,Lion,LOMO,Lookahead,MADGRAD,MSVAG,Nero,NovoGrad,PAdam,PCGrad,PID,PNM,Prodigy,QHAdam,QHM,RAdam,Ranger,Ranger21,RotoGrad,SAM,SGDP,Shampoo,ScalableShampoo,SGDW,SignSGD,SM3,SopihaH,SRMM,SWATS,Tiger,Yogi,BCE,BCEFocal,Focal,FocalCosine,SoftF1,Dice,LDAM,Jaccard,Bi-Tempered,Tversky,FocalTversky,LovaszHinge
8
8
  Author: kozistr
9
9
  Author-email: kozistr@gmail.com
10
10
  Maintainer: kozistr
@@ -54,7 +54,7 @@ pytorch-optimizer
54
54
 
55
55
  | **pytorch-optimizer** is optimizer & lr scheduler collections in PyTorch.
56
56
  | I just re-implemented (speed & memory tweaks, plug-ins) the algorithm while based on the original paper. Also, It includes useful and practical optimization ideas.
57
- | Currently, 57 optimizers, 6 lr schedulers, and 10 loss functions are supported!
57
+ | Currently, **60 optimizers**, **10 lr schedulers**, and **13 loss functions** are supported!
58
58
  |
59
59
  | Highly inspired by `pytorch-optimizer <https://github.com/jettify/pytorch-optimizer>`__.
60
60
 
@@ -69,20 +69,20 @@ So, please double-check the license before using it at your work.
69
69
  Installation
70
70
  ~~~~~~~~~~~~
71
71
 
72
- ::
72
+ .. code-block:: bash
73
73
 
74
74
  $ pip3 install -U pytorch-optimizer
75
75
 
76
76
  If there's a version issue when installing the package, try with `--no-deps` option.
77
77
 
78
- ::
78
+ .. code-block:: bash
79
79
 
80
80
  $ pip3 install -U --no-deps pytorch-optimizer
81
81
 
82
82
  Simple Usage
83
83
  ~~~~~~~~~~~~
84
84
 
85
- ::
85
+ .. code-block:: python
86
86
 
87
87
  from pytorch_optimizer import AdamP
88
88
 
@@ -99,7 +99,7 @@ Simple Usage
99
99
 
100
100
  Also, you can load the optimizer via `torch.hub`
101
101
 
102
- ::
102
+ .. code-block:: python
103
103
 
104
104
  import torch
105
105
 
@@ -109,7 +109,7 @@ Also, you can load the optimizer via `torch.hub`
109
109
 
110
110
  If you want to build the optimizer with parameters & configs, there's `create_optimizer()` API.
111
111
 
112
- ::
112
+ .. code-block:: python
113
113
 
114
114
  from pytorch_optimizer import create_optimizer
115
115
 
@@ -127,144 +127,148 @@ Supported Optimizers
127
127
 
128
128
  You can check the supported optimizers with below code.
129
129
 
130
- ::
130
+ .. code-block:: python
131
131
 
132
132
  from pytorch_optimizer import get_supported_optimizers
133
133
 
134
134
  supported_optimizers = get_supported_optimizers()
135
135
 
136
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
137
- | Optimizer | Description | Official Code | Paper | Citation |
138
- +==============+===================================================================================================+===================================================================================+===============================================================================================+======================================================================================================================+
139
- | AdaBelief | *Adapting Step-sizes by the Belief in Observed Gradients* | `github <https://github.com/juntang-zhuang/Adabelief-Optimizer>`__ | `https://arxiv.org/abs/2010.07468 <https://arxiv.org/abs/2010.07468>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv201007468Z/exportcitation>`__ |
140
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
141
- | AdaBound | *Adaptive Gradient Methods with Dynamic Bound of Learning Rate* | `github <https://github.com/Luolc/AdaBound/blob/master/adabound/adabound.py>`__ | `https://openreview.net/forum?id=Bkg3g2R9FX <https://openreview.net/forum?id=Bkg3g2R9FX>`__ | `cite <https://github.com/Luolc/AdaBound#citing>`__ |
142
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
143
- | AdaHessian | *An Adaptive Second Order Optimizer for Machine Learning* | `github <https://github.com/amirgholami/adahessian>`__ | `https://arxiv.org/abs/2006.00719 <https://arxiv.org/abs/2006.00719>`__ | `cite <https://github.com/amirgholami/adahessian#citation>`__ |
144
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
145
- | AdamD | *Improved bias-correction in Adam* | | `https://arxiv.org/abs/2110.10828 <https://arxiv.org/abs/2110.10828>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv211010828S/exportcitation>`__ |
146
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
147
- | AdamP | *Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights* | `github <https://github.com/clovaai/AdamP>`__ | `https://arxiv.org/abs/2006.08217 <https://arxiv.org/abs/2006.08217>`__ | `cite <https://github.com/clovaai/AdamP#how-to-cite>`__ |
148
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
149
- | diffGrad | *An Optimization Method for Convolutional Neural Networks* | `github <https://github.com/shivram1987/diffGrad>`__ | `https://arxiv.org/abs/1909.11015v3 <https://arxiv.org/abs/1909.11015v3>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190911015D/exportcitation>`__ |
150
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
151
- | MADGRAD | *A Momentumized, Adaptive, Dual Averaged Gradient Method for Stochastic* | `github <https://github.com/facebookresearch/madgrad>`__ | `https://arxiv.org/abs/2101.11075 <https://arxiv.org/abs/2101.11075>`__ | `cite <https://github.com/facebookresearch/madgrad#tech-report>`__ |
152
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
153
- | RAdam | *On the Variance of the Adaptive Learning Rate and Beyond* | `github <https://github.com/LiyuanLucasLiu/RAdam>`__ | `https://arxiv.org/abs/1908.03265 <https://arxiv.org/abs/1908.03265>`__ | `cite <https://github.com/LiyuanLucasLiu/RAdam#citation>`__ |
154
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
155
- | Ranger | *a synergistic optimizer combining RAdam and LookAhead, and now GC in one optimizer* | `github <https://github.com/lessw2020/Ranger-Deep-Learning-Optimizer>`__ | `https://bit.ly/3zyspC3 <https://bit.ly/3zyspC3>`__ | `cite <https://github.com/lessw2020/Ranger-Deep-Learning-Optimizer#citing-this-work>`__ |
156
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
157
- | Ranger21 | *a synergistic deep learning optimizer* | `github <https://github.com/lessw2020/Ranger21>`__ | `https://arxiv.org/abs/2106.13731 <https://arxiv.org/abs/2106.13731>`__ | `cite <https://github.com/lessw2020/Ranger21#referencing-this-work>`__ |
158
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
159
- | Lamb | *Large Batch Optimization for Deep Learning* | `github <https://github.com/cybertronai/pytorch-lamb>`__ | `https://arxiv.org/abs/1904.00962 <https://arxiv.org/abs/1904.00962>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190400962Y/exportcitation>`__ |
160
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
161
- | Shampoo | *Preconditioned Stochastic Tensor Optimization* | `github <https://github.com/moskomule/shampoo.pytorch>`__ | `https://arxiv.org/abs/1802.09568 <https://arxiv.org/abs/1802.09568>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180209568G/exportcitation>`__ |
162
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
163
- | Nero | *Learning by Turning: Neural Architecture Aware Optimisation* | `github <https://github.com/jxbz/nero>`__ | `https://arxiv.org/abs/2102.07227 <https://arxiv.org/abs/2102.07227>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210207227L/exportcitation>`__ |
164
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
165
- | Adan | *Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models* | `github <https://github.com/sail-sg/Adan>`__ | `https://arxiv.org/abs/2208.06677 <https://arxiv.org/abs/2208.06677>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv220806677X/exportcitation>`__ |
166
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
167
- | Adai | *Disentangling the Effects of Adaptive Learning Rate and Momentum* | `github <https://github.com/zeke-xie/adaptive-inertia-adai>`__ | `https://arxiv.org/abs/2006.15815 <https://arxiv.org/abs/2006.15815>`__ | `cite <https://github.com/zeke-xie/adaptive-inertia-adai#citing>`__ |
168
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
169
- | SAM | *Sharpness-Aware Minimization* | `github <https://github.com/davda54/sam>`__ | `https://arxiv.org/abs/2010.01412 <https://arxiv.org/abs/2010.01412>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv201001412F/exportcitation>`__ |
170
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
171
- | ASAM | *Adaptive Sharpness-Aware Minimization* | `github <https://github.com/davda54/sam>`__ | `https://arxiv.org/abs/2102.11600 <https://arxiv.org/abs/2102.11600>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210211600K/exportcitation>`__ |
172
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
173
- | GSAM | *Surrogate Gap Guided Sharpness-Aware Minimization* | `github <https://github.com/juntang-zhuang/GSAM>`__ | `https://openreview.net/pdf?id=edONMAnhLu- <https://openreview.net/pdf?id=edONMAnhLu->`__ | `cite <https://github.com/juntang-zhuang/GSAM#citation>`__ |
174
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
175
- | D-Adaptation | *Learning-Rate-Free Learning by D-Adaptation* | `github <https://github.com/facebookresearch/dadaptation>`__ | `https://arxiv.org/abs/2301.07733 <https://arxiv.org/abs/2301.07733>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2023arXiv230107733D/exportcitation>`__ |
176
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
177
- | AdaFactor | *Adaptive Learning Rates with Sublinear Memory Cost* | `github <https://github.com/DeadAt0m/adafactor-pytorch>`__ | `https://arxiv.org/abs/1804.04235 <https://arxiv.org/abs/1804.04235>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180404235S/exportcitation>`__ |
178
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
179
- | Apollo | *An Adaptive Parameter-wise Diagonal Quasi-Newton Method for Nonconvex Stochastic Optimization* | `github <https://github.com/XuezheMax/apollo>`__ | `https://arxiv.org/abs/2009.13586 <https://arxiv.org/abs/2009.13586>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv200913586M/exportcitation>`__ |
180
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
181
- | NovoGrad | *Stochastic Gradient Methods with Layer-wise Adaptive Moments for Training of Deep Networks* | `github <https://github.com/lonePatient/NovoGrad-pytorch>`__ | `https://arxiv.org/abs/1905.11286 <https://arxiv.org/abs/1905.11286>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190511286G/exportcitation>`__ |
182
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
183
- | Lion | *Symbolic Discovery of Optimization Algorithms* | `github <https://github.com/google/automl/tree/master/lion>`__ | `https://arxiv.org/abs/2302.06675 <https://arxiv.org/abs/2302.06675>`__ | `cite <https://github.com/google/automl/tree/master/lion#citation>`__ |
184
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
185
- | Ali-G | *Adaptive Learning Rates for Interpolation with Gradients* | `github <https://github.com/oval-group/ali-g>`__ | `https://arxiv.org/abs/1906.05661 <https://arxiv.org/abs/1906.05661>`__ | `cite <https://github.com/oval-group/ali-g#adaptive-learning-rates-for-interpolation-with-gradients>`__ |
186
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
187
- | SM3 | *Memory-Efficient Adaptive Optimization* | `github <https://github.com/google-research/google-research/tree/master/sm3>`__ | `https://arxiv.org/abs/1901.11150 <https://arxiv.org/abs/1901.11150>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190111150A/exportcitation>`__ |
188
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
189
- | AdaNorm | *Adaptive Gradient Norm Correction based Optimizer for CNNs* | `github <https://github.com/shivram1987/AdaNorm>`__ | `https://arxiv.org/abs/2210.06364 <https://arxiv.org/abs/2210.06364>`__ | `cite <https://github.com/shivram1987/AdaNorm/tree/main#citation>`__ |
190
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
191
- | RotoGrad | *Gradient Homogenization in Multitask Learning* | `github <https://github.com/adrianjav/rotograd>`__ | `https://openreview.net/pdf?id=T8wHz4rnuGL <https://openreview.net/pdf?id=T8wHz4rnuGL>`__ | `cite <https://github.com/adrianjav/rotograd#citing>`__ |
192
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
193
- | A2Grad | *Optimal Adaptive and Accelerated Stochastic Gradient Descent* | `github <https://github.com/severilov/A2Grad_optimizer>`__ | `https://arxiv.org/abs/1810.00553 <https://arxiv.org/abs/1810.00553>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv181000553D/exportcitation>`__ |
194
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
195
- | AccSGD | *Accelerating Stochastic Gradient Descent For Least Squares Regression* | `github <https://github.com/rahulkidambi/AccSGD>`__ | `https://arxiv.org/abs/1704.08227 <https://arxiv.org/abs/1704.08227>`__ | `cite <https://github.com/rahulkidambi/AccSGD#citation>`__ |
196
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
197
- | SGDW | *Decoupled Weight Decay Regularization* | `github <https://github.com/loshchil/AdamW-and-SGDW>`__ | `https://arxiv.org/abs/1711.05101 <https://arxiv.org/abs/1711.05101>`__ | `cite <https://github.com/loshchil/AdamW-and-SGDW#contact>`__ |
198
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
199
- | ASGD | *Adaptive Gradient Descent without Descent* | `github <https://github.com/ymalitsky/adaptive_GD>`__ | `https://arxiv.org/abs/1910.09529 <https://arxiv.org/abs/1910.09529>`__ | `cite <https://github.com/ymalitsky/adaptive_GD#reference>`__ |
200
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
201
- | Yogi | *Adaptive Methods for Nonconvex Optimization* | | `NIPS 2018 <https://papers.nips.cc/paper/8186-adaptive-methods-for-nonconvex-optimization>`__ | `cite <https://proceedings.neurips.cc/paper_files/paper/2018/hash/90365351ccc7437a1309dc64e4db32a3-Abstract.html>`__ |
202
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
203
- | SWATS | *Improving Generalization Performance by Switching from Adam to SGD* | | `https://arxiv.org/abs/1712.07628 <https://arxiv.org/abs/1712.07628>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2017arXiv171207628S/exportcitation>`__ |
204
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
205
- | Fromage | *On the distance between two neural networks and the stability of learning* | `github <https://github.com/jxbz/fromage>`__ | `https://arxiv.org/abs/2002.03432 <https://arxiv.org/abs/2002.03432>`__ | `cite <https://github.com/jxbz/fromage#citation>`__ |
206
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
207
- | MSVAG | *Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradients* | `github <https://github.com/lballes/msvag>`__ | `https://arxiv.org/abs/1705.07774 <https://arxiv.org/abs/1705.07774>`__ | `cite <https://github.com/lballes/msvag#citation>`__ |
208
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
209
- | AdaMod | *An Adaptive and Momental Bound Method for Stochastic Learning* | `github <https://github.com/lancopku/AdaMod>`__ | `https://arxiv.org/abs/1910.12249 <https://arxiv.org/abs/1910.12249>`__ | `cite <https://github.com/lancopku/AdaMod#citation>`__ |
210
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
211
- | AggMo | *Aggregated Momentum: Stability Through Passive Damping* | `github <https://github.com/AtheMathmo/AggMo>`__ | `https://arxiv.org/abs/1804.00325 <https://arxiv.org/abs/1804.00325>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180400325L/exportcitation>`__ |
212
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
213
- | QHAdam | *Quasi-hyperbolic momentum and Adam for deep learning* | `github <https://github.com/facebookresearch/qhoptim>`__ | `https://arxiv.org/abs/1810.06801 <https://arxiv.org/abs/1810.06801>`__ | `cite <https://github.com/facebookresearch/qhoptim#reference>`__ |
214
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
215
- | PID | *A PID Controller Approach for Stochastic Optimization of Deep Networks* | `github <https://github.com/tensorboy/PIDOptimizer>`__ | `CVPR 18 <http://www4.comp.polyu.edu.hk/~cslzhang/paper/CVPR18_PID.pdf>`__ | `cite <https://github.com/tensorboy/PIDOptimizer#citation>`__ |
216
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
217
- | Gravity | *a Kinematic Approach on Optimization in Deep Learning* | `github <https://github.com/dariush-bahrami/gravity.optimizer>`__ | `https://arxiv.org/abs/2101.09192 <https://arxiv.org/abs/2101.09192>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210109192B/exportcitation>`__ |
218
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
219
- | AdaSmooth | *An Adaptive Learning Rate Method based on Effective Ratio* | | `https://arxiv.org/abs/2204.00825v1 <https://arxiv.org/abs/2204.00825v1>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv220400825L/exportcitation>`__ |
220
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
221
- | SRMM | *Stochastic regularized majorization-minimization with weakly convex and multi-convex surrogates* | `github <https://github.com/HanbaekLyu/SRMM>`__ | `https://arxiv.org/abs/2201.01652 <https://arxiv.org/abs/2201.01652>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv220101652L/exportcitation>`__ |
222
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
223
- | AvaGrad | *Domain-independent Dominance of Adaptive Methods* | `github <https://github.com/lolemacs/avagrad>`__ | `https://arxiv.org/abs/1912.01823 <https://arxiv.org/abs/1912.01823>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv191201823S/exportcitation>`__ |
224
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
225
- | PCGrad | *Gradient Surgery for Multi-Task Learning* | `github <https://github.com/tianheyu927/PCGrad>`__ | `https://arxiv.org/abs/2001.06782 <https://arxiv.org/abs/2001.06782>`__ | `cite <https://github.com/tianheyu927/PCGrad#reference>`__ |
226
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
227
- | AMSGrad | *On the Convergence of Adam and Beyond* | | `https://openreview.net/pdf?id=ryQu7f-RZ <https://openreview.net/pdf?id=ryQu7f-RZ>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190409237R/exportcitation>`__ |
228
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
229
- | Lookahead | *k steps forward, 1 step back* | `github <https://github.com/pytorch/examples/tree/main/imagenet>`__ | `https://arxiv.org/abs/1907.08610 <https://arxiv.org/abs/1907.08610>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190708610Z/exportcitation>`__ |
230
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
231
- | PNM | *Manipulating Stochastic Gradient Noise to Improve Generalization* | `github <https://github.com/zeke-xie/Positive-Negative-Momentum>`__ | `https://arxiv.org/abs/2103.17182 <https://arxiv.org/abs/2103.17182>`__ | `cite <https://github.com/zeke-xie/Positive-Negative-Momentum#citing>`__ |
232
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
233
- | GC | *Gradient Centralization* | `github <https://github.com/Yonghongwei/Gradient-Centralization>`__ | `https://arxiv.org/abs/2004.01461 <https://arxiv.org/abs/2004.01461>`__ | `cite <https://github.com/Yonghongwei/Gradient-Centralization#citation>`__ |
234
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
235
- | AGC | *Adaptive Gradient Clipping* | `github <https://github.com/deepmind/deepmind-research/tree/master/nfnets>`__ | `https://arxiv.org/abs/2102.06171 <https://arxiv.org/abs/2102.06171>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210206171B/exportcitation>`__ |
236
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
237
- | Stable WD | *Understanding and Scheduling Weight Decay* | `github <https://github.com/zeke-xie/stable-weight-decay-regularization>`__ | `https://arxiv.org/abs/2011.11152 <https://arxiv.org/abs/2011.11152>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv201111152X/exportcitation>`__ |
238
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
239
- | Softplus T | *Calibrating the Adaptive Learning Rate to Improve Convergence of ADAM* | | `https://arxiv.org/abs/1908.00700 <https://arxiv.org/abs/1908.00700>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190800700T/exportcitation>`__ |
240
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
241
- | Un-tuned w/u | *On the adequacy of untuned warmup for adaptive optimization* | | `https://arxiv.org/abs/1910.04209 <https://arxiv.org/abs/1910.04209>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv191004209M/exportcitation>`__ |
242
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
243
- | Norm Loss | *An efficient yet effective regularization method for deep neural networks* | | `https://arxiv.org/abs/2103.06583 <https://arxiv.org/abs/2103.06583>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210306583G/exportcitation>`__ |
244
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
245
- | AdaShift | *Decorrelation and Convergence of Adaptive Learning Rate Methods* | `github <https://github.com/MichaelKonobeev/adashift>`__ | `https://arxiv.org/abs/1810.00143v4 <https://arxiv.org/abs/1810.00143v4>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv181000143Z/exportcitation>`__ |
246
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
247
- | AdaDelta | *An Adaptive Learning Rate Method* | | `https://arxiv.org/abs/1212.5701v1 <https://arxiv.org/abs/1212.5701v1>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2012arXiv1212.5701Z/exportcitation>`__ |
248
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
249
- | Amos | *An Adam-style Optimizer with Adaptive Weight Decay towards Model-Oriented Scale* | `github <https://github.com/google-research/jestimator>`__ | `https://arxiv.org/abs/2210.11693 <https://arxiv.org/abs/2210.11693>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv221011693T/exportcitation>`__ |
250
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
251
- | SignSGD | *Compressed Optimisation for Non-Convex Problems* | `github <https://github.com/jxbz/signSGD>`__ | `https://arxiv.org/abs/1802.04434 <https://arxiv.org/abs/1802.04434>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180204434B/exportcitation>`__ |
252
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
253
- | Sophia | *A Scalable Stochastic Second-order Optimizer for Language Model Pre-training* | `github <https://github.com/Liuhong99/Sophia>`__ | `https://arxiv.org/abs/2305.14342 <https://arxiv.org/abs/2305.14342>`__ | `cite <https://github.com/Liuhong99/Sophia>`__ |
254
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
255
- | Prodigy | *An Expeditiously Adaptive Parameter-Free Learner* | `github <https://github.com/konstmish/prodigy>`__ | `https://arxiv.org/abs/2306.06101 <https://arxiv.org/abs/2306.06101>`__ | `cite <https://github.com/konstmish/prodigy#how-to-cite>`__ |
256
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
257
- | PAdam | *Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep Neural Networks* | `github <https://github.com/uclaml/Padam>`__ | `https://arxiv.org/abs/1806.06763 <https://arxiv.org/abs/1806.06763>`__ | `cite <https://github.com/uclaml/Padam#citation>`__ |
258
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
259
- | LOMO | *Full Parameter Fine-tuning for Large Language Models with Limited Resources* | `github <https://github.com/OpenLMLab/LOMO>`__ | `https://arxiv.org/abs/2306.09782 <https://arxiv.org/abs/2306.09782>`__ | `cite <https://github.com/OpenLMLab/LOMO#citation>`__ |
260
- +--------------+---------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
136
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
137
+ | Optimizer | Description | Official Code | Paper | Citation |
138
+ +==============+===================================================================================================+==========================================================================================+===============================================================================================+======================================================================================================================+
139
+ | AdaBelief | *Adapting Step-sizes by the Belief in Observed Gradients* | `github <https://github.com/juntang-zhuang/Adabelief-Optimizer>`__ | `https://arxiv.org/abs/2010.07468 <https://arxiv.org/abs/2010.07468>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv201007468Z/exportcitation>`__ |
140
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
141
+ | AdaBound | *Adaptive Gradient Methods with Dynamic Bound of Learning Rate* | `github <https://github.com/Luolc/AdaBound/blob/master/adabound/adabound.py>`__ | `https://openreview.net/forum?id=Bkg3g2R9FX <https://openreview.net/forum?id=Bkg3g2R9FX>`__ | `cite <https://github.com/Luolc/AdaBound#citing>`__ |
142
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
143
+ | AdaHessian | *An Adaptive Second Order Optimizer for Machine Learning* | `github <https://github.com/amirgholami/adahessian>`__ | `https://arxiv.org/abs/2006.00719 <https://arxiv.org/abs/2006.00719>`__ | `cite <https://github.com/amirgholami/adahessian#citation>`__ |
144
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
145
+ | AdamD | *Improved bias-correction in Adam* | | `https://arxiv.org/abs/2110.10828 <https://arxiv.org/abs/2110.10828>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv211010828S/exportcitation>`__ |
146
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
147
+ | AdamP | *Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights* | `github <https://github.com/clovaai/AdamP>`__ | `https://arxiv.org/abs/2006.08217 <https://arxiv.org/abs/2006.08217>`__ | `cite <https://github.com/clovaai/AdamP#how-to-cite>`__ |
148
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
149
+ | diffGrad | *An Optimization Method for Convolutional Neural Networks* | `github <https://github.com/shivram1987/diffGrad>`__ | `https://arxiv.org/abs/1909.11015v3 <https://arxiv.org/abs/1909.11015v3>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190911015D/exportcitation>`__ |
150
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
151
+ | MADGRAD | *A Momentumized, Adaptive, Dual Averaged Gradient Method for Stochastic* | `github <https://github.com/facebookresearch/madgrad>`__ | `https://arxiv.org/abs/2101.11075 <https://arxiv.org/abs/2101.11075>`__ | `cite <https://github.com/facebookresearch/madgrad#tech-report>`__ |
152
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
153
+ | RAdam | *On the Variance of the Adaptive Learning Rate and Beyond* | `github <https://github.com/LiyuanLucasLiu/RAdam>`__ | `https://arxiv.org/abs/1908.03265 <https://arxiv.org/abs/1908.03265>`__ | `cite <https://github.com/LiyuanLucasLiu/RAdam#citation>`__ |
154
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
155
+ | Ranger | *a synergistic optimizer combining RAdam and LookAhead, and now GC in one optimizer* | `github <https://github.com/lessw2020/Ranger-Deep-Learning-Optimizer>`__ | `https://bit.ly/3zyspC3 <https://bit.ly/3zyspC3>`__ | `cite <https://github.com/lessw2020/Ranger-Deep-Learning-Optimizer#citing-this-work>`__ |
156
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
157
+ | Ranger21 | *a synergistic deep learning optimizer* | `github <https://github.com/lessw2020/Ranger21>`__ | `https://arxiv.org/abs/2106.13731 <https://arxiv.org/abs/2106.13731>`__ | `cite <https://github.com/lessw2020/Ranger21#referencing-this-work>`__ |
158
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
159
+ | Lamb | *Large Batch Optimization for Deep Learning* | `github <https://github.com/cybertronai/pytorch-lamb>`__ | `https://arxiv.org/abs/1904.00962 <https://arxiv.org/abs/1904.00962>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190400962Y/exportcitation>`__ |
160
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
161
+ | Shampoo | *Preconditioned Stochastic Tensor Optimization* | `github <https://github.com/moskomule/shampoo.pytorch>`__ | `https://arxiv.org/abs/1802.09568 <https://arxiv.org/abs/1802.09568>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180209568G/exportcitation>`__ |
162
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
163
+ | Nero | *Learning by Turning: Neural Architecture Aware Optimisation* | `github <https://github.com/jxbz/nero>`__ | `https://arxiv.org/abs/2102.07227 <https://arxiv.org/abs/2102.07227>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210207227L/exportcitation>`__ |
164
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
165
+ | Adan | *Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models* | `github <https://github.com/sail-sg/Adan>`__ | `https://arxiv.org/abs/2208.06677 <https://arxiv.org/abs/2208.06677>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv220806677X/exportcitation>`__ |
166
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
167
+ | Adai | *Disentangling the Effects of Adaptive Learning Rate and Momentum* | `github <https://github.com/zeke-xie/adaptive-inertia-adai>`__ | `https://arxiv.org/abs/2006.15815 <https://arxiv.org/abs/2006.15815>`__ | `cite <https://github.com/zeke-xie/adaptive-inertia-adai#citing>`__ |
168
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
169
+ | SAM | *Sharpness-Aware Minimization* | `github <https://github.com/davda54/sam>`__ | `https://arxiv.org/abs/2010.01412 <https://arxiv.org/abs/2010.01412>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv201001412F/exportcitation>`__ |
170
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
171
+ | ASAM | *Adaptive Sharpness-Aware Minimization* | `github <https://github.com/davda54/sam>`__ | `https://arxiv.org/abs/2102.11600 <https://arxiv.org/abs/2102.11600>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210211600K/exportcitation>`__ |
172
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
173
+ | GSAM | *Surrogate Gap Guided Sharpness-Aware Minimization* | `github <https://github.com/juntang-zhuang/GSAM>`__ | `https://openreview.net/pdf?id=edONMAnhLu- <https://openreview.net/pdf?id=edONMAnhLu->`__ | `cite <https://github.com/juntang-zhuang/GSAM#citation>`__ |
174
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
175
+ | D-Adaptation | *Learning-Rate-Free Learning by D-Adaptation* | `github <https://github.com/facebookresearch/dadaptation>`__ | `https://arxiv.org/abs/2301.07733 <https://arxiv.org/abs/2301.07733>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2023arXiv230107733D/exportcitation>`__ |
176
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
177
+ | AdaFactor | *Adaptive Learning Rates with Sublinear Memory Cost* | `github <https://github.com/DeadAt0m/adafactor-pytorch>`__ | `https://arxiv.org/abs/1804.04235 <https://arxiv.org/abs/1804.04235>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180404235S/exportcitation>`__ |
178
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
179
+ | Apollo | *An Adaptive Parameter-wise Diagonal Quasi-Newton Method for Nonconvex Stochastic Optimization* | `github <https://github.com/XuezheMax/apollo>`__ | `https://arxiv.org/abs/2009.13586 <https://arxiv.org/abs/2009.13586>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv200913586M/exportcitation>`__ |
180
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
181
+ | NovoGrad | *Stochastic Gradient Methods with Layer-wise Adaptive Moments for Training of Deep Networks* | `github <https://github.com/lonePatient/NovoGrad-pytorch>`__ | `https://arxiv.org/abs/1905.11286 <https://arxiv.org/abs/1905.11286>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190511286G/exportcitation>`__ |
182
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
183
+ | Lion | *Symbolic Discovery of Optimization Algorithms* | `github <https://github.com/google/automl/tree/master/lion>`__ | `https://arxiv.org/abs/2302.06675 <https://arxiv.org/abs/2302.06675>`__ | `cite <https://github.com/google/automl/tree/master/lion#citation>`__ |
184
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
185
+ | Ali-G | *Adaptive Learning Rates for Interpolation with Gradients* | `github <https://github.com/oval-group/ali-g>`__ | `https://arxiv.org/abs/1906.05661 <https://arxiv.org/abs/1906.05661>`__ | `cite <https://github.com/oval-group/ali-g#adaptive-learning-rates-for-interpolation-with-gradients>`__ |
186
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
187
+ | SM3 | *Memory-Efficient Adaptive Optimization* | `github <https://github.com/google-research/google-research/tree/master/sm3>`__ | `https://arxiv.org/abs/1901.11150 <https://arxiv.org/abs/1901.11150>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190111150A/exportcitation>`__ |
188
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
189
+ | AdaNorm | *Adaptive Gradient Norm Correction based Optimizer for CNNs* | `github <https://github.com/shivram1987/AdaNorm>`__ | `https://arxiv.org/abs/2210.06364 <https://arxiv.org/abs/2210.06364>`__ | `cite <https://github.com/shivram1987/AdaNorm/tree/main#citation>`__ |
190
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
191
+ | RotoGrad | *Gradient Homogenization in Multitask Learning* | `github <https://github.com/adrianjav/rotograd>`__ | `https://openreview.net/pdf?id=T8wHz4rnuGL <https://openreview.net/pdf?id=T8wHz4rnuGL>`__ | `cite <https://github.com/adrianjav/rotograd#citing>`__ |
192
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
193
+ | A2Grad | *Optimal Adaptive and Accelerated Stochastic Gradient Descent* | `github <https://github.com/severilov/A2Grad_optimizer>`__ | `https://arxiv.org/abs/1810.00553 <https://arxiv.org/abs/1810.00553>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv181000553D/exportcitation>`__ |
194
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
195
+ | AccSGD | *Accelerating Stochastic Gradient Descent For Least Squares Regression* | `github <https://github.com/rahulkidambi/AccSGD>`__ | `https://arxiv.org/abs/1704.08227 <https://arxiv.org/abs/1704.08227>`__ | `cite <https://github.com/rahulkidambi/AccSGD#citation>`__ |
196
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
197
+ | SGDW | *Decoupled Weight Decay Regularization* | `github <https://github.com/loshchil/AdamW-and-SGDW>`__ | `https://arxiv.org/abs/1711.05101 <https://arxiv.org/abs/1711.05101>`__ | `cite <https://github.com/loshchil/AdamW-and-SGDW#contact>`__ |
198
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
199
+ | ASGD | *Adaptive Gradient Descent without Descent* | `github <https://github.com/ymalitsky/adaptive_GD>`__ | `https://arxiv.org/abs/1910.09529 <https://arxiv.org/abs/1910.09529>`__ | `cite <https://github.com/ymalitsky/adaptive_GD#reference>`__ |
200
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
201
+ | Yogi | *Adaptive Methods for Nonconvex Optimization* | | `NIPS 2018 <https://papers.nips.cc/paper/8186-adaptive-methods-for-nonconvex-optimization>`__ | `cite <https://proceedings.neurips.cc/paper_files/paper/2018/hash/90365351ccc7437a1309dc64e4db32a3-Abstract.html>`__ |
202
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
203
+ | SWATS | *Improving Generalization Performance by Switching from Adam to SGD* | | `https://arxiv.org/abs/1712.07628 <https://arxiv.org/abs/1712.07628>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2017arXiv171207628S/exportcitation>`__ |
204
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
205
+ | Fromage | *On the distance between two neural networks and the stability of learning* | `github <https://github.com/jxbz/fromage>`__ | `https://arxiv.org/abs/2002.03432 <https://arxiv.org/abs/2002.03432>`__ | `cite <https://github.com/jxbz/fromage#citation>`__ |
206
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
207
+ | MSVAG | *Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradients* | `github <https://github.com/lballes/msvag>`__ | `https://arxiv.org/abs/1705.07774 <https://arxiv.org/abs/1705.07774>`__ | `cite <https://github.com/lballes/msvag#citation>`__ |
208
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
209
+ | AdaMod | *An Adaptive and Momental Bound Method for Stochastic Learning* | `github <https://github.com/lancopku/AdaMod>`__ | `https://arxiv.org/abs/1910.12249 <https://arxiv.org/abs/1910.12249>`__ | `cite <https://github.com/lancopku/AdaMod#citation>`__ |
210
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
211
+ | AggMo | *Aggregated Momentum: Stability Through Passive Damping* | `github <https://github.com/AtheMathmo/AggMo>`__ | `https://arxiv.org/abs/1804.00325 <https://arxiv.org/abs/1804.00325>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180400325L/exportcitation>`__ |
212
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
213
+ | QHAdam | *Quasi-hyperbolic momentum and Adam for deep learning* | `github <https://github.com/facebookresearch/qhoptim>`__ | `https://arxiv.org/abs/1810.06801 <https://arxiv.org/abs/1810.06801>`__ | `cite <https://github.com/facebookresearch/qhoptim#reference>`__ |
214
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
215
+ | PID | *A PID Controller Approach for Stochastic Optimization of Deep Networks* | `github <https://github.com/tensorboy/PIDOptimizer>`__ | `CVPR 18 <http://www4.comp.polyu.edu.hk/~cslzhang/paper/CVPR18_PID.pdf>`__ | `cite <https://github.com/tensorboy/PIDOptimizer#citation>`__ |
216
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
217
+ | Gravity | *a Kinematic Approach on Optimization in Deep Learning* | `github <https://github.com/dariush-bahrami/gravity.optimizer>`__ | `https://arxiv.org/abs/2101.09192 <https://arxiv.org/abs/2101.09192>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210109192B/exportcitation>`__ |
218
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
219
+ | AdaSmooth | *An Adaptive Learning Rate Method based on Effective Ratio* | | `https://arxiv.org/abs/2204.00825v1 <https://arxiv.org/abs/2204.00825v1>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv220400825L/exportcitation>`__ |
220
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
221
+ | SRMM | *Stochastic regularized majorization-minimization with weakly convex and multi-convex surrogates* | `github <https://github.com/HanbaekLyu/SRMM>`__ | `https://arxiv.org/abs/2201.01652 <https://arxiv.org/abs/2201.01652>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv220101652L/exportcitation>`__ |
222
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
223
+ | AvaGrad | *Domain-independent Dominance of Adaptive Methods* | `github <https://github.com/lolemacs/avagrad>`__ | `https://arxiv.org/abs/1912.01823 <https://arxiv.org/abs/1912.01823>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv191201823S/exportcitation>`__ |
224
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
225
+ | PCGrad | *Gradient Surgery for Multi-Task Learning* | `github <https://github.com/tianheyu927/PCGrad>`__ | `https://arxiv.org/abs/2001.06782 <https://arxiv.org/abs/2001.06782>`__ | `cite <https://github.com/tianheyu927/PCGrad#reference>`__ |
226
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
227
+ | AMSGrad | *On the Convergence of Adam and Beyond* | | `https://openreview.net/pdf?id=ryQu7f-RZ <https://openreview.net/pdf?id=ryQu7f-RZ>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190409237R/exportcitation>`__ |
228
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
229
+ | Lookahead | *k steps forward, 1 step back* | `github <https://github.com/pytorch/examples/tree/main/imagenet>`__ | `https://arxiv.org/abs/1907.08610 <https://arxiv.org/abs/1907.08610>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190708610Z/exportcitation>`__ |
230
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
231
+ | PNM | *Manipulating Stochastic Gradient Noise to Improve Generalization* | `github <https://github.com/zeke-xie/Positive-Negative-Momentum>`__ | `https://arxiv.org/abs/2103.17182 <https://arxiv.org/abs/2103.17182>`__ | `cite <https://github.com/zeke-xie/Positive-Negative-Momentum#citing>`__ |
232
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
233
+ | GC | *Gradient Centralization* | `github <https://github.com/Yonghongwei/Gradient-Centralization>`__ | `https://arxiv.org/abs/2004.01461 <https://arxiv.org/abs/2004.01461>`__ | `cite <https://github.com/Yonghongwei/Gradient-Centralization#citation>`__ |
234
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
235
+ | AGC | *Adaptive Gradient Clipping* | `github <https://github.com/deepmind/deepmind-research/tree/master/nfnets>`__ | `https://arxiv.org/abs/2102.06171 <https://arxiv.org/abs/2102.06171>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210206171B/exportcitation>`__ |
236
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
237
+ | Stable WD | *Understanding and Scheduling Weight Decay* | `github <https://github.com/zeke-xie/stable-weight-decay-regularization>`__ | `https://arxiv.org/abs/2011.11152 <https://arxiv.org/abs/2011.11152>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv201111152X/exportcitation>`__ |
238
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
239
+ | Softplus T | *Calibrating the Adaptive Learning Rate to Improve Convergence of ADAM* | | `https://arxiv.org/abs/1908.00700 <https://arxiv.org/abs/1908.00700>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190800700T/exportcitation>`__ |
240
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
241
+ | Un-tuned w/u | *On the adequacy of untuned warmup for adaptive optimization* | | `https://arxiv.org/abs/1910.04209 <https://arxiv.org/abs/1910.04209>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv191004209M/exportcitation>`__ |
242
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
243
+ | Norm Loss | *An efficient yet effective regularization method for deep neural networks* | | `https://arxiv.org/abs/2103.06583 <https://arxiv.org/abs/2103.06583>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210306583G/exportcitation>`__ |
244
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
245
+ | AdaShift | *Decorrelation and Convergence of Adaptive Learning Rate Methods* | `github <https://github.com/MichaelKonobeev/adashift>`__ | `https://arxiv.org/abs/1810.00143v4 <https://arxiv.org/abs/1810.00143v4>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv181000143Z/exportcitation>`__ |
246
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
247
+ | AdaDelta | *An Adaptive Learning Rate Method* | | `https://arxiv.org/abs/1212.5701v1 <https://arxiv.org/abs/1212.5701v1>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2012arXiv1212.5701Z/exportcitation>`__ |
248
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
249
+ | Amos | *An Adam-style Optimizer with Adaptive Weight Decay towards Model-Oriented Scale* | `github <https://github.com/google-research/jestimator>`__ | `https://arxiv.org/abs/2210.11693 <https://arxiv.org/abs/2210.11693>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv221011693T/exportcitation>`__ |
250
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
251
+ | SignSGD | *Compressed Optimisation for Non-Convex Problems* | `github <https://github.com/jxbz/signSGD>`__ | `https://arxiv.org/abs/1802.04434 <https://arxiv.org/abs/1802.04434>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180204434B/exportcitation>`__ |
252
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
253
+ | Sophia | *A Scalable Stochastic Second-order Optimizer for Language Model Pre-training* | `github <https://github.com/Liuhong99/Sophia>`__ | `https://arxiv.org/abs/2305.14342 <https://arxiv.org/abs/2305.14342>`__ | `cite <https://github.com/Liuhong99/Sophia>`__ |
254
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
255
+ | Prodigy | *An Expeditiously Adaptive Parameter-Free Learner* | `github <https://github.com/konstmish/prodigy>`__ | `https://arxiv.org/abs/2306.06101 <https://arxiv.org/abs/2306.06101>`__ | `cite <https://github.com/konstmish/prodigy#how-to-cite>`__ |
256
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
257
+ | PAdam | *Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep Neural Networks* | `github <https://github.com/uclaml/Padam>`__ | `https://arxiv.org/abs/1806.06763 <https://arxiv.org/abs/1806.06763>`__ | `cite <https://github.com/uclaml/Padam#citation>`__ |
258
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
259
+ | LOMO | *Full Parameter Fine-tuning for Large Language Models with Limited Resources* | `github <https://github.com/OpenLMLab/LOMO>`__ | `https://arxiv.org/abs/2306.09782 <https://arxiv.org/abs/2306.09782>`__ | `cite <https://github.com/OpenLMLab/LOMO#citation>`__ |
260
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
261
+ | Tiger | *A Tight-fisted Optimizer, an optimizer that is extremely budget-conscious* | `github <https://github.com/bojone/tiger>`__ | | `cite <https://github.com/bojone/tiger/blob/main/README_en.md#citation>`__ |
262
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
263
+ | CAME | *Confidence-guided Adaptive Memory Efficient Optimization* | `github <https://github.com/huawei-noah/Pretrained-Language-Model/tree/master/CAME>`__ | `https://aclanthology.org/2023.acl-long.243/ <https://aclanthology.org/2023.acl-long.243/>`__ | `cite <https://github.com/huawei-noah/Pretrained-Language-Model/tree/master/CAME#citation>`__ |
264
+ +--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
261
265
 
262
266
  Supported LR Scheduler
263
267
  ----------------------
264
268
 
265
269
  You can check the supported learning rate schedulers with below code.
266
270
 
267
- ::
271
+ .. code-block:: python
268
272
 
269
273
  from pytorch_optimizer import get_supported_lr_schedulers
270
274
 
@@ -283,7 +287,7 @@ Supported Loss Function
283
287
 
284
288
  You can check the supported loss functions with below code.
285
289
 
286
- ::
290
+ .. code-block:: python
287
291
 
288
292
  from pytorch_optimizer import get_supported_loss_functions
289
293
 
@@ -304,6 +308,10 @@ You can check the supported loss functions with below code.
304
308
  +---------------------+-------------------------------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
305
309
  | Bi-Tempered | *The Principle of Unchanged Optimality in Reinforcement Learning Generalization* | | `https://arxiv.org/abs/1906.03361 <https://arxiv.org/abs/1906.03361>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190600336I/exportcitation>`__ |
306
310
  +---------------------+-------------------------------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
311
+ | Tversky | *Tversky loss function for image segmentation using 3D fully convolutional deep networks* | | `https://arxiv.org/abs/1706.05721 <https://arxiv.org/abs/1706.05721>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2017arXiv170605721S/exportcitation>`__ |
312
+ +---------------------+-------------------------------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
313
+ | Lovasz Hinge | *A tractable surrogate for the optimization of the intersection-over-union measure in neural networks* | `github <https://github.com/bermanmaxim/LovaszSoftmax>`__ | `https://arxiv.org/abs/1705.08790 <https://arxiv.org/abs/1705.08790>`__ | `cite <https://github.com/bermanmaxim/LovaszSoftmax#citation>`__ |
314
+ +---------------------+-------------------------------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
307
315
 
308
316
  Useful Resources
309
317
  ----------------
@@ -434,21 +442,22 @@ Improved bias-correction in Adam
434
442
  Adaptive Gradient Norm Correction
435
443
  ---------------------------------
436
444
 
437
- | Correcting the norm of gradient in each iteration based on the adaptive training history of gradient norm.
445
+ | Correcting the norm of a gradient in each iteration based on the adaptive training history of gradient norm.
438
446
 
439
447
  Citation
440
448
  --------
441
449
 
442
- Please cite original authors of optimization algorithms. You can easily find it from the above table!
443
- If you use this software, please cite it as below. Or you can get from "cite this repository" button.
450
+ Please cite the original authors of optimization algorithms. You can easily find it in the above table!
451
+ If you use this software, please cite it below. Or you can get it from "cite this repository" button.
444
452
 
445
453
  ::
446
454
 
447
- @software{Kim_pytorch_optimizer_Optimizer_and_2022,
455
+ @software{Kim_pytorch_optimizer_optimizer_2022,
448
456
  author = {Kim, Hyeongchan},
449
- month = {1},
450
- title = {{pytorch_optimizer: optimizer and lr scheduler collections in PyTorch}},
451
- version = {1.0.0},
457
+ month = jan,
458
+ title = {{pytorch_optimizer: optimizer & lr scheduler & loss function collections in PyTorch}},
459
+ url = {https://github.com/kozistr/pytorch_optimizer},
460
+ version = {2.11.0},
452
461
  year = {2022}
453
462
  }
454
463
 
@@ -462,9 +471,9 @@ Hyeongchan Kim / `@kozistr <http://kozistr.tech/about>`__
462
471
  :target: https://pytorch-optimizers.readthedocs.io/en/latest/?badge=latest
463
472
  .. |PyPI version| image:: https://badge.fury.io/py/pytorch-optimizer.svg
464
473
  :target: https://badge.fury.io/py/pytorch-optimizer
465
- .. |PyPi download| image:: https://pepy.tech/badge/pytorch-optimizer
474
+ .. |PyPi download| image:: https://static.pepy.tech/badge/pytorch-optimizer
466
475
  :target: https://pepy.tech/project/pytorch-optimizer
467
- .. |PyPi month download| image:: https://pepy.tech/badge/pytorch-optimizer/month
476
+ .. |PyPi month download| image:: https://static.pepy.tech/badge/pytorch-optimizer/month
468
477
  :target: https://pepy.tech/project/pytorch-optimizer
469
478
  .. |PyPI pyversions| image:: https://img.shields.io/pypi/pyversions/pytorch-optimizer.svg
470
479
  :target: https://pypi.python.org/pypi/pytorch-optimizer/