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

[thumbnail of FINAL VERSION]
FINAL_VERSION.pdf - Accepted Version
Available under License Creative Commons Attribution-NonCommercial.

Download (862kB)


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%.

Item Type:
Journal Article
Journal or Publication Title:
IEEE Transactions on Industrial Informatics
Additional Information:
©2021 IEEE. Personal use of this material is permitted. However, permission to reprint/republish this material for advertising or promotional purposes or for creating new collective works for resale or redistribution to servers or lists, or to reuse any copyrighted component of this work in other works must be obtained from the IEEE.
Uncontrolled Keywords:
?? computational modelingfederated learningindustrial internet of thingsoptimizationresource allocationresource managementserverstrainingwireless communicationeconomic and social effectsnp-hardprocessingautomated manufactureconvergence timedistributed device ??
ID Code:
Deposited By:
Deposited On:
17 May 2021 13:00
Last Modified:
10 Jan 2024 00:31