arXiv Machine Learning

Nonlocal Mean Field Schr\"{o}dinger Bridge with Learned Interactions

arXiv:2606. 04265v1 Announce Type: cross Abstract: The Schr\"odinger Bridge Problem constructs a stochastic process that connects an initial distribution to a terminal distribution with minimum energy.

arXiv Machine Learning
1d ago

Cost-augmented Schr\"odinger bridges on graphs are exactly solvable: a Feynman-Kac tilt replaces learned control

The paper presents a new formulation of the Schr"odinger bridge problem on graphs that incorporates state costs via a Feynman‑Kac tilt, eliminating the need for learned control or temporal‑difference penalties. The resulting cost‑augmented bridge is solved exactly by alternating two endpoint rescalings, each requiring only a sparse matrix‑exponential application, and the method scales linearly with network size. Experiments on a protein‑folding model and a large road‑network demonstrate that the exact bridge reduces expected energy barriers and matches target distributions within sampling error.

By Akshay Balsubramani
arXiv Statistics ML
Aug 27

Schr\"odinger Bridges over Kinetic Swarming Models

The paper studies finite‑horizon minimum‑energy steering of inertial swarms under stochastic disturbances, focusing on mean‑field models with Cucker–Smale alignment or Morse attraction–repulsion interactions. It formulates the problem as a Schr"odinger bridge, deriving nonlinear, time‑symmetric optimality systems and proposing nested fixed‑point schemes for numerical solution. Numerical experiments demonstrate that the optimal corrective drift can either exploit or counteract the natural interaction forces, depending on their alignment with the steering objective.

By Asmaa Eldesoukey, Md Zulfiqur Haider, Italo Napolitano, Yongxin Chen, Abhishek Halder
arXiv Machine Learning
Aug 20

Self-supervised In-context Operator Learning for Stochastic Mean-Field Control

The paper introduces a mesh‑free, self‑supervised neural operator—called the Normalizing Flow Invertible Solution Transformer (NFIST)—for stochastic mean‑field control (MFC). By reformulating the controlled Fokker–Planck dynamics as a deterministic continuity equation via a probability‑flow ODE and an invertible normalizing‑flow transformer, the authors enable closed‑form score evaluation with linear cost per particle. The resulting operator learns from task prompts (distribution parameters or particle clouds) and can solve unseen MFC tasks in a single forward pass, achieving zero‑shot generalization across applications such as stochastic optimal control, Schrödinger bridges, systemic‑risk control, and obstacle‑avoiding path planning.

By Suyi Gao, Mo Zhou, Rongjie Lai
arXiv Machine Learning
Jun 16

Amortized mean-shift interacting particles

arXiv:2606. 15871v1 Announce Type: cross Abstract: Bayesian inference for inverse problems is run to evaluate integrals -- posterior expectations, tail probabilities, and risks -- across a stream of observations.

By Ali Siahkoohi
arXiv Machine Learning
Sep 4

Towards Scaling Reinforcement Learning to Massive Populations: Learning Mean-Field Representations

The paper proposes a mean‑field reinforcement learning framework that models rewards and transitions as functions of an unknown low‑dimensional aggregate statistic of a large agent population. By learning this low‑dimensional representation in an offline setting, the authors demonstrate a provable method for obtaining near‑optimal policies. Experiments on a one‑step routing game inspired by supply‑chain problems show that, with a fixed neural‑network size and optimization budget, the learned representation improves reward prediction and the quality of Nash equilibria compared to baselines that ignore population structure.

By Aditya Makkar, Benjamin Unger, Jeongyeol Kwon, Mathieu Lauri\`ere, Eugene Vinitsky, Yonathan Efroni
arXiv Machine Learning
Sep 23

DeepSPoC: A Deep Learning Based Sequential Propagation of Chaos

DeepSPoC is a neural particle method that replaces direct particle-particle interactions in sequential propagation of chaos (SPoC) with particle‑network interactions, using a neural density representation (KRnet) to approximate the empirical measure. By simulating particles in batches and embedding a neural network into the mean‑field SDE coefficients, DeepSPoC reduces memory usage and computational cost compared to traditional particle methods. The approach is demonstrated on various mean‑field equations, showing improved scalability for high‑dimensional problems.

By Kai Du, Yongle Xie, Tao Zhou, Yuancheng Zhou
arXiv AI
Jun 2

Equilibrium Propagation for Non-Conservative Systems

arXiv:2602. 03670v2 Announce Type: replace-cross Abstract: Equilibrium Propagation (EP) is a physics-inspired learning algorithm that uses stationary states of a dynamical system both for inference and learning.

By Antonino Emanuele Scurria, Dimitri Vanden Abeele, Bortolo Matteo Mognetti, Serge Massar