Chinese Journal on Internet of Things ›› 2019, Vol. 3 ›› Issue (4): 82-90.doi: 10.11959/j.issn.2096-3750.2019.00135

• Theory and Technology • Previous Articles     Next Articles

D2D computation task offloading for efficient federated learning

Xiaoran CAI,Xiaopeng MO,Jie XU   

  1. School of Information Engineering,Guangdong University of Technology,Guangzhou 510006,China
  • Revised:2019-08-09 Online:2019-12-30 Published:2020-02-05
  • Supported by:
    The National Key R&D Program Subsidized Projects(2018YFB1800800);Research and Development Program Subsidized Projects in Key Areas of Guangdong Province(2018B030338001)

Abstract:

Federated learning is a kind of distributed machine learning technique.The factor of communication and computation resource constraints at the edge node is becoming the performance bottleneck.In particular,when different edge node has distinct computation and communication capabilities,the model training performance may degrade severely,thus necessitating the joint communication and computation optimization.To tackle this challenge,a computational task offloading scheme enabled by device-to-device (D2D) communications was proposed,in which different edge node exchanged data samples via D2D communication links to balance the processing capability and task load,in order to minimize the total time delay for machine learning model training.Simulation results show that compared to the benchmark scheme without such D2D task offloading the training speed and efficiency of federated learning has be improved significantly.

Key words: federated learning, mobile edge computing, task offloading, device-to-device communication

CLC Number: 

No Suggested Reading articles found!