提出一种新分布距离度量,能捕捉数据流形结构,计算更快更稳定。
Mutual Regression Distance
- 通过约束互回归问题定义新距离,利用数据流形平滑性
- 简化版计算复杂度远低于Wasserstein距离,实测性能更优
- 适合流形数据的聚类、生成模型和域适应任务
最大均值差异和Wasserstein距离是常用分布距离度量,在度量学习、生成建模、域自适应和聚类中作用重要。但它们依赖于两点间成对距离,无法利用数据的流形特性(如平滑性),在流形形式下衡量分布差异效果有限。本文提出一种新距离——互回归距离(Mutual Regression Distance, MRD),基于约束互回归问题构建,可有效利用数据流形性质。证明MRD为拟度量,满足几乎全部度量公理。由于原始MRD优化成本高,我们提出紧致版和简化版MRD,据此设计启发式算法。还提供核版本的MRD,更适用于非线性数据。所提MRD,尤其是简化版,计算复杂度显著低于Wasserstein距离。理论分析提供了鲁棒性等保证。应用于分布聚类、生成模型和域适应任务,数值实验表明其性能优于基线方法。
原文摘要 · Abstract (English)
The maximum mean discrepancy and Wasserstein distance are popular distance measures between distributions and play important roles in many machine learning problems such as metric learning, generative modeling, domain adaption, and clustering. However, since they are functions of pair-wise distances between data points in two distributions, they do not exploit the potential manifold properties of data such as smoothness and hence are not effective in measuring the dissimilarity between the two distributions in the form of manifolds. In this paper, different from existing measures, we propose a novel distance called Mutual Regression Distance (MRD) induced by a constrained mutual regression problem, which can exploit the manifold property of data. We prove that MRD is a pseudometric that satisfies almost all the axioms of a metric. Since the optimization of the original MRD is costly, we provide a tight MRD and a simplified MRD, based on which a heuristic algorithm is established. We also provide kernel variants of MRDs that are more effective in handling nonlinear data. Our MRDs especially the simplified MRDs have much lower computational complexity than the Wasserstein distance. We provide theoretical guarantees, such as robustness, for MRDs. Finally, we apply MRDs to distribution clustering, generative models, and domain adaptation. The numerical results demonstrate the effectiveness and superiority of MRDs compared to the baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。