Cross-silo federated learning (FL) frequently faces challenges with non-independent and identically distributed (non-IID) data across clients, leading to a decline in the accuracy of FL model training. Although personalized FL (PFL) methods have been proposed to address this issue, they generally focus on specific levels of statistical heterogeneity. Furthermore, privacy constraints frequently obscure data heterogeneity, making it difficult to select a targeted PFL. To overcome this issue, we propose FedASH, a personalized cross-silo FL framework that adapts to diverse degrees of heterogeneity with two key components. The first component is adaptive proximal directed relationships, which dynamically adjust each client’s reliance on private and shared models, effectively adapting to different levels of data heterogeneity. The second component is client weight evaluation based on gradient, which assesses corresponding weights based on the similarity between the client and server update directions, achieving more effective aggregation. We conduct experiments on multiple datasets, including Fashion-MNIST, CIFAR-10, CIFAR-100, and a Parkinson’s disease dataset, and benchmark FedASH against several other FL methods. Experimental results confirm that our framework consistently achieves superior accuracy across diverse datasets.

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

Personalized Cross-Silo Federated Learning with Adaptive Proximal Relationships and Gradient-Based Aggregation

  • Zhiqiang Lin,
  • Lingyu Liang,
  • Jiaxuan Cheng,
  • Shuai Jiang,
  • Siying Li,
  • Yong Xu

摘要

Cross-silo federated learning (FL) frequently faces challenges with non-independent and identically distributed (non-IID) data across clients, leading to a decline in the accuracy of FL model training. Although personalized FL (PFL) methods have been proposed to address this issue, they generally focus on specific levels of statistical heterogeneity. Furthermore, privacy constraints frequently obscure data heterogeneity, making it difficult to select a targeted PFL. To overcome this issue, we propose FedASH, a personalized cross-silo FL framework that adapts to diverse degrees of heterogeneity with two key components. The first component is adaptive proximal directed relationships, which dynamically adjust each client’s reliance on private and shared models, effectively adapting to different levels of data heterogeneity. The second component is client weight evaluation based on gradient, which assesses corresponding weights based on the similarity between the client and server update directions, achieving more effective aggregation. We conduct experiments on multiple datasets, including Fashion-MNIST, CIFAR-10, CIFAR-100, and a Parkinson’s disease dataset, and benchmark FedASH against several other FL methods. Experimental results confirm that our framework consistently achieves superior accuracy across diverse datasets.