A Volumetric Approach to Monge's Optimal Transport on Surfaces

Richard Tsai,Axel G. R. Turnquist

2024-05-15

Abstract:We propose a volumetric formulation for computing the Optimal Transport problem defined on surfaces in $\mathbb{R}^3$, found in disciplines like optics, computer graphics, and computational methodologies. Instead of directly tackling the original problem on the surface, we define a new Optimal Transport problem on a thin tubular region, $T_{\epsilon}$, adjacent to the surface. This extension offers enhanced flexibility and simplicity for numerical discretization on Cartesian grids. The Optimal Transport mapping and potential function computed on $T_{\epsilon}$ are consistent with the original problem on surfaces. We demonstrate that, with the proposed volumetric approach, it is possible to use simple and straightforward numerical methods to solve Optimal Transport for $\Gamma = \mathbb{S}^2$ and the $2$-torus.

Numerical Analysis

What problem does this paper attempt to address?

The paper aims to address the Optimal Transport (OT) problem on surfaces in three-dimensional space and proposes a novel volumetric method to compute such problems. Specifically, the authors' goal is to define the optimal transport problem on a smooth three-dimensional surface \(\Gamma\) and solve the corresponding Monge-Ampère type optimal transport partial differential equation.

To tackle this problem, the authors do not directly handle the original optimal transport problem on the surface. Instead, they define a new optimal transport problem in a thin tubular neighborhood \(T_\epsilon\) around the surface \(\Gamma\). This approach enhances the flexibility and simplicity of numerical discretization and allows computations to be performed on a Cartesian grid. The optimal transport map and potential function computed in \(T_\epsilon\) are consistent with the problem on the original surface.

The main advantages of this method include:
- The ability to use simple and intuitive numerical methods to solve the optimal transport problem in \(T_\epsilon\).
- Effective handling of the optimal transport problem in specific application scenarios, such as optics, computer graphics, and computational methods.
- The method allows for easier numerical handling of issues like covariant derivatives on surfaces.

Additionally, the paper discusses the application cases of this method in different fields, including the design of mirrors or lenses in freeform optics and the use of optimal transport as a tool in statistical analysis. Finally, the paper provides a detailed explanation of how to set up the new optimal transport problem in \(T_\epsilon\) and demonstrates that the optimal transport map on the original surface \(\Gamma\) can be extracted from this new problem.