Learning Contrastive Multimodal Fusion with Improved Modality Dropout for Disease Detection and Prediction

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Gu, Yi, Saito, Kuniaki, Ma, Jiaxin
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914051387293696
author Gu, Yi
Saito, Kuniaki
Ma, Jiaxin
author_facet Gu, Yi
Saito, Kuniaki
Ma, Jiaxin
contents As medical diagnoses increasingly leverage multimodal data, machine learning models are expected to effectively fuse heterogeneous information while remaining robust to missing modalities. In this work, we propose a novel multimodal learning framework that integrates enhanced modalities dropout and contrastive learning to address real-world limitations such as modality imbalance and missingness. Our approach introduces learnable modality tokens for improving missingness-aware fusion of modalities and augments conventional unimodal contrastive objectives with fused multimodal representations. We validate our framework on large-scale clinical datasets for disease detection and prediction tasks, encompassing both visual and tabular modalities. Experimental results demonstrate that our method achieves state-of-the-art performance, particularly in challenging and practical scenarios where only a single modality is available. Furthermore, we show its adaptability through successful integration with a recent CT foundation model. Our findings highlight the effectiveness, efficiency, and generalizability of our approach for multimodal learning, offering a scalable, low-cost solution with significant potential for real-world clinical applications. The code is available at https://github.com/omron-sinicx/medical-modality-dropout.
format Preprint
id arxiv_https___arxiv_org_abs_2509_18284
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning Contrastive Multimodal Fusion with Improved Modality Dropout for Disease Detection and Prediction
Gu, Yi
Saito, Kuniaki
Ma, Jiaxin
Computer Vision and Pattern Recognition
As medical diagnoses increasingly leverage multimodal data, machine learning models are expected to effectively fuse heterogeneous information while remaining robust to missing modalities. In this work, we propose a novel multimodal learning framework that integrates enhanced modalities dropout and contrastive learning to address real-world limitations such as modality imbalance and missingness. Our approach introduces learnable modality tokens for improving missingness-aware fusion of modalities and augments conventional unimodal contrastive objectives with fused multimodal representations. We validate our framework on large-scale clinical datasets for disease detection and prediction tasks, encompassing both visual and tabular modalities. Experimental results demonstrate that our method achieves state-of-the-art performance, particularly in challenging and practical scenarios where only a single modality is available. Furthermore, we show its adaptability through successful integration with a recent CT foundation model. Our findings highlight the effectiveness, efficiency, and generalizability of our approach for multimodal learning, offering a scalable, low-cost solution with significant potential for real-world clinical applications. The code is available at https://github.com/omron-sinicx/medical-modality-dropout.
title Learning Contrastive Multimodal Fusion with Improved Modality Dropout for Disease Detection and Prediction
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2509.18284