Optimizing Tradeoff Between Learning Speed and Cost for Federated Learning-Enabled Industrial IoT

Long Zhang,Suiyuan Wu,Haitao Xu,Qilie Liu,Choong Seon Hong,Zhu Han
DOI: https://doi.org/10.1109/jiot.2023.3330754
IF: 10.6
2023-01-01
IEEE Internet of Things Journal
Abstract:Combination of industrial Internet of Things (IIoT) and federated learning (FL) is deemed as a promising solution to realize Industry 4.0 and beyond. However, scheduling more IIoT devices engaged in FL contributes to accelerated learning speed, but resulting in increased learning cost in terms of energy consumption and model accuracy reduction. In this paper, we investigate the tradeoff between learning speed and cost in a three-layer FL-enabled IIoT system. Particularly, a weighted learning utility function is designed by capturing such a tradeoff. We aim to maximize the weighted learning utility in an FL training round by jointly optimizing the edge association as well as the allocations of resource block, computation capacity, and transmit power of IIoT device. The resulting problem is a non-convex and mixed integer optimization problem, and consequently it is difficult to solve. We thereby decompose the original problem into three subproblems, and then propose an overall alternating optimization algorithm to solve the subproblems iteratively until convergence. Via experimental results, it is demonstrated that the proposed scheme significantly improves the system-wide learning utility as compared to other baseline schemes. It is also shown that the proposed scheme can achieve the optimized tradeoff between learning speed and learning cost.
computer science, information systems,telecommunications,engineering, electrical & electronic
What problem does this paper attempt to address?