On the pythagorean structure of the optimal transport for separable cost functions

Abstract

In this paper, we study the optimal transport problem induced by two measures supported over two polish spaces, namely, X and Y , which are the product of n smaller polish spaces, that is, X={\Large{\times}}_{j=1}^n X_j and Y={\Large{\times}}_{j=1}^n Y_j . In particular, we focus on problems induced by a cost function c:X\times Y\to [0,+\infty) that is separable; i.e., c is such that c=c_1+\dots+c_n , where each c_j depends only on the couple (x_j,y_j) , and thus c_j:X_j\times Y_j\to [0,+\infty) . Noticeably, if X=Y=\mathbb{R}^n , this class of cost functions includes all the l_p^p costs. Our main result proves that the optimal transportation plan with respect to a separable cost function between two given measures can be expressed as the composition of n different lower-dimensional transports, one for each pair of coordinates (x_i,y_i) in X\times Y . This allows us to decompose the entire Wasserstein cost as the sum of n lower-dimensional Wasserstein costs and to prove that there always exists an optimal transportation plan whose random variable enjoys a conditional independence property with respect to its marginals. We then show that our formalism allows us to explicitly compute the optimal transportation plan between two probability measures when each measure has independent marginals. Finally, we focus on two specific frameworks. In the first one, the cost function is a separable distance, i.e., d=d_1+d_2 , where both d_1 and d_2 are distances themselves. In the second one, both measures are supported over \mathbb{R}^n and the cost function is of the form c(x,y)=h(\vert x_1-y_1\vert)+h(\vert x_2-y_2\vert) , where h is a convex function such that h(0)=0 .