FedCLCC: A personalized federated learning algorithm for edge cloud collaboration based on contrastive learning and conditional computing

Abstract

Federated learning (FL) is a distributed machine learning paradigm for edge cloud computing. FL can facilitate data-driven decision-making in tactical scenarios, effectively addressing both data volume and infrastructure challenges in edge environments. However, the diversity of clients in edge cloud computing presents significant challenges for FL. Personalized federated learning (pFL) received considerable attention in recent years. One example of pFL involves exploiting the global and local information in the local model. Current pFL algorithms experience limitations such as slow convergence speed, catastrophic forgetting, and poor performance in complex tasks, which still have significant shortcomings compared to the centralized learning. To achieve high pFL performance, we propose FedCLCC: Federated Contrastive Learning and Conditional Computing. The core of FedCLCC is the use of contrastive learning and conditional computing. Contrastive learning determines the feature representation similarity to adjust the local model. Conditional computing separates the global and local information and feeds it to their corresponding heads for global and local handling. Our comprehensive experiments demonstrate that FedCLCC outperforms other state-of-the-art FL algorithms.

Keywords

• Federated learning
• Statistical heterogeneity
• Personalized model
• Conditional computing
• Contrastive learning