AdEMAMix optimizer improves model performance and convergence speed.
problem Suboptimal use of single EMA in momentum-based optimizers.
method Proposes AdEMAMix, a modified Adam optimizer with a mixture of two EMAs.
result Empirically shows gradients can remain relevant for tens of thousands of steps, leading to faster convergence and lower minima.