Multimodal Federated Learning With Missing Modalities through Feature Imputation Network

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Poudel, Pranav, Chhetri, Aavash, Gyawali, Prashnna, Leontidis, Georgios, Bhattarai, Binod
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916759939842048
author Poudel, Pranav
Chhetri, Aavash
Gyawali, Prashnna
Leontidis, Georgios
Bhattarai, Binod
author_facet Poudel, Pranav
Chhetri, Aavash
Gyawali, Prashnna
Leontidis, Georgios
Bhattarai, Binod
contents Multimodal federated learning holds immense potential for collaboratively training models from multiple sources without sharing raw data, addressing both data scarcity and privacy concerns, two key challenges in healthcare. A major challenge in training multimodal federated models in healthcare is the presence of missing modalities due to multiple reasons, including variations in clinical practice, cost and accessibility constraints, retrospective data collection, privacy concerns, and occasional technical or human errors. Previous methods typically rely on publicly available real datasets or synthetic data to compensate for missing modalities. However, obtaining real datasets for every disease is impractical, and training generative models to synthesize missing modalities is computationally expensive and prone to errors due to the high dimensionality of medical data. In this paper, we propose a novel, lightweight, low-dimensional feature translator to reconstruct bottleneck features of the missing modalities. Our experiments on three different datasets (MIMIC-CXR, NIH Open-I, and CheXpert), in both homogeneous and heterogeneous settings consistently improve the performance of competitive baselines. The code and implementation details are available at: https://github.com/bhattarailab/FedFeatGen
format Preprint
id arxiv_https___arxiv_org_abs_2505_20232
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Multimodal Federated Learning With Missing Modalities through Feature Imputation Network
Poudel, Pranav
Chhetri, Aavash
Gyawali, Prashnna
Leontidis, Georgios
Bhattarai, Binod
Machine Learning
Computer Vision and Pattern Recognition
Multimodal federated learning holds immense potential for collaboratively training models from multiple sources without sharing raw data, addressing both data scarcity and privacy concerns, two key challenges in healthcare. A major challenge in training multimodal federated models in healthcare is the presence of missing modalities due to multiple reasons, including variations in clinical practice, cost and accessibility constraints, retrospective data collection, privacy concerns, and occasional technical or human errors. Previous methods typically rely on publicly available real datasets or synthetic data to compensate for missing modalities. However, obtaining real datasets for every disease is impractical, and training generative models to synthesize missing modalities is computationally expensive and prone to errors due to the high dimensionality of medical data. In this paper, we propose a novel, lightweight, low-dimensional feature translator to reconstruct bottleneck features of the missing modalities. Our experiments on three different datasets (MIMIC-CXR, NIH Open-I, and CheXpert), in both homogeneous and heterogeneous settings consistently improve the performance of competitive baselines. The code and implementation details are available at: https://github.com/bhattarailab/FedFeatGen
title Multimodal Federated Learning With Missing Modalities through Feature Imputation Network
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.20232