Wasserstein 1 最適輸送マップは、2 つの確率分布 μ と ν からのポイント間の自然な対応を提供します。これは、多くのアプリケーションで役立ちます。これらのマップを計算するために利用可能なアルゴリズムは、高次元にうまくスケーリングできないようです。深層学習アプリケーションでは、ニューラル ネットワークを使用して、カントロビッチ ポテンシャルとして知られる双対問題の解を近似するための効率的なアルゴリズムが開発されています ([Gulrajani et al., 2017] など)。重要なのは、このようなアルゴリズムが高次元でうまく機能することです。この論文では、Kantorovich ポテンシャルのみに依存する Wasserstein 1 最適輸送マップの計算に向けたアプローチを提示します。一般に、Wasserstein 1 最適輸送マップは一意ではなく、ポテンシャルだけから計算することはできません。私たちの主な結果は、μが密度を持ち、νが余次元の少なくとも2の部分多様体でサポートされている場合、最適な輸送マップは一意であり、ポテンシャルに関して明示的に記述できることを証明することです.これらの仮定は、多くの画像処理コンテキストやその他のアプリケーションでは当然のことです。カントロビッチ ポテンシャルがおおよそしかわかっていない場合、我々の結果は、データが最適な方向に正しい平均変位で移動する反復手順を動機付けます。これは、ある分布を別の分布に変換するためのアプローチを提供するため、さまざまなトランスポートの問題に対する多目的アルゴリズムとして使用できます。いくつかの概念実証実験を通じて、このアルゴリズムが、通常は特殊な技術を必要とするノイズ除去、生成、変換、ブレ除去などのさまざまなイメージング タスクを正常に実行することを示します。
Wasserstein 1 optimal transport maps provide a natural correspondence between points from two probability distributions, μ and ν, which is useful in many applications. Available algorithms for computing these maps do not appear to scale well to high dimensions. In deep learning applications, efficient algorithms have been developed for approximating solutions of the dual problem, known as Kantorovich potentials, using neural networks (e.g. [Gulrajani et al., 2017]). Importantly, such algorithms work well in high dimensions. In this paper we present an approach towards computing Wasserstein 1 optimal transport maps that relies only on Kantorovich potentials. In general, a Wasserstein 1 optimal transport map is not unique and is not computable from a potential alone. Our main result is to prove that if μ has a density and ν is supported on a submanifold of codimension at least 2, an optimal transport map is unique and can be written explicitly in terms of a potential. These assumptions are natural in many image processing contexts and other applications. When the Kantorovich potential is only known approximately, our result motivates an iterative procedure wherein data is moved in optimal directions and with the correct average displacement. Since this provides an approach for transforming one distribution to another, it can be used as a multipurpose algorithm for various transport problems; we demonstrate through several proof of concept experiments that this algorithm successfully performs various imaging tasks, such as denoising, generation, translation and deblurring, which normally require specialized techniques.