Cooperative Distribution Alignment via JSD Upper Bound
Cooperative Distribution Alignment via JSD Upper Bound
复制标题
DOI:
10.48550/arxiv.2207.02286
复制
发表时间:
2022-07
期刊:
影响因子:
--
通讯作者:
Wonwoong Cho;Ziyu Gong;David I. Inouye
中科院分区:
文献类型:
--
作者:
Wonwoong Cho;Ziyu Gong;David I. Inouye
Unsupervised distribution alignment estimates a transformation that maps two or more source distributions to a shared aligned distribution given only samples from each distribution. This task has many applications including generative modeling, unsupervised domain adaptation, and socially aware learning. Most prior works use adversarial learning (i.e., min-max optimization), which can be challenging to optimize and evaluate. A few recent works explore non-adversarial flow-based (i.e., invertible) approaches, but they lack a unified perspective and are limited in efficiently aligning multiple distributions. Therefore, we propose to unify and generalize previous flow-based approaches under a single non-adversarial framework, which we prove is equivalent to minimizing an upper bound on the Jensen-Shannon Divergence (JSD). Importantly, our problem reduces to a min-min, i.e., cooperative, problem and can provide a natural evaluation metric for unsupervised distribution alignment. We show empirical results on both simulated and real-world datasets to demonstrate the benefits of our approach. Code is available at https://github.com/inouye-lab/alignment-upper-bound.