We study the estimation of optimal transport (OT) maps between an arbitrary source probability measure and a log-concave target probability measure. Our contributions are twofold. First, we propose a new evolution equation in the set of transport maps. It can be seen as the gradient flow of a lift of some user-chosen divergence (e.g., the KL divergence, or relative entropy) to the space of transport maps, constrained to the convex set of optimal transport maps. We prove the existence of long-time solutions to this flow as well as its convergence toward the OT map as time goes to infinity, under standard convexity conditions on the divergence. Second, we study the practical implementation of this constrained gradient flow. We propose two time-discrete computational schemes-one explicit, one implicit-, and we prove the convergence of the latter to the OT map as time goes to infinity. We then parameterize the OT maps with convexity-constrained neural networks and train them with these discretizations of the constrained gradient flow. We show that this is equivalent to performing a natural gradient descent of the lift of the chosen divergence in the neural networks’ parameter space. Empirically, our scheme outperforms the standard Euclidean gradient descent methods used to train convexity-constrained neural networks in terms of approximation results for the OT map and convergence stability, and it still yields better results than the same approach combined with the widely used adam optimizer.
@article{dumont2026learning,title={Learning Monge maps by lifting and constraining Wasserstein gradient flows},author={Dumont, Th{\'e}o and Lacombe, Th{\'e}o and Vialard, Fran{\c{c}}ois-Xavier},year={2026},journal={preprint},}
2024
FoCMBest poster
On the existence of Monge maps for the Gromov-Wasserstein problem
The Gromov–Wasserstein problem is a non-convex optimization problem over the polytope of transportation plans between two probability measures supported on two spaces, each equipped with a cost function evaluating similarities between points. Akin to the standard optimal transportation problem, it is natural to ask for conditions guaranteeing some structure on the optimizers, for instance if these are induced by a (Monge) map. We study this question in Euclidean spaces when the cost functions are either given by (i) inner products or (ii) squared distances, two standard choices in the literature. We establish the existence of an optimal map in case (i) and of an optimal 2-map (the union of the graphs of two maps) in case (ii), both under an absolute continuity condition on the source measure. Additionally, in case (ii) and in dimension one, we numerically design situations where optimizers of the Gromov–Wasserstein problem are 2-maps but are not maps. This suggests that our result cannot be improved in general for this cost. Still in dimension one, we additionally establish the optimality of monotone maps under some conditions on the measures, thereby giving insight on why such maps often appear to be optimal in numerical experiments.
@article{dumont2022existence,title={On the existence of Monge maps for the Gromov-Wasserstein problem},author={Dumont, Th{\'e}o and Lacombe, Th{\'e}o and Vialard, Fran{\c{c}}ois-Xavier},year={2024},journal={Foundations of Computational Mathematics},}
2023
CVPR
Modular memorability: tiered representations for video memorability prediction
Théo Dumont, Juan S. Hevia, and Camilo L. Fosco
In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023
@inproceedings{dumont2023modular,title={Modular memorability: tiered representations for video memorability prediction},author={Dumont, Th{\'e}o and Hevia, Juan S. and Fosco, Camilo L.},booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},pages={10751--10760},year={2023}}
2022
eLife
The rapid developmental rise of somatic inhibition disengages hippocampal dynamics from self-motion
Robin F. Dard, Erwan Leprince, Julien Denis, Shrisha Rao Balappa, and 11 more authors
@article{dard2022rapid,title={The rapid developmental rise of somatic inhibition disengages hippocampal dynamics from self-motion},author={Dard, Robin F. and Leprince, Erwan and Denis, Julien and Balappa, Shrisha Rao and Suchkov, Dmitrii and Boyce, Richard and Lopez, Catherine and Giorgi-Kurz, Marie and Szwagier, Tom and Dumont, Th{\'e}o and Rouault, Herv{\'e} and Minlebaev, Marat and Baude, Agn{\`e}s and Cossart, Rosa and Picardo, Michel A.},journal={eLife},volume={11},pages={e78116},year={2022},publisher={eLife Sciences Publications Limited},}
