NeurIPS2020

Multimodal Generative Learning Utilizing Jensen-Shannon-Divergence

Thomas M. Sutter, Imant Daunhawer, Julia E. Vogt

95 citations

Abstract

Learning from different data types is a long-standing goal in machine learning research, as multiple information sources co-occur when describing natural phenomena. However, existing generative models that approximate a multimodal ELBO rely on difficult or inefficient training schemes to learn a joint distribution and the dependencies between modalities. In this work, we propose a novel, efficient objective function that utilizes the Jensen-Shannon divergence for multiple distributions. It simultaneously approximates the unimodal and joint multimodal posteriors directly via a dynamic prior. In addition, we theoretically prove that the new multimodal JS-divergence (mmJSD) objective optimizes an ELBO. In extensive experiments, we demonstrate the advantage of the proposed mmJSD model compared to previous work in unsupervised, generative learning tasks. Joint and Conditional Generation. [27] implemented a multimodal VAE and introduced the idea that the distribution of the unimodal approximation should be close to the multimodal approximation function. [31] introduced the triple ELBO as an additional improvement. Both define labels as second modality and are not scalable in the number of modalities.