FedSiam-DA: Dual-aggregated Federated Learning Via Siamese Network for Non-Iid Data

Xin Wang,Yanhan Wang,Ming Yang,Feng li,Xiaoming Wu,Lisheng Fan,Shibo He
DOI: https://doi.org/10.1109/tmc.2024.3472898
IF: 6.075
2024-01-01
IEEE Transactions on Mobile Computing
Abstract:Federated learning (FL) is an effective mobile edge computing framework that enables multiple participants to collaboratively train intelligent models, without requiring large amounts of data transmission while protecting privacy. However, FL encounters challenges due to non-independent and identically distributed (non-IID) data from different participants. The existing methods, whether focusing on local training or global aggregation, often suffer from insufficient unilateral optimization. Achieving effective local-global collaborative optimization, particularly in the absence of additional reference models or datasets, is both crucial and challenging. To address this, we propose a novel approach: D ual- A ggregated Fed erated learning based on a triple Siam ese network ( FedSiam-DA ). This method enhances the FL algorithm on both client and server sides. On the client side, we establish a triple Siamese network incorporating a stop-gradient scheme, which leverages a contrastive learning strategy to control the update directions of local models. On the server side, we introduce a dual aggregation mechanism with dynamic weights for local updates, improving the global model's ability to assimilate personalized knowledge from local models. Extensive experiments on multiple benchmark datasets demonstrate that FedSiam-DA significantly improves model performance under non-IID data conditions compared to existing methods.
What problem does this paper attempt to address?