Learning more expressive joint distributions in multimodal variational methods
This addresses the problem of modeling complex joint distributions in multimodal data for computer vision applications, representing an incremental improvement over existing variational inference methods.
The paper tackles the limited expressive power of multimodal variational methods by introducing a method that uses normalizing flows to approximate the joint posterior, improving state-of-the-art performance on tasks like colorization and edge detection.
Data often are formed of multiple modalities, which jointly describe the observed phenomena. Modeling the joint distribution of multimodal data requires larger expressive power to capture high-level concepts and provide better data representations. However, multimodal generative models based on variational inference are limited due to the lack of flexibility of the approximate posterior, which is obtained by searching within a known parametric family of distributions. We introduce a method that improves the representational capacity of multimodal variational methods using normalizing flows. It approximates the joint posterior with a simple parametric distribution and subsequently transforms into a more complex one. Through several experiments, we demonstrate that the model improves on state-of-the-art multimodal methods based on variational inference on various computer vision tasks such as colorization, edge and mask detection, and weakly supervised learning. We also show that learning more powerful approximate joint distributions improves the quality of the generated samples. The code of our model is publicly available at https://github.com/SashoNedelkoski/BPFDMVM.