This paper studies the problem of distribution matching (DM), which is a fundamental machine learning problem seeking to robustly align two probability distributions. Our approach is established on a relaxed formulation, called partial distribution matching (PDM), which seeks to match a fraction of the distributions instead of matching them completely. We theoretically derive the Kantorovich-Rubinstein duality for the partial Wasserstain-1 (PW) discrepancy, and develop a partial Wasserstein adversarial network (PWAN) that efficiently approximates the PW discrepancy based on this dual form. Partial matching can then be achieved by optimizing the network using gradient descent. Two practical tasks, point set registration and partial domain adaptation are investigated, where the goals are to partially match distributions in 3D space and high-dimensional feature space respectively. The experiment results confirm that the proposed PWAN effectively produces highly robust matching results, performing better or on par with the state-of-the-art methods.

本研究解决了分布匹配问题，即如何稳健地对齐两个概率分布。提出了一种新的部分分布匹配方法（PDM），通过部分Wasserstein对抗网络（PWAN）实现高效近似，实验表明该方法在点集配准和部分领域适应任务中表现出色，具有较强的实用性和性能优势。

通过部分Wasserstein对抗网络进行部分分布匹配