arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.01296cs.LGstat.ML

Regression via Implicit Models and Optimal Transport Cost Minimization

University of Minnesota(明尼苏达大学);Virginia Tech(弗吉尼亚理工大学);Criteo AI Lab(Criteo人工智能实验室);LinkedIn Corporation(领英公司)

Saurav Manchanda, Khoa Doan, Pranjul Yadav, S. Sathiya Keerthi

英文摘要:

This paper addresses the classic problem of regression, which involves the inductive learning of a map, $y=f(x,z)$, $z$ denoting noise, $f:\mathbb{R}^n\times \mathbb{R}^k \rightarrow \mathbb{R}^m$. Recently, Conditional GAN (CGAN) has been applied for regression and has shown to be advantageous over the other standard approaches like Gaussian Process Regression, given its ability to implicitly model complex noise forms. However, the current CGAN implementation for regression uses the classical generator-discriminator architecture with the minimax optimization approach, which is notorious for being difficult to train due to issues like training instability or failure to converge. In this paper, we take another step towards regression models that implicitly model the noise, and propose a solution which directly optimizes the optimal transport cost between the true probability distribution $p(y|x)$ and the estimated distribution $\hat{p}(y|x)$ and does not suffer from the issues associated with the minimax approach. On a variety of synthetic and real-world datasets, our proposed solution achieves state-of-the-art results. The code accompanying this paper is available at "https://github.com/gurdaspuriya/ot_regression".

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑