スライスされたワッサースタイン (SW) 距離は、2 つの確率尺度間の統計的に効果的で計算効率の良いメトリックとして広く認識されています。 SW 距離の重要な要素は、スライス分布です。この分布を選択するには、2 つの既存のアプローチがあります。最初のアプローチは、固定事前分布を使用することです。 2 番目のアプローチは、分布のパラメトリック ファミリに属し、期待される距離を最大化できる最適な分布を最適化することです。ただし、どちらのアプローチにも限界があります。固定事前分布は、2 つの一般的な確率尺度を区別できる射影方向を強調するという点で、有益ではありません。最良の分布を得るために最適化を行うことは、多くの場合、高価で不安定です。さらに、候補分布のパラメトリック ファミリの設計は、簡単に誤って指定される可能性があります。この問題に対処するために、スライス分布を、パラメーターがなく、投影された 1 次元のワッサースタイン距離のエネルギー関数に比例する密度を持つエネルギーベースの分布として設計することを提案します。次に、新しいスライス ワッサースタイン メトリック、エネルギーベースのスライス ワッサースタイン (EBSW) 距離を導出し、重要度サンプリング、サンプリング重要度リサンプリング、およびマルコフ連鎖法を介して、その位相的、統計的、および計算上の特性を調査します。最後に、ポイント クラウド勾配フロー、カラー トランスファー、およびポイント クラウド再構築に関する実験を行い、EBSW の良好なパフォーマンスを示します。
The sliced Wasserstein (SW) distance has been widely recognized as a statistically effective and computationally efficient metric between two probability measures. A key component of the SW distance is the slicing distribution. There are two existing approaches for choosing this distribution. The first approach is using a fixed prior distribution. The second approach is optimizing for the best distribution which belongs to a parametric family of distributions and can maximize the expected distance. However, both approaches have their limitations. A fixed prior distribution is non-informative in terms of highlighting projecting directions that can discriminate two general probability measures. Doing optimization for the best distribution is often expensive and unstable. Moreover, designing the parametric family of the candidate distribution could be easily misspecified. To address the issues, we propose to design the slicing distribution as an energy-based distribution that is parameter-free and has the density proportional to an energy function of the projected one-dimensional Wasserstein distance. We then derive a novel sliced Wasserstein metric, energy-based sliced Waserstein (EBSW) distance, and investigate its topological, statistical, and computational properties via importance sampling, sampling importance resampling, and Markov Chain methods. Finally, we conduct experiments on point-cloud gradient flow, color transfer, and point-cloud reconstruction to show the favorable performance of the EBSW.