arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1908.03015cs.LGcs.AIcs.NEstat.ML

Augmenting Variational Autoencoders with Sparse Labels: A Unified Framework for Unsupervised, Semi-(un)supervised, and Supervised Learning

Relayr (GmbH)(Relayr有限公司)

Felix Berkhahn, Richard Keys, Wajih Ouertani, Nikhil Shetty, Dominik Geißler

英文摘要:

We present a new flavor of Variational Autoencoder (VAE) that interpolates seamlessly between unsupervised, semi-supervised and fully supervised learning domains. We show that unlabeled datapoints not only boost unsupervised tasks, but also the classification performance. Vice versa, every label not only improves classification, but also unsupervised tasks. The proposed architecture is simple: A classification layer is connected to the topmost encoder layer, and then combined with the resampled latent layer for the decoder. The usual evidence lower bound (ELBO) loss is supplemented with a supervised loss target on this classification layer that is only applied for labeled datapoints. This simplicity allows for extending any existing VAE model to our proposed semi-supervised framework with minimal effort. In the context of classification, we found that this approach even outperforms a direct supervised setup.

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

↑