Accelerating the Sinkhorn algorithm for sparse multi-marginal optimal transport by fast Fourier transforms

DOI10.3390/A15090311arXiv2208.03120MaRDI QIDQ6407074FDOQ6407074

Authors: Fatima Antarou Ba, Michael Quellmalz 

Publication date: 5 August 2022

Abstract: We consider the numerical solution of the discrete multi-marginal optimal transport (MOT) by means of the Sinkhorn algorithm. In general, the Sinkhorn algorithm suffers from the curse of dimensionality with respect to the number of marginals. If the MOT cost function decouples according to a tree or circle, its complexity is linear in the number of marginal measures. In this case, we speed up the convolution with the radial kernel required in the Sinkhorn algorithm by non-uniform fast Fourier methods. Each step of the proposed accelerated Sinkhorn algorithm with a tree-structured cost function has a complexity of $mathcalO(KN)$ instead of the classical $mathcalO(K{N}^2)$ for straightforward matrix-vector operations, where $K$ is the number of marginals and each marginal measure is supported on at most $N$ points. In case of a circle-structured cost function, the complexity improves from $mathcalO(K{N}^3)$ to $mathcalO(K{N}^2)$. This is confirmed by numerical experiments.

Mathematics Subject Classification ID

Optimal transportation (49Q22) Numerical aspects of computer graphics, image analysis, and computational geometry (65D18) Numerical methods of relaxation type (49M20) Numerical methods for discrete and fast Fourier transforms (65T50)

This page was built for publication: Accelerating the Sinkhorn algorithm for sparse multi-marginal optimal transport by fast Fourier transforms

Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q6407074)