Knowledge Transfer Scaling Laws for 3D Medical Imaging
Not on the map. Why?
Abstract (arXiv)
Vision foundation models are increasingly moving beyond 2D to volumetric domains such as 3D medical imaging, where unified pretraining across different imaging modalities (i.e. CT, MRI, and PET) could provide foundational models for diverse clinical tasks. However, training such models requires mixing heterogeneous imaging domains, and current mixture strategies remain largely heuristic. In this work, we observe that different medical imaging domains scale at variable rates during pretraining, and knowledge transfer between domains is strongly asymmetric: training on one domain can substantially improve another, but the reverse may be much weaker. Interestingly, both MAE reconstruction loss and cross-domain transfer follow predictable power-law trends with domain-specific behaviors. Motivated by these findings, we formulate data allocation as a scaling-law optimization problem. The derived allocations reveal an interpretable hub-and-island structure: highly transferable domains emerge as hubs that benefit many others and deserve strategic allocation, while isolated domains act as islands requiring direct investment. Empirically, transfer-aware allocation outperforms data-proportional sampling by up to 58% and generalizes well to unseen budgets with r=0.989. Downstream validation on disease classification and organ/lesion segmentation further confirms that the derived transfer-aware mixtures provide stronger pretrained representations for clinical 3D medical imaging tasks.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 38 of its 48 references that are in the map. First posted on arXiv 7 May 2026.
Builds on
38 of 48 references.
References · 38
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2020mT5: A massively multilingual pre-trained text-to-text transformercited by 1,818
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 959
- 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
- 2021MedMNIST v2 -- A large-scale lightweight benchmark for 2D and 3D biomedical image classificationcited by 690
- 2021Self-Supervised Pre-Training of Swin Transformers for 3D Medical Image Analysiscited by 422
- 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 399
- 2022AMOS: A Large-Scale Abdominal Multi-Organ Benchmark for Versatile Medical Image Segmentationcited by 351
- 2019Med3D: Transfer Learning for 3D Medical Image Analysiscited by 288
26 more
- 2021Scaling Laws for Transfercited by 272
- 2019Massively Multilingual Neural Machine Translation in the Wild: Findings and Challengescited by 257
- 2024Data Mixing Laws: Optimizing Data Mixtures by Predicting Language Modeling Performancecited by 184
- 2024RegMix: Data Mixture as Regression for Language Model Pre-trainingcited by 177
- 2023Benchmarking Foundation Models with Language-Model-as-an-Examinercited by 172
- 2023UniMax: Fairer and more Effective Language Sampling for Large-Scale Multilingual Pretrainingcited by 151
- 20203D Self-Supervised Methods for Medical Imagingcited by 134
- 20223D UX-Net: A Large Kernel Volumetric ConvNet Modernizing Hierarchical Transformer for Medical Image Segmentationcited by 94
- 2021UniMiSS: Universal Medical Self-Supervised Learning via Breaking Dimensionality Barriercited by 55
- 2025Scaling Laws for Optimal Data Mixturescited by 52
- 2023The Brain Tumor Segmentation (BraTS) Challenge 2023: Glioma Segmentation in Sub-Saharan Africa Patient Population (BraTS-Africa)cited by 51
- 2024FairMedFM: Fairness Benchmarking for Medical Imaging Foundation Modelscited by 47
- 2023MedFMC: A Real-world Dataset and Benchmark For Foundation Model Adaptation in Medical Image Classificationcited by 38
- 2023Continual Self-supervised Learning: Towards Universal Multi-modal Medical Data Representation Learningcited by 34
- 2023Training Like a Medical Resident: Context-Prior Learning Toward Universal Medical Image Segmentationcited by 33
- 2024Touchstone Benchmark: Are We on the Right Way for Evaluating AI Algorithms for Medical Segmentation?cited by 31
- 2025A generalizable 3D framework and model for self-supervised learning in medical imagingcited by 31
- 2023BHSD: A 3D Multi-Class Brain Hemorrhage Segmentation Datasetcited by 27
- 2020Models Genesiscited by 26
- 2023MIS-FM: 3D Medical Image Segmentation using Foundation Models Pretrained on a Large-Scale Unannotated Datasetcited by 26
- 2024An OpenMind for 3D medical vision self-supervised learningcited by 26
- 2024BiMix: A Bivariate Data Mixing Law for Language Model Pretrainingcited by 25
- 2024Multi-modal Vision Pre-training for Medical Image Analysiscited by 23
- 2023Scaling Up 3D Kernels with Bayesian Frequency Re-parameterization for Medical Image Segmentationcited by 12
- 2024FEDMEKI: A Benchmark for Scaling Medical Foundation Models via Federated Knowledge Injectioncited by 8
- 2025Data Scaling Laws for Radiology Foundation Modelscited by 5
Built on by
Not cited yet.