Client Scheduling for Multiserver Federated Learning in Industrial IoT With Unreliable Communications

Haitao Zhao, Yuhao Tan, Kun Guo, Wenchao Xia, Bo Xu, Tony Q.S. Quek

研究成果: Article同行評審

摘要

The Industrial Internet of Things (IIoT) is emerging as a promising technology that can accelerate the application of industrial intelligence to smart factories. Because of the sensitive nature of user data, federated learning (FL) which performs distributed machine learning while preserving data privacy, is leveraged to meet the accuracy and privacy requirements of IIoT end devices/clients. However, the unreliable communications in IIoT may result in possible single-point failures in the typical single-server FL framework, thereby negatively affecting the training efficiency. In this article, we study on the client scheduling problem in a multiserver FL framework for the communication reliability and training efficiency improvement. Specifically, we focus on a semi-decentralized FL (SD-FL) framework, where edge servers and clients collaborate to train a shared global model through unreliable intracluster model aggregation and intercluster model consensus because of the model transmission error in client-server and server-server communication. Then, a client-server association optimization problem is formulated, with the objective of minimizing the global training loss. Resorting to the convergence analysis of SD-FL, the original problem is simplified and transformed into an integer nonlinear programming problem to guide us to design a high-efficiency client scheduling scheme. Finally, experimental results show that the proposed scheme significantly outperforms the baselines in terms of the test accuracy and training loss.

原文English
頁(從 - 到)16478-16490
頁數13
期刊IEEE Internet of Things Journal
11
發行號9
DOIs
出版狀態Published - 2024 5月 1

All Science Journal Classification (ASJC) codes

  • 訊號處理
  • 資訊系統
  • 硬體和架構
  • 電腦科學應用
  • 電腦網路與通信

指紋

深入研究「Client Scheduling for Multiserver Federated Learning in Industrial IoT With Unreliable Communications」主題。共同形成了獨特的指紋。

引用此