cs.LGMay 24, 2026

Leveraging Gauge Freedom for Learning Non-Gradient Population Dynamics of Stochastic Systems

Authors: Jules BermanTobias BlickhanBenjamin Peherstorfer

Abstract

Existing work on population dynamics inference often focuses on flows arising from vector fields that are the gradients of scalar potentials. Among all admissible flows that are compatible with the population dynamics, gradient flows are optimal in a specific sense: they minimize kinetic energy. The selection of fields based on different criteria corresponds to a gauge freedom when determining population dynamics, which we leverage in this work. We propose Non-Gradient Inference Flows (NGIF), an algorithm to infer non-gradient population dynamics using a weak formulation of the continuity equation. This allows us to parameterize general vector fields and choose other selection criteria beyond minimal kinetic energy. We demonstrate on a variety of low- and high-dimensional physics problems that this more general approach improves distributional accuracy over gradient-restricted baselines and better captures non-potential transport.

Explore similar work

CardsList
  1. Two-Parameter Flows for Learning Population Dynamics of Physical Systems

    May 25, 2026Paul Schwerdtner, Tobias Blickhan, Benjamin PeherstorferPopulation Dynamics