Abstract
Generative models inspired by dynamical transport of measure -- such as flowsand diffusions -- construct a continuous-time map between two probabilitydensities. Conventionally, one of these is the target density, only accessiblethrough samples, while the other is taken as a simple base density that isdata-agnostic. In this work, using the framework of stochastic interpolants, weformalize how to \textit{couple} the base and the target densities. Thisenables us to incorporate information about class labels or continuousembeddings to construct dynamical transport maps that serve as conditionalgenerative models. We show that these transport maps can be learned by solvinga simple square loss regression problem analogous to the standard independentsetting. We demonstrate the usefulness of constructing dependent couplings inpractice through experiments in super-resolution and in-painting.