Large-scale optimal transport map estimation using projection pursuit
Cheng Meng · Yuan Ke · Jingyi Zhang · Mengrui Zhang · Wenxuan Zhong · Ping Ma

Tue Dec 10th 05:30 -- 07:30 PM @ East Exhibition Hall B + C #27

This paper studies the estimation of large-scale optimal transport maps (OTM), which is a well known challenging problem owing to the curse of dimensionality. Existing literature approximates the large-scale OTM by a series of one-dimensional OTM problems through iterative random projection. Such methods, however, suffer from slow or none convergence in practice due to the nature of randomly selected projection directions. Instead, we propose an estimation method of large-scale OTM by combining the idea of projection pursuit regression and sufficient dimension reduction. The proposed method, named projection pursuit Monge map (PPMM), adaptively selects the most informative'' projection direction in each iteration. We theoretically show the proposed dimension reduction method can consistently estimate the mostinformative'' projection direction in each iteration. Furthermore, the PPMM algorithm weakly convergences to the target large-scale OTM in a reasonable number of steps. Empirically, PPMM is computationally easy and converges fast. We assess its finite sample performance through the applications of Wasserstein distance estimation and generative models.

Author Information

Cheng Meng (University of Georgia)
Yuan Ke (University of Georgia)
Jingyi Zhang (The University of Georgia)
Mengrui Zhang (University of Georgia)
Wenxuan Zhong
Ping Ma (University of Georgia)