Multimodal Variational Autoencoder: a Barycentric View

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Qiu, Peijie, Zhu, Wenhui, Kumar, Sayantan, Chen, Xiwen, Sun, Xiaotong, Yang, Jin, Razi, Abolfazl, Wang, Yalin, Sotiras, Aristeidis
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913628963209216
author Qiu, Peijie
Zhu, Wenhui
Kumar, Sayantan
Chen, Xiwen
Sun, Xiaotong
Yang, Jin
Razi, Abolfazl
Wang, Yalin
Sotiras, Aristeidis
author_facet Qiu, Peijie
Zhu, Wenhui
Kumar, Sayantan
Chen, Xiwen
Sun, Xiaotong
Yang, Jin
Razi, Abolfazl
Wang, Yalin
Sotiras, Aristeidis
contents Multiple signal modalities, such as vision and sounds, are naturally present in real-world phenomena. Recently, there has been growing interest in learning generative models, in particular variational autoencoder (VAE), to for multimodal representation learning especially in the case of missing modalities. The primary goal of these models is to learn a modality-invariant and modality-specific representation that characterizes information across multiple modalities. Previous attempts at multimodal VAEs approach this mainly through the lens of experts, aggregating unimodal inference distributions with a product of experts (PoE), a mixture of experts (MoE), or a combination of both. In this paper, we provide an alternative generic and theoretical formulation of multimodal VAE through the lens of barycenter. We first show that PoE and MoE are specific instances of barycenters, derived by minimizing the asymmetric weighted KL divergence to unimodal inference distributions. Our novel formulation extends these two barycenters to a more flexible choice by considering different types of divergences. In particular, we explore the Wasserstein barycenter defined by the 2-Wasserstein distance, which better preserves the geometry of unimodal distributions by capturing both modality-specific and modality-invariant representations compared to KL divergence. Empirical studies on three multimodal benchmarks demonstrated the effectiveness of the proposed method.
format Preprint
id arxiv_https___arxiv_org_abs_2412_20487
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Multimodal Variational Autoencoder: a Barycentric View
Qiu, Peijie
Zhu, Wenhui
Kumar, Sayantan
Chen, Xiwen
Sun, Xiaotong
Yang, Jin
Razi, Abolfazl
Wang, Yalin
Sotiras, Aristeidis
Machine Learning
Computer Vision and Pattern Recognition
Information Theory
Multiple signal modalities, such as vision and sounds, are naturally present in real-world phenomena. Recently, there has been growing interest in learning generative models, in particular variational autoencoder (VAE), to for multimodal representation learning especially in the case of missing modalities. The primary goal of these models is to learn a modality-invariant and modality-specific representation that characterizes information across multiple modalities. Previous attempts at multimodal VAEs approach this mainly through the lens of experts, aggregating unimodal inference distributions with a product of experts (PoE), a mixture of experts (MoE), or a combination of both. In this paper, we provide an alternative generic and theoretical formulation of multimodal VAE through the lens of barycenter. We first show that PoE and MoE are specific instances of barycenters, derived by minimizing the asymmetric weighted KL divergence to unimodal inference distributions. Our novel formulation extends these two barycenters to a more flexible choice by considering different types of divergences. In particular, we explore the Wasserstein barycenter defined by the 2-Wasserstein distance, which better preserves the geometry of unimodal distributions by capturing both modality-specific and modality-invariant representations compared to KL divergence. Empirical studies on three multimodal benchmarks demonstrated the effectiveness of the proposed method.
title Multimodal Variational Autoencoder: a Barycentric View
topic Machine Learning
Computer Vision and Pattern Recognition
Information Theory
url https://arxiv.org/abs/2412.20487