Abstract
Monotone gradient functions play a central role in solving the Mongeformulation of the optimal transport problem, which arises in modernapplications ranging from fluid dynamics to robot swarm control. When thetransport cost is the squared Euclidean distance, Brenier's theorem guaranteesthat the unique optimal map is the gradient of a convex function, namely amonotone gradient map, and it satisfies a Monge-Amp\`ere equation. In[arXiv:2301.10862] [arXiv:2404.07361], we proposed Monotone Gradient Networks(mGradNets), neural networks that directly parameterize the space of monotonegradient maps. In this work, we leverage mGradNets to directly learn theoptimal transport mapping by minimizing a training loss function defined usingthe Monge-Amp\`ere equation. We empirically show that the structural bias ofmGradNets facilitates the learning of optimal transport maps and employ ourmethod for a robot swarm control problem.