Skip to yearly menu bar Skip to main content


Poster

Keep the Momentum: Conservation Laws beyond Euclidean Gradient Flows

Sibylle Marcotte · Rémi Gribonval · Gabriel Peyré


Abstract:

Conservation laws are well-established in the context of Euclidean gradient flow dynamics, notably for linear or ReLU neural network training. Yet, their existence and principles for non-Euclidean geometries and momentum-based dynamics remain largely unknown. In this paper, we characterize "all" conservation laws pertinent in such contexts. In stark contrast to the case of gradient flows, we prove that the conservation laws for momentum-based dynamics exhibit temporal dependence. Additionally, we identify a consistent pattern of "conservation loss" when transitioning from gradient flow scenarios to momentum cases. Specifically, for linear networks, our framework allows us to identify all conservation laws, which are less numerous than in the gradient flow case. With ReLU networks, no conservation law remains. This phenomenon also manifests in non-Euclidean metrics, used e.g. for Nonnegative Matrix Factorization (NMF): all conservation laws can be determined in the gradient flow context, yet none persists in the momentum case.

Live content is unavailable. Log in and register to view live content