For federated learning systems, the Non-IID data distribution across multiple clients brings enormous challenges to the model’s generalization ability. Most existing solutions rely on simple statistical indicators or heuristic rules. Due to the lack of comprehensive capture and modeling capabilities of the potential relationships between client data distributions, they fail to effectively leverage the feature distribution relevance between clients to promote model training. In this paper, we propose a novel approach for personalized federated learning, FedCRA, equipped with an adaptive aggregation strategy. Specifically, we construct a client relevance-aware graph based on the underlying data distribution differences among clients and propose a collaborative graph-assisted personalized aggregation method. We conducted extensive experiments on multiple public datasets, validating the effectiveness and superior performance of our proposed method.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Client Relevance-Aware Adaptive Aggregation for Personalized Federated Learning

  • Zeyao Liu,
  • Zhendong Zhao,
  • Xiaojun Chen,
  • Xin Zhao,
  • Yuexin Xuan,
  • Bisheng Tang,
  • Xiaoshuang Ji

摘要

For federated learning systems, the Non-IID data distribution across multiple clients brings enormous challenges to the model’s generalization ability. Most existing solutions rely on simple statistical indicators or heuristic rules. Due to the lack of comprehensive capture and modeling capabilities of the potential relationships between client data distributions, they fail to effectively leverage the feature distribution relevance between clients to promote model training. In this paper, we propose a novel approach for personalized federated learning, FedCRA, equipped with an adaptive aggregation strategy. Specifically, we construct a client relevance-aware graph based on the underlying data distribution differences among clients and propose a collaborative graph-assisted personalized aggregation method. We conducted extensive experiments on multiple public datasets, validating the effectiveness and superior performance of our proposed method.