M-IDoL: Information Decomposition for Modality-Specific and Diverse Representation Learning in Medical Foundation Model
Organizations: School of Computer Science and Technique, Tongji University, Shanghai 201804, China. · School of Communications and Electronic Engineering, East China Normal University, Shanghai 200241, China.
Abstract
Medical foundation models (MFMs) aim to learn universal representations from multimodal medical images that can generalize effectively to diverse downstream clinical tasks. However, most existing MFMs suffer from information ambiguity that blends multimodal representations in a single embedding space, leading to the degradation of modality specificity and diversity. In this paper, we propose M-IDoL, a self-supervised MFM that introduces Information Decomposition for multimodal representation Learning via two objectives: i) maximizing inter-modality entropy by dispersing multimodal representations into separable Mixture-of-Experts (MoE) subspaces to achieve representation specificity across modalities; and ii) minimizing intra-modality uncertainty by performing fine-grained semantic discrimination within each MoE subspace to enrich representation diversity per modality. By pre-training on 1.15 million medical images, M-IDoL i) delivers superior generalization across 21 downstream clinical tasks, outperforming 20 foundation models on five imaging modalities (e.g., X-ray, fundus, OCT, dermoscopy and pathology), and ii) learns modality-specific and diverse representations, showing clearer separation of feature clusters across modalities and finer-grained feature discrimination within each modality.
Figures & tables
| Pre-train | Downstream | ||||
|---|---|---|---|---|---|
| Dataset | Images | Dataset | Train | Val | Test |
| Retinology (Fundus) | Retinology (Fundus) | ||||
| APTOS ( Karthik et al., 2019 ) | 2,048 | 514 | 1,100 | ||
| EYEPACS ( Gulshan et al., 2016 ) | 88,702 | Glaucoma ( Kim, 2018 ) | 861 | 218 | 465 |
| PAPILA 1 1 1 PAPILA (2022), a retinal dataset for glaucomatous diagnosis. URL https://doi.org/10.6084/m9.figshare.14798004.v1. | 311 | 79 | 98 | ||
| AIROGS ( De Vente et al., 2023 ) | 101,442 | Retina ( Ret, ) | 336 | 84 | 181 |
| MoE | RSNA | APTOS | OCTID | Mitosis | HAM10000 | ||
|---|---|---|---|---|---|---|---|
| ✗ | ✗ | ✗ | 87.42 ± 0.57 | 90.36 ± 0.22 | 96.98 ± 0.47 | 86.72 ± 0.34 | 92.77 ± 0.19 |
| ✔ | ✗ | ✗ | 88.31 ± 0.34 | 87.97 ± 0.52 | 96.77 ± 0.59 | 88.64 ± 0.60 | 90.86 ± 0.28 |
| ✔ | ✔ | ✗ | 91.45 ± 0.16 | 93.46 ± 0.32 | 98.84 ± 0.31 | 91.75 ± 0.27 | 96.66 ± 0.19 |
| ✔ | ✔ | ✔ | 93.23 ± 0.22 | 95.19 ± 0.62 | 99.34 ± 0.35 | 95.62 ± 0.24 | 97.96 ± 0.36 |
| Retinology (Fundus) | ||||||||
| APTOS | Glaucoma | PAPILA | Retina | |||||
| Methods | ACC ( ) | AUC ( ) | ACC ( ) | AUC ( ) | ACC ( ) | AUC ( ) | ACC ( ) | AUC ( ) |
| RETFound ( Zhou et al., 2023b ) | 92.17 ± 0.35 ‡ | 94.36 ± 0.14 ‡ | 90.18 ± 0.38 ‡ | 94.36 ± 0.19 ‡ | 87.25 ± 0.26 | 84.04 ± 0.16 | 84.98 ± 0.13 ‡ | 85.27 ± 0.53 |
| KeepFIT ( Wu et al., 2024 ) | 92.05 ± 0.63 | 94.26 ± 0.61 | 87.62 ± 1.01 | 91.95 ± 0.41 | 86.64 ± 0.19 | 84.88 ± 0.39 ‡ | 84.06 ± 0.46 | 87.02 ± 0.31 ‡ |
| FLAIR ( Silva-Rodriguez et al., 2025 ) | 91.99 ± 0.31 | 93.08 ± 0.84 | 89.05 ± 0.34 | 93.99 ± 0.73 | 87.07 ± 0.59 ‡ | 83.08 ± 0.42 | 85.31 ± 0.39 | 84.28 ± 0.39 |
| M-IDoL(ours) | 93.43 ± 0.85 | 95.19 ± 0.62 | 90.97 ± 0.58 | 95.05 ± 0.29 | 88.30 ± 0.19 | 86.61 ± 0.34 | 88.4 ± 0.26 | 88.89 ± 0.47 |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| APTOS | Glaucoma | PAPILA | Retina | OCTDL | OCTID | TVHL-DME | |
| lr | 5e-4 | 5e-4 | 1e-3 | 8e-4 | 1e-4 | 1e-4 | 1e-4 |
| epoch | 50 | 50 | 50 | 50 | 50 | 50 | 50 |
| shape | 224 × 224 | 224 × 224 | 224 × 224 | 224 × 224 | 224 × 224 | 224 × 224 | 224 × 224 |
| batch size | 64 | 64 | 64 | 64 | 64 | 64 | 64 |
| task | Multi-class cls. | Multi-class cls. | Multi-class cls. | Multi-class cls. | Multi-class cls. | Multi-class cls. | Multi-class cls. |
| TVHL-DR | BreakHis-Binary | BreakHis-8C | Mitosis | MHIST | ISIC2016 | ISIC2017 |
| Retinology (Fundus) | ||||||||
| APTOS | Glaucoma | PAPILA | Retina | |||||
| Methods | ACC | AUC | ACC | AUC | ACC | AUC | ACC | AUC |
| SwinViT | 89.84 ± 0.57 | 92.30 ± 0.76 | 86.27 ± 0.53 | 91.64 ± 1.41 | 81.37 ± 0.31 | 77.24 ± 0.97 | 77.88 ± 0.14 | 73.17 ± 0.10 |
| DINO | 90.13 ± 0.62 | 92.71 ± 0.29 | 89.54 ± 0.35 | 89.79 ± 0.68 | 81.66 ± 0.14 | 79.81 ± 0.19 | 82.32 ± 0.29 | 76.43 ± 0.29 |
| MAE | 91.21 ± 0.35 | 92.22 ± 0.44 | 88.41 ± 0.72 | 92.14 ± 0.24 | 79.52 ± 0.04 | 73.64 ± 0.07 | 78.31 ± 0.17 | 72.61 ± 0.08 |
| Unimiss+ | 87.52 ± 0.88 | 89.52 ± 0.19 | 86.72 ± 0.37 | 90.26 ± 0.29 | 82.39 ± 0.14 | 73.24 ± 0.85 | 81.32 ± 0.48 | 76.19 ± 0.38 |
Medical Imaging Foundation Models
Medical imaging foundation models learn reusable representations for tasks across clinical images, modalities, and institutions.
See all 234 papers