H\"older++: Improving the Quality-Coherence Trade-off in Multimodal VAEs
About
Existing approaches for multimodal variational autoencoders (VAEs) face a trade-off between generative quality and coherence-i.e., they struggle to generate realistic and diverse samples that, at the same time, are semantically consistent across modalities. A recent work shows that using a simple approximation to H\"older pooling as an aggregation method improves coherence over the SOTA MMVAE+, despite assuming a single shared representation across all modalities. Yet, it slightly compromises sample diversity. Inspired by this insight, we propose H\"older++, a novel multimodal VAE that improves the generative quality-coherence trade-off through: (i) the first implementation of H\"older pooling without any approximation for multimodal VAEs; (ii) an extended architecture that models distinct shared and private (i.e., modality-specific) representations (H\"older+); and (iii) hierarchical inference that further enhances the disentanglement between the shared and private representations (H\"older++). Our experiments corroborate that H\"older++ consistently improves the generative quality-coherence trade-off, yields more structured latent spaces, and learns shared representations that are informative for downstream tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Generation | CelebAMask-HQ | FID72.32 | 12 | |
| Attribute Prediction | CelebAMask-HQ | F1 Score66.5 | 8 | |
| Mask Prediction | CelebAMask-HQ | F1 Score89.6 | 8 | |
| Bird Species Classification | CUBICC | Joint Score0.827 | 7 | |
| Clustering | CUBICC | Accuracy (ACC)59.1 | 7 | |
| Clustering | CUBICC Caption Representation | Accuracy (ACC)45.3 | 7 | |
| Clustering | CUBICC Joint Representation | Accuracy65.3 | 7 | |
| Disentanglement | CUBICC | W Content Accuracy19.5 | 7 | |
| Bird Species Classification | CUBICC Caption Representation | Joint Information0.62 | 7 | |
| Digit Classification on Latent Representations | MNIST-SVHN SVHN | Joint Accuracy92.3 | 6 |