Federated Learning (FL) is an innovative machine learning approach that facilitates collaborative model training across distributed devices while safeguarding individual data privacy. However, traditional FL algorithms encounter challenges of model instability in practice, stemming from issues such as uneven data distribution and client heterogeneity. To address these challenges, this study proposes a FL algorithm based on Bhattacharyya regularization (FedBC). The algorithm effectively improves the consistency of models among local clients, and demonstrates promising results in handling the heterogeneity of non-independent identically distributed (non-iid) data in FL. Experimental results show that the proposed algorithm achieves excellent convergence speed and model performance, significantly enhancing the efficiency and robustness of FL while preserving data privacy.

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

Federated Learning Optimization Algorithm Based on Bhattacharyya Regularization

  • Xiaowen Duan,
  • Rui Zhao,
  • Hongtao Nie,
  • Qingguo Zhou,
  • Xin Liu

摘要

Federated Learning (FL) is an innovative machine learning approach that facilitates collaborative model training across distributed devices while safeguarding individual data privacy. However, traditional FL algorithms encounter challenges of model instability in practice, stemming from issues such as uneven data distribution and client heterogeneity. To address these challenges, this study proposes a FL algorithm based on Bhattacharyya regularization (FedBC). The algorithm effectively improves the consistency of models among local clients, and demonstrates promising results in handling the heterogeneity of non-independent identically distributed (non-iid) data in FL. Experimental results show that the proposed algorithm achieves excellent convergence speed and model performance, significantly enhancing the efficiency and robustness of FL while preserving data privacy.