解决多模态联邦学习中缺失模态与数据异质性问题
FedRecon: Missing Modality Reconstruction in Heterogeneous Distributed Environments
- 用轻量级MVAE重建缺失模态,保持跨模态一致性
- 通过分布映射实现样本级对齐,提升数据完整性和一致性
- 冻结全局生成器防止灾难性遗忘,适合异构数据场景
多模态数据在真实场景中常因部分模态缺失而呈现非独立同分布(Non-IID)特性,导致模态异质性与数据异质性并存,给联邦学习带来根本挑战。为此,我们提出FedRecon,首个同时应对缺失模态重建与Non-IID适应的多模态联邦学习方法。该方法首先采用轻量级多模态变分自编码器(MVAE)重建缺失模态,并保持跨模态一致性;不同于传统插补方法,通过新颖的分布映射机制实现样本级对齐,确保数据一致性和完整性。此外,引入全局生成器冻结策略,防止灾难性遗忘,缓解Non-IID波动。在多个多模态数据集上的大量实验表明,FedRecon在非独立同分布条件下显著优于现有先进方法。代码将在论文接受后公开。
原文摘要 · Abstract (English)
Multimodal data are often incomplete and exhibit Non-Independent and Identically Distributed (Non-IID) characteristics in real-world scenarios. These inherent limitations lead to both modality heterogeneity through partial modality absence and data heterogeneity from distribution divergence, creating fundamental challenges for effective federated learning (FL). To address these coupled challenges, we propose FedRecon, the first method targeting simultaneous missing modality reconstruction and Non-IID adaptation in multimodal FL. Our approach first employs a lightweight Multimodal Variational Autoencoder (MVAE) to reconstruct missing modalities while preserving cross-modal consistency. Distinct from conventional imputation methods, we achieve sample-level alignment through a novel distribution mapping mechanism that guarantees both data consistency and completeness. Additionally, we introduce a strategy employing global generator freezing to prevent catastrophic forgetting, which in turn mitigates Non-IID fluctuations. Extensive evaluations on multimodal datasets demonstrate FedRecon's superior performance in modality reconstruction under Non-IID conditions, surpassing state-of-the-art methods. The code will be released upon paper acceptance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。