English

Covariant Gradient Descent

Machine Learning 2025-04-15 v2 High Energy Physics - Theory Optimization and Control

Abstract

We present a manifestly covariant formulation of the gradient descent method, ensuring consistency across arbitrary coordinate systems and general curved trainable spaces. The optimization dynamics is defined using a covariant force vector and a covariant metric tensor, both computed from the first and second statistical moments of the gradients. These moments are estimated through time-averaging with an exponential weight function, which preserves linear computational complexity. We show that commonly used optimization methods such as RMSProp, Adam and AdaBelief correspond to special limits of the covariant gradient descent (CGD) and demonstrate how these methods can be further generalized and improved.

Keywords

Cite

@article{arxiv.2504.05279,
  title  = {Covariant Gradient Descent},
  author = {Dmitry Guskov and Vitaly Vanchurin},
  journal= {arXiv preprint arXiv:2504.05279},
  year   = {2025}
}

Comments

12 pages, 2 figures, 2 tables

R2 v1 2026-06-28T22:49:44.029Z