Journal on Communications ›› 2022, Vol. 43 ›› Issue (8): 65-77.doi: 10.11959/j.issn.1000-436x.2022126

• Papers • Previous Articles     Next Articles

Federated learning resource management for energy-constrained industrial IoT devices

Shaoshuai FAN, Jianbo WU, Hui TIAN   

  1. State Key Laboratory of Networking and Switching Technology, Beijing University of Posts and Telecommunications, Beijing 100876, China
  • Revised:2022-06-02 Online:2022-08-25 Published:2022-08-01
  • Supported by:
    The National Key Research and Development Program of China(2020YFB1807800)

Abstract:

Given the impact of limited wireless resources, a dynamic multi-dimensional resource joint management algorithm was proposed, which intended to tackle the problem of device failure and training interruption caused by the limited battery energy in federated learning network in industrial Internet of things (IIoT).Firstly, the optimization problem was decoupled into battery energy allocation, equipment resource allocation and communication resource allocation sub-problems which were interdependent with the goal of maximizing the fixed-time learning accuracy.Then, the equipment transmission and computing resource allocation problem were solved based on particle swarm optimization algorithm under the given energy budget.Thereafter, the resource block iterative matching algorithm was proposed to optimize the optimal communication resource allocation strategy.Finally, the online energy allocation algorithm was proposed to adjust the energy budget allocation.Simulation results validate the proposed algorithm can improve the model learning accuracy compared with other benchmarks, and can perform better in energy shortage scenarios.

Key words: federated learning, battery-powered, resource allocation, learning efficiency

CLC Number: 

No Suggested Reading articles found!