Learning Reconfigurable Representations for Multimodal Federated Learning with Missing Data
Duong M. Nguyen, Trong Nghia Hoang, Thanh Trung Huynh, Quoc Viet Hung Nguyen, Phi Le Nguyen
Abstract
Multimodal federated learning in real-world settings often encounters incomplete and heterogeneous data across clients. This results in misaligned local feature representations that limit the effectiveness of model aggregation. Unlike prior work that assumes either differing modality sets without missing input features or a shared modality set with missing features across clients, we consider a more general and realistic setting where each client observes a different subset of modalities and might also have missing input features within each modality. To address the resulting misalignment in learned representations, we propose a new federated learning framework featuring locally adaptive representations based on learnable client-side embedding controls that encode each client's data-missing patterns.
These embeddings serve as reconfiguration signals that align the globally aggregated representation with each client's local context, enabling more effective use of shared information. Furthermore, the embedding controls can be algorithmically aggregated across clients with similar data-missing patterns to enhance the robustness of reconfiguration signals in adapting the global representation. Empirical results on multiple federated multimodal benchmarks with diverse data-missing patterns across clients demonstrate the efficacy of the proposed method, achieving up to 36.45% performance improvement under severe data incompleteness. The method is also supported by a theoretical analysis with an explicit performance bound that matches our empirical observations. Our source codes are provided at https://github.com/nmduonggg/PEPSY
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers4
- FedAFD: Multimodal Federated Learning via Adversarial Fusion and DistillationMin Tan, Junchao Ma, Yinfu FENG, Jiajun Ding et al.CVPR 2026 · 1 citation
- GeoEvo: Identity-Aware Potential Game with Geometric Evolution for Personalized Multimodal Federated LearningChen Wang, Yongli Hu, Huajie Jiang, Kan Guo et al.ICML 2026
- FedCDWA: Decoupled Federated Prototype Distillation with Hierarchical Wasserstein AggregationZhenshen Liu, Kai Fan, Wenjie Li, Kuan Zhang et al.ICML 2026
- FedSSM: State Space Model-based Proactive Inference for Heterogeneous Multimodal Federated LearningHengyi Ren, Yuchen Xie, Changlong Wang, Xin Li et al.ICML 2026
Builds on28
- On the Convergence of FedAvg on Non-IID DataXiang Li, Kaixuan Huang, Wenhao Yang, Shusen Wang et al.ICLR 2020 · 2,930 citations
- Personalized Federated Learning with Moreau EnvelopesCanh T. Dinh, Nguyen Hoang Tran, Tuan Dung NguyenNeurIPS 2020 · 1,542 citations
- A decoder-only foundation model for time-series forecastingAbhimanyu Das, Weihao Kong, Rajat Sen, Yichen ZhouICML 2024 · 601 citations
- Unified Training of Universal Time Series Forecasting TransformersGerald Woo, Chenghao Liu, Akshat Kumar, Caiming Xiong et al.ICML 2024 · 513 citations
- MOMENT: A Family of Open Time-series Foundation ModelsMononito Goswami, Konrad Szafer, Arjun Choudhry, Yifu Cai et al.ICML 2024 · 442 citations
Related papers
- Federated Prompt-Tuning with Heterogeneous and Incomplete Multimodal Client DataThu Hang Phung, Duong M. Nguyen, Thanh Trung Huynh, Quoc Viet Hung Nguyen et al.ICCV 2025 · 7 citations
- LAUA: Handling Missing Modalities and Unpaired Data in Multimodal Federated LearningYi Wei, Xiaokai Zhou, Shanshan Feng, Chuang Hu et al.KDD 2026
- Client-Adaptive Cross-Model Reconstruction Network for Modality-Incomplete Multimodal Federated LearningBaochen Xiong, Xiaoshan Yang, Yaguang Song, Yaowei Wang et al.ACM MM 2023 · 19 citations
- Adaptive Hyper-graph Aggregation for Modality-Agnostic Federated LearningQ. Fan, L. ShuaiCVPR 2024
- FedMobile: Enabling Knowledge Contribution-aware Multi-modal Federated Learning with Incomplete ModalitiesYi Liu, Cong Wang, Xingliang YuanWWW 2025 · 9 citations
