Generating samples given a specific label requires estimating conditional distributions. We derive a tractable upper bound of the Wasserstein distance between conditional distributions to lay the theoretical groundwork to learn conditional distributions. Based on this result, we propose a novel conditional generation algorithm where conditional distributions are fully characterized by a metric space defined by a statistical distance. We employ optimal transport theory to propose the \textit{Wasserstein geodesic generator}, a new conditional generator that learns the Wasserstein geodesic. The proposed method learns both conditional distributions for observed domains and optimal transport maps between them. The conditional distributions given unobserved intermediate domains are on the Wasserstein geodesic between conditional distributions given two observed domain labels. Experiments on face images with light conditions as domain labels demonstrate the efficacy of the proposed method.
翻译:给定特定标签生成样本需要估计条件分布。我们推导了条件分布之间Wasserstein距离的一个可处理上界,为学习条件分布奠定理论基础。基于这一结果,我们提出了一种新颖的条件生成算法,其中条件分布完全由统计距离定义的度量空间来刻画。我们运用最优传输理论,提出了\textit{Wasserstein测地生成器},这是一种学习Wasserstein测地线的全新条件生成器。该方法同时学习观测域的条件分布以及它们之间的最优传输映射。给定两个观测域标签的条件分布之间的未观测中间域的条件分布,位于Wasserstein测地线上。以光照条件作为域标签的人脸图像实验验证了所提方法的有效性。