FlowAdam: Implicit Regularization via Geometry-Aware Soft Momentum Injection

Devender Singh, Tarun Sheel

Adaptive moment methods such as Adam use a diagonal, coordinate-wise preconditioner based on exponential moving averages of squared gradients. This diagonal scaling is coordinate-s…