物联网学报 ›› 2019, Vol. 3 ›› Issue (4): 82-90.doi: 10.11959/j.issn.2096-3750.2019.00135

• 理论与技术 • 上一篇    下一篇

面向联合学习的D2D计算任务卸载

蔡晓然,莫小鹏,许杰   

  1. 广东工业大学信息工程学院,广东 广州 510006
  • 修回日期:2019-08-09 出版日期:2019-12-30 发布日期:2020-02-05
  • 作者简介:蔡晓然(1996- ),女,广东阳江人,广东工业大学硕士生,主要研究方向为无线通信、机器学习和移动边缘计算|莫小鹏(1996- ),男,广东湛江人,广东工业大学硕士生,主要研究方向为无线通信、机器学习、无人机通信和移动边缘计算|许杰(1985- ),男,四川内江人,广东工业大学教授、博士生导师,主要研究方向为无线通信、机器学习、无线能量传输、无人机通信以及移动边缘计算
  • 基金资助:
    国家重点研发计划资助项目(2018YFB1800800);广东省重点领域研发计划资助项目(2018B030338001)

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)

摘要:

联合学习是一种分布式机器学习,边缘节点的计算和通信资源受限等因素是限制其性能优化的瓶颈。当边缘节点的计算和通信能力异构时,需要对通信和计算进行联合优化。提出了一种面向联合学习的D2D计算任务卸载方案,不同边缘节点通过D2D通信交换数据样本,平衡节点的处理能力和任务负载,以最小化联合学习模型训练过程的总时延。仿真结果表明,所提出的D2D计算任务卸载方案能显著提高联合学习的模型训练速度和效率。

关键词: 联合学习, 移动边缘计算, 任务卸载, D2D通信

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

中图分类号: 

No Suggested Reading articles found!