1: \begin{abstract}

2: We study first order methods to compute the barycenter of a probability distribution $P$ over the space of probability measures with finite second moment. We develop a framework to derive global rates of convergence for both gradient descent and stochastic gradient descent despite the fact that the barycenter functional is not geodesically convex. Our analysis overcomes this technical hurdle by employing a Polyak-\L{}ojasiewicz (PL) inequality and relies on tools from optimal transport and metric geometry. In turn, we establish a PL inequality when $P$ is supported on the Bures-Wasserstein manifold of Gaussian probability measures. It leads to the first global rates of convergence for first order methods in this context. 

3: 

4: % We study first order methods to compute the barycenter of a probability distribution over Wasserstein

5: % space. We develop a framework to analyze the convergence of gradient descent and stochastic gradient descent for a general barycenter functional, based on a Polyak- PL inequality and an improved variance inequality. To achieve concrete rates of convergence these results to the

6: % Bures-Wasserstein manifold and

7: % derive global rates of convergence for both gradient descent and stochastic gradient descent despite the fact that the barycenter functional is not geodesically convex. 

8: 

9: %Our analysis overcomes this technical hurdle by developing a Polyak-\L{}ojasiewicz (PL) inequality, using tools from 

10: 

11: %\scnote{I changed ``non-Euclidean data'' to ``geometric data'' because I think it is more correct.}

12: \end{abstract}

13: