MAFM^3: Modular Adaptation of Foundation Models for Multi-Modal Medical AI

TL;DR

This paper introduces MAFM^3, a modular framework that adapts foundation models for multi-modal medical imaging, enabling efficient multitask and multimodality capabilities with improved performance.

Contribution

It presents a novel modular adaptation framework allowing a single foundation model to expand into diverse medical imaging tasks and modalities efficiently.

Findings

Improved performance on prognosis and segmentation tasks.

Achieved 5% Dice score improvement with PET scans.

Demonstrated effective multitask and multimodality adaptation.

Abstract

Foundational models are trained on extensive datasets to capture the general trends of a domain. However, in medical imaging, the scarcity of data makes pre-training for every domain, modality, or task challenging. Instead of building separate models, we propose MAFM^3 (Modular Adaptation of Foundation Models for Multi-Modal Medical AI), a framework that enables a single foundation model to expand into diverse domains, tasks, and modalities through lightweight modular components. These components serve as specialized skill sets that allow the system to flexibly activate the appropriate capability at the inference time, depending on the input type or clinical objective. Unlike conventional adaptation methods that treat each new task or modality in isolation, MAFM^3 provides a unified and expandable framework for efficient multitask and multimodality adaptation. Empirically, we validate our approach by adapting a chest CT foundation model initially trained for classification into prognosis and segmentation modules. Our results show improved performance on both tasks. Furthermore, by incorporating PET scans, MAFM^3 achieved an improvement in the Dice score 5% compared to the respective baselines. These findings establish that foundation models, when equipped with modular components, are not inherently constrained to their initial training scope but can evolve into multitask, multimodality systems for medical imaging. The code implementation of this work can be found at https://github.com/Areeb2735/CTscan_prognosis_VLM