ProMoE-FL Enhances Multimodal Federated Learning with Missing Data

Aavash Chhetri, Bibek Niroula, Eduard Vazquez, Yash Raj Shrestha, Prashnna Gyawali, Loris Bazzani, Binod Bhattarai· July 9, 2026 View original

Key takeaways

  • ProMoE-FL offers a robust solution for multimodal federated learning with missing data.
  • It uses a prototype-conditioned Mixture-of-Experts for intelligent feature synthesis.
  • The framework builds a global client-aware prototype bank to capture modality priors.
  • ProMoE-FL outperforms existing methods, particularly in healthcare imaging applications.

Who benefits

HealthcareMedical ImagingPharmaceuticalsLife Sciences

Summary

This paper introduces ProMoE-FL, a Prototype-conditioned Mixture-of-Experts framework for robust multimodal federated learning, specifically addressing missing modalities. It uses a global client-aware prototype bank and dynamic expert routing to synthesize missing features, outperforming state-of-the-art methods on chest X-ray datasets.

This research addresses a significant challenge in multimodal federated learning: dealing with missing data modalities. Existing approaches often rely on public datasets or simplistic feature synthesis, which can be inadequate. The authors propose ProMoE-FL, a novel framework that leverages a Prototype-conditioned Mixture-of-Experts architecture to robustly synthesize missing features. ProMoE-FL establishes a global prototype bank that is aware of client-specific characteristics, capturing clinically relevant modality priors across different institutions. This prototype bank then conditions a Mixture of Experts, which uses modality indices to enable dynamic, direction-aware routing of experts. This mechanism allows for the intelligent synthesis of missing features. Extensive evaluations on four public chest X-ray datasets (MIMIC-CXR, NIH Open-I, PadChest, and CheXpert) demonstrate that ProMoE-FL consistently surpasses current state-of-the-art methods in both homogeneous and more complex heterogeneous settings.

Why it matters

This advancement is critical for deploying robust AI models in real-world scenarios, especially in healthcare, where data is often incomplete and distributed across multiple institutions, enabling more effective and privacy-preserving AI applications.

How to implement this in your domain

  1. 1Investigate ProMoE-FL for federated learning projects involving multimodal data with potential missing components.
  2. 2Pilot the framework in healthcare settings for diagnostic AI models that combine various imaging or patient data types.
  3. 3Collaborate with research teams to adapt and integrate this approach into existing federated learning pipelines.
  4. 4Assess the privacy implications and benefits of using client-aware prototype banks in distributed learning.

Original post by Aavash Chhetri, Bibek Niroula, Eduard Vazquez, Yash Raj Shrestha, Prashnna Gyawali, Loris Bazzani, Binod Bhattarai

"arXiv:2607.06633v1 Announce Type: cross Abstract: In this paper, we address the problem of multimodal federated learning with missing modality. Existing methods utilize an additional public dataset or perform naive feature synthesis that is based solely on the available modality.…"

View on X

Originally posted by Aavash Chhetri, Bibek Niroula, Eduard Vazquez, Yash Raj Shrestha, Prashnna Gyawali, Loris Bazzani, Binod Bhattarai on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses