FedSEMA: Similarity-Aware for Representation Consistency in Federated Contrastive Learning

Yanbing Zhou,Yingbo Wu,Jiyang Zhou,Xin Zheng
DOI: https://doi.org/10.1007/s10489-023-05193-0
IF: 5.3
2023-01-01
Applied Intelligence
Abstract:Contrastive learning has emerged as a promising method for addressing the non-independent and identically distributed (non-IID) problem in federated learning. Existing methods make the limited assumption that all clients are consistently available in each communication round. However, in the more constrained scenario of cross-device settings, resource-constrained clients intermittently participate in the training process, resulting in the updates of the local models being delayed and thus reducing the representation consistency in the federated contrastive loss. In this paper, we analyse the superiority of the federated contrastive loss over traditional methods in terms of addressing the non-IID problem: the federated contrastive loss not only corrects the local objective towards the global objective but also debiases the local updates. To address the representation inconsistency issue, we propose a novel method called Federated Similarity-aware Exponential Moving Average update (FedSEMA), which incorporates a similarity-aware function into the EMA update process. First, FedSEMA adaptively facilitates the underlying pairwise collaborations between clients to generate personalized knowledge based on the similarity-aware EMA update procedure. Second, FedSEMA effectively exploits personalized knowledge to update the delayed local models, maintaining representation consistency to maximally benefit representation learning. Our extensive experiments conducted on various datasets under different non-IID settings demonstrate that FedSEMA is an effective and robust method for tackling the representation inconsistency issue.
What problem does this paper attempt to address?