ProMoE-FL: Prototype-conditioned Mixture of Experts for Multimodal Federated Learning with Missing Modalities

Aavash Chhetri, Bibek Niroula, Eduard Vazquez, Yash Raj Shrestha, Prashnna Gyawali +2 more
7/7/2026
cs.CVcs.AI

No official code found. Best implementation: bhattarailab/ProMoE-FL (1 star, MIT, last commit 2026-06-24).

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. To address these limitations, we propose ProMoE-FL, a Prototype-conditioned Mixture-of-Experts framework for robust missing-modality feature synthesis in multimodal federated learning. ProMoE-FL builds a global client-aware prototype bank that captures clinically meaningful modality priors across institutions. Our Mixture of Experts is conditioned on these prototypes and modality indices to enable direction-aware expert routing for dynamically synthesizing missing features. We perform extensive quantitative and qualitative evaluations on four public chest X-ray datasets (MIMIC-CXR, NIH Open-I, PadChest, and CheXpert) and demonstrate that ProMoE-FL consistently outperforms state-of-the-art methods in both homogeneous as well as the more challenging heterogeneous settings.

View on arXivView PDF

Code Implementations(1)

MIT

This repository contains the official implementation of our MICCAI 2026 paper "ProMoE-FL: Prototype-conditioned Mixture of Experts for Multimodal Federated Learning with Missing Modalities"

10Jun 16, 20263 months agoMIT

Cite this paper

@article{chhetri2026promoefl,
  title  = {ProMoE-FL: Prototype-conditioned Mixture of Experts for Multimodal Federated Learning with Missing Modalities},
  author = {Aavash Chhetri and Bibek Niroula and Eduard Vazquez and Yash Raj Shrestha and Prashnna Gyawali and Loris Bazzani and Binod Bhattarai},
  year   = {2026},
  eprint = {2607.06633},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2607.06633v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2607.06633.svg)](https://codeofpaper.com/paper/2607.06633)

Discussion