Resource Allocation for Latency-aware Federated Learning in Industrial Internet-of-Things

Gao, W. and Zhao, Z. and Min, G. and Ni, Q. and Jiang, Y. (2021) Resource Allocation for Latency-aware Federated Learning in Industrial Internet-of-Things. IEEE Transactions on Industrial Informatics, 17 (12). pp. 8505-8513. ISSN 1551-3203

Text (FINAL VERSION)
FINAL_VERSION.pdf - Accepted Version
Available under License Creative Commons Attribution-NonCommercial.
Download (862kB)

Abstract

Federated Learning (FL) has been employed for tremendous privacy-sensitive applications, where distributed devices collaboratively train a global model. In Industrial Internet-of-Things (IIoT) systems, training latency is the key performance metric as the automated manufacture usually requires timely processing. The existing works increase the number of effective devices to accelerate the training. However, devices in IIoT systems are usually densely deployed, increasing the number of clients can potentially cause serious interference and prolonged training latency. In this paper, we propose RaFed, a resource allocation scheme for FL. We formulate the problem of reducing training latency as an optimization problem, which is proved to be NP-hard. We propose a heuristic to select appropriate devices to achieve a good trade-off between the interference and convergence time. We conduct experiments using an RGB-D dataset in an IIoT system. The results show that compared to the state-of-the-art works, Rafed significantly reduces the latency by 29.9%.