Learning Emergent Modular Representations in Multi-modality Medical Vision Foundation Models
Not on the map. Why?
Abstract (arXiv)
Multi-modality medical vision (MV) foundation models (FM) are fundamentally challenged by pronounced Non-IID feature statistics across heterogeneous imaging modalities. Monolithic self-supervised optimization on such data induces conflicting gradients, driving representations to collapse toward modality-dominant shortcuts. This work reframes this failure as an imbalance between specialization and coordination in emergent modularity, and proposes Director-Experts (DEX), a modular network that explicitly regulates these dynamics in stacked modules. Each DEX module comprises a pool of experts, dynamically adapted by our image-wise activation strategy, autonomously specializing in modality-dominant statistics, together with a director, updated via our group exponential moving average, which distills multi-expert knowledge into a shared space for semantic integration across modalities, thus driving the emergence of modular representations. We curate a new benchmark, Medical Vision Universe, over 4 million images across 10 modalities, which provides a FM-level pre-training with the broadest coverage of distinct imaging modalities to our DEX. Extensive evaluations on 26 downstream tasks demonstrate improved optimization behavior and transferability, indicating DEX as a principled step toward general-purpose multi-modality medical AI. Our code and dataset will be opened at https://github.com/YutingHe-list/DEX.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 50 of its 136 references that are in the map. First posted on arXiv 21 May 2026.
Builds on
50 of 136 references.
References · 50
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2021On the Opportunities and Risks of Foundation Modelscited by 4,033
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2020Shortcut Learning in Deep Neural Networkscited by 1,969
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
- 2017ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseasescited by 972
- 2021FLAVA: A Foundational Language And Vision Alignment Modelcited by 649
- 2023Are Emergent Abilities of Large Language Models a Mirage?cited by 523
- 2016Cross-stitch Networks for Multi-task Learningcited by 467
38 more
- 2021Domain Adaptation for Medical Image Analysis: A Surveycited by 352
- 2021Federated Learning on Non-IID Data: A Surveycited by 346
- 2020Knowledge Distillation and Student-Teacher Learning for Visual Intelligence: A Review and New Outlookscited by 316
- 2023Multimodal Foundation Models: From Specialists to General-Purpose Assistantscited by 247
- 2019REFUGE Challenge: A Unified Framework for Evaluating Automated Methods for Glaucoma Assessment from Fundus Photographscited by 228
- 2023Segment Anything in Medical Imagescited by 227
- 2024Uni-MoE: Scaling Unified Multimodal LLMs with Mixture of Expertscited by 108
- 2020CholecSeg8k: A Semantic Segmentation Dataset for Laparoscopic Cholecystectomy Based on Cholec80cited by 106
- 2019Evaluation of Algorithms for Multi-Modality Whole Heart Segmentation: An Open-Access Grand Challengecited by 100
- 2020Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Maskscited by 98
- 2020ROSE: A Retinal OCT-Angiography Vessel Segmentation Dataset and New Modelcited by 78
- 2024The 2024 Brain Tumor Segmentation (BraTS) Challenge: Glioma Segmentation on Post-treatment MRIcited by 69
- 2023USFM: A Universal Ultrasound Foundation Model Generalized to Tasks and Organs towards Label Efficient Image Analysiscited by 68
- 2023LVM-Med: Learning Large-Scale Self-Supervised Vision Models for Medical Imaging via Second-order Graph Matchingcited by 60
- 2020An automatic multi-tissue human fetal brain segmentation benchmark using the Fetal Tissue Annotation Datasetcited by 59
- 2024TotalSegmentator MRI: Robust Sequence-independent Segmentation of Multiple Anatomic Structures in MRIcited by 59
- 2019Multimodal Self-Supervised Learning for Medical Image Analysiscited by 57
- 2022ImageCAS: A Large-Scale Dataset and Benchmark for Coronary Artery Segmentation based on Computed Tomography Angiography Imagescited by 55
- 2024A Multimodal Vision Foundation Model for Clinical Dermatologycited by 50
- 2022XMorpher: Full Transformer for Deformable Medical Image Registration via Cross Attentioncited by 46
- 2021Multi-Modal Transformer for Accelerated MR Imagingcited by 43
- 2022REFUGE2 Challenge: A Treasure Trove for Multi-Dimension Analysis and Evaluation in Glaucoma Screeningcited by 38
- 2023Emergent Modularity in Pre-trained Transformerscited by 37
- 2022Mitigating Modality Collapse in Multimodal VAEs via Impartial Optimizationcited by 32
- 2025A Closer Look at Multimodal Representation Collapsecited by 28
- 2023Unlocking Emergent Modularity in Large Language Modelscited by 17
- 2024SDR-Former: A Siamese Dual-Resolution Transformer for Liver Lesion Classification Using 3D Multi-Phase Imagingcited by 17
- 2023Multimodal Machine Learning in Image-Based and Clinical Biomedicine: Survey and Prospectscited by 15
- 2024Medical Vision Generalist: Unifying Medical Imaging Tasks in Contextcited by 13
- 2025Aligning Multimodal Representations through an Information Bottleneckcited by 13
- 2025SegAnyPET: Universal Promptable Segmentation from Positron Emission Tomography Imagescited by 11
- 2023DIAS: A Dataset and Benchmark for Intracranial Artery Segmentation in DSA sequencescited by 10
- 2025MedMoE: Modality-Specialized Mixture of Experts for Medical Vision-Language Understandingcited by 10
- 2024DSCA: A Digital Subtraction Angiography Sequence Dataset and Spatio-Temporal Model for Cerebral Artery Segmentationcited by 8
- 2024BSAFusion: A Bidirectional Stepwise Feature Alignment Network for Unaligned Medical Image Fusioncited by 6
- 2025Vector Contrastive Learning For Pixel-Wise Pretraining In Medical Visioncited by 3
- 2020CoDiNet: Path Distribution Modeling with Consistency and Diversity for Dynamic Routingcited by 2
- 2024On Calibration in Multi-Distribution Learningcited by 1
Built on by
Not cited yet.