Characterization of barycenters in the Wasserstein space by averaging optimal transport maps

Bigot, Jérémie; Klein, Thierry

ESAIM: Probability and Statistics, Tome 22 (2018), pp. 35-57.

This paper is concerned by the study of barycenters for random probability measures in the Wasserstein space. Using a duality argument, we give a precise characterization of the population barycenter for various parametric classes of random probability measures with compact support. In particular, we make a connection between averaging in the Wasserstein space as introduced in Agueh and Carlier [SIAM J. Math. Anal. 43 (2011) 904–924], and taking the expectation of optimal transport maps with respect to a fixed reference measure. We also discuss the usefulness of this approach in statistics for the analysis of deformable models in signal and image processing. In this setting, the problem of estimating a population barycenter from n independent and identically distributed random probability measures is also considered.

[1] B. Afsari,Riemannian l^p center of mass: existence, uniqueness, and convexity. Proc. Am. Math. Soc. 139 (2011) 655–673. | DOI | MR | Zbl

[2] M. Agueh and G. Carlier, Barycenters in the Wasserstein space. SIAM J. Math. Anal. 43 (2011) 904–924. | DOI | MR | Zbl

[3] A. Agulló-Antolín, J.A. Cuesta-Albertos, H. Lescornel and J.-M. Loubes, A parametric registration model for warped distributions with Wasserstein distance. J. Multivar. Anal. 135 (2015) 117–130. | DOI | MR | Zbl

[4] S. Allassonnière, Y. Amit and A. Trouvé, Towards a coherent statistical framework for dense deformable template estimation. J. R. Stat. Soc. Series B 69 (2007) 3–29. | DOI | MR | Zbl

[5] S. Allassonniére, J. Bigot, J. Glauns, F. Maire and F. Richard, Statistical models for deformable templates in image and shape analysis. Ann. Math. Blaise Pascal 20 (2013) 1–35. | DOI | Numdam | MR | Zbl

[6] P.C. Álvarez-Esteban, E. Del Barrio, J.A. Cuesta-Albertos and C. Matrán, Uniqueness and approximate computation of optimal incomplete transportation plans. Ann. Inst. Henri Poincaré Probab. Stat. 47 (2011) 358–375. | DOI | Numdam | MR | Zbl

[7] P.C. Álvarez-Esteban, E.Del Barrio, J.A. Cuesta-Albertos and C. Matrán, A fixed-point approach to barycenters in Wasserstein space. J. Math. Anal. Appl. 441 (2016) 744–762. | DOI | MR | Zbl

[8] M. Arnaudon, C. Dombry, A. Phan and L. Yang, Stochastic algorithms for computing means of probability measures. Stochastic Process. Appl. 122 (2012) 1437–1455. | DOI | MR | Zbl

[9] J.-D. Benamou, G. Carlier, M. Cuturi, L. Nenna and G. Peyré, Iterative bregman projections for regularized transportation problems. SIAM J. Sci. Comput. 37 (2015). | MR | Zbl

[10] R. Bhattacharya and V. Patrangenaru, Large sample theory of intrinsic and extrinsic sample means on manifolds (i). Ann. Stat. 31 (2003) 1–29. | DOI | MR | Zbl

[11] R. Bhattacharya and V. Patrangenaru, Large sample theory of intrinsic and extrinsic sample means on manifolds (ii). Ann. Stat. 33 (2005) 1225–1259. | DOI | MR | Zbl

[12] J. Bigot and B. Charlier. On the consistency of Fréchet means in deformable models for curve and image analysis. Electron. J. Stat. 5 (2011) 1054–1089. | DOI | MR | Zbl

[13] J. Bigot and S. Gadat. A deconvolution approach to estimation of a common shape in a shifted curves model. Ann. Stat. 38 (2010) 2422–2464. | DOI | MR | Zbl

[14] J. Bigot, S. Gadat and J.M. Loubes, Statistical M-estimation and consistency in large deformable models for image warping. J. Math. Imaging Vis. 34 (2009) 270–290. | DOI | MR | Zbl

[15] J. Bigot, J.M. Loubes and M. Vimond, Semiparametric estimation of shifts on compact Lie groups for image registration. Probab. Theory Relat. Fields 152 (2010) 425–473. | DOI | MR | Zbl

[16] S. Bobkov and M. Ledoux, One-dimensional empirical measures, order statistics and Kantorovich transport distances. Memoirs of the American Mathematical Society (2017). Available at https://perso.math.univ-toulouse.fr/ledoux/files/ 2016/12/MEMO.pdf. | MR | Zbl

[17] E. Boissard, T. Le Gouic and J.-M. Loubes, Distribution’s template estimate with Wasserstein metrics. Bernoulli 21 (2015) 740–759. | DOI | MR | Zbl

[18] N. Bonneel, J. Rabin, G. Peyré and H. Pfister, Sliced and radon Wasserstein barycenters of measures. J. Math. Imaging Vis. 51 (2015) 22–45. | DOI | MR | Zbl

[19] Y. Brenier, Polar factorization and monotone rearrangement of vector-valued functions. Comm. Pure Appl. Math. 44 (1991) 375–417. | DOI | MR | Zbl

[20] M. Cuturi and A. Doucet, Fast computation of Wasserstein barycenters, in Proc. of the 31st International Conference on Machine Learning (ICML-14), edited by I.T. Jebara and E.P. Xing. JMLR Workshop and Conference Proceedings (2014) 685–693.

[21] G. De Philippis and A. Figalli, The Monge-Ampère equation and its link to optimal transportation. Bull. Amer. Math. Soc. 51 (2014) 527–580. | DOI | MR | Zbl

[22] I. Ekeland and R. Témam, Convex analysis and variational problems. English edition. Translated from the French. Vol. 28 of Classics in Applied Mathematics. Society for Industrial and Applied Mathematics SIAM, Philadelphia, PA (1999). | MR | Zbl

[23] J. Fontbona, H. Guérin and S. Méléard, Measurability of optimal transportation and strong coupling of martingale measures. Electron. Commun. Probab. 15 (2010) 124–133. | DOI | MR | Zbl

[24] M. Fréchet, Les éléments aléatoires de nature quelconque dans un espace distancié. Ann. Inst. Henri Poincaré, Sect. B, Prob. Stat. 10 (1948) 235–310. | Numdam | MR | Zbl

[25] F. Gamboa, J.-M. Loubes and E. Maza, Semi-parametric estimation of shifts. Electron. J. Stat. 1 (2007) 616–640. | DOI | MR | Zbl

[26] C. Goodall, Procrustes methods in the statistical analysis of shape. J. R. Stat. Soc. Series B 53 (1991) 285–339. | MR | Zbl

[27] U. Grenander, General Pattern Theory – A Mathematical Study of Regular Structures. Clarendon Press, Oxford (1993). | MR | Zbl

[28] U. Grenander and M. Miller, Pattern Theory: From Representation to Inference. Oxford Univ. Press, Oxford (2007). | MR | Zbl

[29] S. Haker and A. Tannenbaum, On the Monge-Kantorovich problem and image warping. In Vol. 133 of Mathematical Methods in Computer Vision. IMA Vol. Math. Appl. Springer, New York (2003) 65–85. | MR | Zbl

[30] S. Haker, L. Zhu, A. Tannenbaum and S. Angenent, Optimal mass transport for registration and warping. Int. J. Comput. Vis. 60 (2004) 225–240. | DOI | Zbl

[31] S.F. Huckemann, Intrinsic inference on the mean geodesic of planar shapes and tree discrimination by leaf growth. Ann. Stat. 39 (2011) 1098–1124. | DOI | MR | Zbl

[32] D.G. Kendall, Shape manifolds, procrustean metrics, and complex projective spaces. Bull. London Math Soc. 16 (1984) 81–121. | DOI | MR | Zbl

[33] Y.H. Kim and B. Pass, Wasserstein barycenters over Riemannian manifolds. Adv. Math. 307 (2017) 640–683. | DOI | MR | Zbl

[34] T. Le Gouic and J.-M. Loubes, Existence and Consistency of Wasserstein Barycenters. Probab. Theory Relat. Fields 168 (2017) 901–917. | DOI | MR | Zbl

[35] A.D. Loffe and V.M. Tihomirov, Duality of convex functions and extremum problems. Uspehi Mat. Nauk 23 (1968) 51–116. | MR | Zbl

[36] A.R. Meenakshi and C. Rajian, On a product of positive semidefinite matrices. Linear Algebra Appl. 295 (1999) 3–6. | DOI | MR | Zbl

[37] B. Pass, Optimal transportation with infinitely many marginals. J. Funct. Anal. 264 (2013) 947–963. | DOI | MR | Zbl

[38] J. Rabin, G. Peyré, J. Delon and M. Bernot, Wassertein Barycenter and its Applications to Texture Mixing, Vol. 6667 of Lect. Notes Comput. Sci., Proc. SSVM’11. Springer (2011) 435–446. | DOI

[39] R. Ranga Rao, Relations between weak and uniform convergence of measures with applications. Ann. Math. Stat. 33 (1962) 659–680. | DOI | MR | Zbl

[40] K.-T. Sturm, Probability measures on metric spaces of nonpositive curvature. In Heat kernels and analysis on manifolds, graphs, and metric spaces (Paris, 2002), Vol. 338 of Contemporary Mathematics. Am. Math. Soc., Providence, RI (2003) 357–390. | DOI | MR | Zbl

[41] H. Sverdrup-Thygeson, Strong law of large numbers for measures of central tendency and dispersion of random variables in compact metric spaces. Ann. Stat. 9 (1981) 141–145. | DOI | MR | Zbl

[42] A. Trouvé and L. Younes, Local geometry of deformable templates. SIAM J. Math. Anal. 37 (2005) 17–59. | DOI | MR | Zbl

[43] A. Trouvé and L. Younes, Shape spaces, In Handbook of Mathematical Methods in Imaging. Springer (2011). | DOI | Zbl

[44] C. Villani, Topics in Optimal Transportation. American Mathematical Society (2003). | MR | Zbl

[45] M. Vimond, Efficient estimation for a subclass of shape invariant models. Ann. Stat. 38 (2010) 1885–1912. | DOI | MR | Zbl