We develop a notion of projections between sets of probability measures using the geometric properties of the 2-Wasserstein space. It is designed for general multivariate probability measures, is computationally efficient to implement, and provides a unique solution in regular settings. The idea is to work on regular tangent cones of the Wasserstein space using generalized geodesics. Its structure and computational properties make the method applicable in a variety of settings, from causal inference to the analysis of object data. An application to estimating causal effects yields a generalization of the notion of synthetic controls to multivariate data with individual-level heterogeneity, as well as a way to estimate optimal weights jointly over all time periods.
翻译:我们利用2-Wasserstein空间的几何特性对几组概率计量方法进行预测,设计用于一般的多变量概率计量,在计算上有效,在正常情况下提供独特的解决办法。设想是利用通用的大地测量学,对瓦塞尔斯坦空间的常规正切锥体进行工作。其结构和计算特性使这种方法适用于各种环境,从因果推断到分析天体数据。应用估计因果关系,可以概括合成控制概念到具有个体异质性的多变量数据,并且可以共同估计所有时间段的最佳重量。