Diffusion-Guided Semantic Consistency for Multimodal Heterogeneity
Authors: Jing Liu, Zhengliang Guo, Yan Wang, Xiaoguang Zhu, Yao Du, Zehua Wang, Victor C. M. Leung
Organizations: The University of British Columbia · Fudan University · Duke Kunshan University · East China Normal University · University of California, Davis · China University of Mining and Technology · SMBU · Shenzhen University
Federated learning (FL) is severely challenged by non-independent and identically distributed (non-IID) client data, a problem that degrades global model performance, especially in multimodal perception settings. Conventional methods often fail to address the underlying semantic discrepancies between clients, leading to suboptimal performance for multimedia systems requiring robust perception. To overcome this, we introduce SemanticFL, a novel framework that leverages the rich semantic representations of pre-trained diffusion models to provide privacy-preserving guidance for local training. Our approach leverages multi-layer semantic representations from a pre-trained Stable Diffusion model (including VAE-encoded latents and U-Net hierarchical features) to create a shared latent space that aligns heterogeneous clients, facilitated by an efficient client-server architecture that offloads heavy computation to the server. A unified consistency mechanism, employing cross-modal contrastive learning, further stabilizes convergence. We conduct extensive experiments on benchmarks including CIFAR-10, CIFAR-100, and TinyImageNet under diverse heterogeneity scenarios. Our results demonstrate that SemanticFL surpasses existing federated learning approaches, achieving accuracy gains of up to 5.49% over FedAvg, validating its effectiveness in learning robust representations for heterogeneous and multimodal data for perception tasks.
Figures & tables
Fig. 1 : Overview of the SemanticFL framework. More details on components and symbols are described in Sec. III .
Method
CIFAR-10
CIFAR-100
TinyImageNet
NID1 0.05
NID1 0.2
NID2
NID1 0.05
NID1 0.2
NID2
NID1 0.05
NID1 0.2
NID2
FedAvg [ 38 ]
78.27
84.65
72.60
55.97
60.08
50.56
40.41
42.84
35.15
FedProx [ 11 ]
78.42
84.59
72.81
56.27
60.21
50.29
40.20
42.16
35.62
MOON [ 12 ]
80.79
86.10
73.35
56.79
61.48
51.81
40.79
43.63
36.11
FedDisco [ 28 ]
81.35
86.63
74.72
57.34
61.79
52.05
40.83
43.92
37.69
FedCDA [ 39 ]
81.70
86.89
75.07
57.76
62.17
52.35
41.18
44.09
38.24
TABLE I : Comparison of top-1 accuracy (%) on CIFAR-10, CIFAR-100, and TinyImageNet across various non-IID scenarios.
AMDA
SFE
NID1 0.05
NID1 0.2
NID2
Average
77.91
84.12
73.08
78.37
✓
82.15
85.67
76.21
81.34
✓
81.02
85.93
74.85
80.60
✓
✓
83.76
88.94
77.12
83.27
TABLE II : Ablation study of components on CIFAR-10.
Fig. 2 : Results of client number K and local epochs E .
Fig. 3 : Effect of temperature τ on CIFAR-10 and CIFAR-100.
Fig. 4 : t-SNE visualization of learned feature representations.
Heterogeneous Federated Learning studies how to train collaborative models when clients differ in data, tasks, model architectures, or computing resources.
Multimodal Federated Learning (MFL) enables clients with heterogeneous data modalities to collaboratively train models without sharing raw data, offering a privacy-preserving framework that leverages complementary cross-modal information. However, existing methods often overlook personalized client performance and struggle with modality/task discrepancies, as well as model heterogeneity. To address these challenges, we propose FedAFD, a unified MFL framework that enhances client and server learning. On the client side, we introduce a bi-level adversarial alignment strategy to align local and global representations within and across modalities, mitigating modality and task gaps. We further design a granularity-aware fusion module to integrate global knowledge into the personalized features adaptively. On the server side, to handle model heterogeneity, we propose a similarity-guided ensemble distillation mechanism that aggregates client representations on shared public data based on feature similarity and distills the fused knowledge into the global model. Extensive experiments conducted under both IID and non-IID settings demonstrate that FedAFD achieves superior performance and efficiency for both the client and the server.
Min Tan, Junchao Ma, Yinfu Feng +6
Zhejiang Key Laboratory of Space Information Sensing and Transmission, Hangzhou Dianzi University · Laboratory of Complex Systems Modeling and Simulation, School of Computer Science and Technology, Hangzhou Dianzi University · Alibaba International Digital Commerce Group +1
Federated learning (FL) enables collaborative model training across distributed devices without sharing raw data, but applying FL to multi-modal settings introduces significant challenges. Clients typically possess heterogeneous modalities and model architectures, making it difficult to align feature spaces efficiently while preserving privacy and minimizing communication costs. To address this, we introduce CoMFed, a Communication-Efficient Multi-Modal Federated Learning framework that uses learnable projection matrices to generate compressed latent representations. A latent-space regularizer aligns these representations across clients, improving cross-modal consistency and robustness to outliers. Experiments on human activity recognition benchmarks show that CoMFed achieves competitive accuracy with minimal overhead.
Mohamed Badi, Chaouki Ben Issaid, Mehdi Bennis
Center for Wireless Communications, University of Oulu, Oulu 90014, Finland
Previous studies on federated learning (FL) often encounter performance degradation due to data heterogeneity among different clients. In light of the recent advances in multimodal large language models (MLLMs), such as GPT-4v and LLaVA, which demonstrate their exceptional proficiency in multimodal tasks, such as image captioning and multimodal question answering. We introduce a novel federated learning framework, named Multimodal Large Language Model Assisted Federated Learning (MLLM-LLaVA-FL), which employs powerful MLLMs at the server end to address the heterogeneous and long-tailed challenges. Owing to the advanced cross-modality representation capabilities and the extensive open-vocabulary prior knowledge of MLLMs, our framework is adept at harnessing the extensive, yet previously underexploited, open-source data accessible from websites and powerful server-side computational resources. Hence, the MLLM-LLaVA-FL not only enhances the performance but also avoids increasing the risk of privacy leakage and the computational burden on local devices, distinguishing it from prior methodologies. Our framework has three key stages. Initially, we conduct global visual-text pretraining of the model. This pretraining is facilitated by utilizing the extensive open-source data available online, with the assistance of MLLMs. Subsequently, the pretrained model is distributed among various clients for local training. Finally, once the locally trained models are transmitted back to the server, a global alignment is carried out under the supervision of MLLMs to further enhance the performance. Experimental evaluations on established benchmarks, show that our framework delivers promising performance in the typical scenarios with data heterogeneity and long-tail distribution across different clients in FL.
Jianyi Zhang, Hao Frank Yang, Ang Li +5
Duke University · Johns Hopkins University · University of Maryland College Park +1