Embedding Empirical Distributions for Computing Optimal Transport Maps

  • 2025-04-24 16:52:48
  • Mingchen Jiang, Peng Xu, Xichen Ye, Xiaohui Chen, Yun Yang, Yifan Chen
  • 0

Abstract

Distributional data have become increasingly prominent in modern signalprocessing, highlighting the necessity of computing optimal transport (OT) mapsacross multiple probability distributions. Nevertheless, recent studies onneural OT methods predominantly focused on the efficient computation of asingle map between two distributions. To address this challenge, we introduce anovel approach to learning transport maps for new empirical distributions.Specifically, we employ the transformer architecture to produce embeddings fromdistributional data of varying length; these embeddings are then fed into ahypernetwork to generate neural OT maps. Various numerical experiments wereconducted to validate the embeddings and the generated OT maps. The modelimplementation and the code are provided onhttps://github.com/jiangmingchen/HOTET.