← Search

Pierre-Cyril Aubin-Frankowski

4 accepted papers

2024

Mirror and Preconditioned Gradient Descent in Wasserstein Space

NeurIPS 2024spotlight

As the problem of minimizing functionals on the Wasserstein space encompasses many applications in machine learning, different optimization algorithms on $\mathbb{R}^d$ have received their counterpart analog on the Wasserstein space. We focus here on lifting two explicit algorithms: mirror descent a…

2022

Mirror Descent with Relative Smoothness in Measure Spaces, with application to Sinkhorn and EM

NeurIPS 2022accept

Many problems in machine learning can be formulated as optimizing a convex functional over a vector space of measures. This paper studies the convergence of the mirror descent algorithm in this infinite-dimensional setting. Defining Bregman divergences through directional derivatives, we derive the…

Cited by 36SourcePDFScholar