FEDKIM: Adaptive Federated Knowledge Injection into Medical Foundation Models
This addresses the problem of scaling medical AI models for healthcare applications while preserving patient privacy, representing a novel method for a known bottleneck.
The study tackled the challenge of developing comprehensive medical foundation models under privacy constraints by introducing FedKIM, a federated knowledge injection approach that enhanced model performance across twelve tasks in seven modalities without direct access to sensitive data.
Foundation models have demonstrated remarkable capabilities in handling diverse modalities and tasks, outperforming conventional artificial intelligence (AI) approaches that are highly task-specific and modality-reliant. In the medical domain, however, the development of comprehensive foundation models is constrained by limited access to diverse modalities and stringent privacy regulations. To address these constraints, this study introduces a novel knowledge injection approach, FedKIM, designed to scale the medical foundation model within a federated learning framework. FedKIM leverages lightweight local models to extract healthcare knowledge from private data and integrates this knowledge into a centralized foundation model using a designed adaptive Multitask Multimodal Mixture Of Experts (M3OE) module. This method not only preserves privacy but also enhances the model's ability to handle complex medical tasks involving multiple modalities. Our extensive experiments across twelve tasks in seven modalities demonstrate the effectiveness of FedKIM in various settings, highlighting its potential to scale medical foundation models without direct access to sensitive data.