用熵正则化沃瑟斯坦距离计算概率测度的均值与分解,提升小样本点云分类效率。
Synthesis and Analysis of Data as Probability Measures with Entropy-Regularized Optimal Transport
- 基于熵正则化沃瑟斯坦-2距离定义测度均值,通过不动点公式求解。
- 在小样本下实现稳定收敛,系数估计误差率不随维度增长。
- 适用于点云数据分类,尤其在训练数据少时优于神经网络方法。
本文研究基于熵正则化沃瑟斯坦-2代价及其无偏版本Sinkhorn散度的概率测度合成与分析问题。合成问题为在给定单纯形系数下计算参考测度的巴里中心;分析问题为寻找使巴里中心最接近目标测度的系数。在文献中最弱假设下,我们推导出熵正则化沃瑟斯坦-2代价的导数,并证明其巴里中心是熵正则化位移映射平均的不动点。该性质导出有限维凸二次规划,用于求解分析问题。当被分析测度本身为巴里中心时,系数及泛函值可从样本中以维度无关速率估计,且系数对沃瑟斯坦-2扰动具有稳定性。我们将巴里中心系数作为特征用于降噪点云分类,实验表明在小样本场景下显著优于神经网络基线。
原文摘要 · Abstract (English)
We consider synthesis and analysis of probability measures using the entropy-regularized Wasserstein-2 cost and its unbiased version, the Sinkhorn divergence. The synthesis problem consists of computing the barycenter, with respect to these costs, of reference measures given a set of coefficients belonging to the simplex. The analysis problem consists of finding the coefficients for the closest barycenter in the Wasserstein-2 distance to a given measure. Under the weakest assumptions on the measures thus far in the literature, we compute the derivative of the entropy-regularized Wasserstein-2 cost. We leverage this to establish a characterization of barycenters with respect to the entropy-regularized Wasserstein-2 cost as solutions that correspond to a fixed point of an average of the entropy-regularized displacement maps. This characterization yields a finite-dimensional, convex, quadratic program for solving the analysis problem when the measure being analyzed is a barycenter with respect to the entropy-regularized Wasserstein-2 cost. We show that these coefficients, as well as the value of the barycenter functional, can be estimated from samples with dimension-independent rates of convergence, and that barycentric coefficients are stable with respect to perturbations in the Wasserstein-2 metric. We employ the barycentric coefficients as features for classification of corrupted point cloud data, and show that compared to neural network baselines, our approach is more efficient in small training data regimes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。