An Information Criterion for Controlled Disentanglement of Multimodal Data
This addresses the problem of interpretability and robustness in multimodal learning for applications such as biology and vision-language tasks, though it appears incremental with a focus on scenarios not covered in prior work.
The paper tackles the challenge of disentangling modality-specific and shared information in multimodal data, proposing DisentangledSSL, which outperforms baselines on tasks like vision-language prediction and molecule-phenotype retrieval.
Multimodal representation learning seeks to relate and decompose information inherent in multiple modalities. By disentangling modality-specific information from information that is shared across modalities, we can improve interpretability and robustness and enable downstream tasks such as the generation of counterfactual outcomes. Separating the two types of information is challenging since they are often deeply entangled in many real-world applications. We propose Disentangled Self-Supervised Learning (DisentangledSSL), a novel self-supervised approach for learning disentangled representations. We present a comprehensive analysis of the optimality of each disentangled representation, particularly focusing on the scenario not covered in prior work where the so-called Minimum Necessary Information (MNI) point is not attainable. We demonstrate that DisentangledSSL successfully learns shared and modality-specific features on multiple synthetic and real-world datasets and consistently outperforms baselines on various downstream tasks, including prediction tasks for vision-language data, as well as molecule-phenotype retrieval tasks for biological data. The code is available at https://github.com/uhlerlab/DisentangledSSL.