1: \begin{abstract}

2:     Modern distribution matching algorithms for training diffusion or flow models directly \emph{prescribe} the time evolution of the marginal distributions between two boundary distributions.

3:     In this work, we consider a generalized distribution matching setup, where these marginals are only \emph{implicitly} described as a solution to some task-specific objective function.

4:     The problem setup, known as the Generalized Schr{\"o}dinger Bridge (GSB),

5:     appears prevalently in many scientific areas both within and without machine learning.

6:     We propose \textbf{\GSBM{} (GSBM)}, a new matching algorithm inspired by recent advances,

7:     generalizing them beyond kinetic energy minimization and to account for task-specific state costs.

8:     We show that such a generalization can be cast as solving \emph{conditional} stochastic optimal control, for which efficient variational approximations can be used, and further debiased with the aid of path integral theory.

9:     Compared to prior methods for solving GSB problems, our GSBM algorithm always preserves a feasible transport map between the boundary distributions throughout training, thereby enabling stable convergence and significantly improved scalability.

10:     We empirically validate our claims on an extensive suite of experimental setups, including crowd navigation, opinion depolarization, LiDAR manifolds, and image domain transfer.

11:     Our work brings new algorithmic opportunities for training diffusion models enhanced with task-specific optimality structures.

12: \end{abstract}

13: