计算机应用 ›› 2021, Vol. 41 ›› Issue (9): 2639-2645.DOI: 10.11772/j.issn.1001-9081.2020111734

所属专题: 先进计算

• 先进计算 • 上一篇    下一篇

移动边缘计算环境中面向机器学习的计算迁移策略

郭棉1, 张锦友2   

  1. 1. 广东技术师范大学 电子与信息学院, 广州 510665;
    2. 广东石油化工学院 电子信息工程学院, 广东 茂名 525000
  • 收稿日期:2020-11-09 修回日期:2021-01-19 出版日期:2021-09-10 发布日期:2021-05-08
  • 通讯作者: 郭棉
  • 作者简介:郭棉(1979-),女,广东茂名人,副教授,博士,CCF会员,主要研究方向:边缘计算、云计算、网络服务质量、深度强化学习;张锦友(1999-),男,广东惠州人,主要研究方向:边缘计算、深度学习。
  • 基金资助:
    国家自然科学基金青年科学基金资助项目(61901128);广东省攀登计划项目(pdjh2020b0392)。

Computation offloading policy for machine learning in mobile edge computing environments

GUO Mian1, ZHANG Jinyou2   

  1. 1. School of Electronics and Information, Guangdong Polytechnic Normal University, Guangzhou Guangdong 510665, China;
    2. College of Electronic and Information Engineering, Guangdong University of Petrochemical Technology, Maoming Guangdong 525000, China
  • Received:2020-11-09 Revised:2021-01-19 Online:2021-09-10 Published:2021-05-08
  • Supported by:
    This work is partially supported by the Youth Program of National Natural Science Foundation of China (61901128), the "Climbing" Program of Guangdong Province (pdjh2020b0392).

摘要: 针对物联网(IoT)数据源的多样化、数据的非独立同分布性、边缘设备计算能力和能耗的异构性,提出一种集中学习和联邦学习共存的移动边缘计算(MEC)网络计算迁移策略。首先,建立与集中学习、联邦学习都关联的计算迁移系统模型,考虑了集中学习、联邦学习模型产生的网络传输延迟、计算延迟以及能耗;然后,以系统平均延迟为优化目标、以能耗和基于机器学习准确率的训练次数为限制条件构建面向机器学习的计算迁移优化模型。接着对所述计算迁移进行了博弈分析,并基于分析结果提出一种能量约束的延迟贪婪(ECDG)算法,通过延迟贪婪决策和能量约束决策更新二阶优化来获取模型的优化解。与集中式贪婪算法和面向联邦学习的客户选择(FedCS)算法相比,ECDG算法的平均学习延迟最低,约为集中式贪婪算法的1/10,为FedCS算法的1/5。实验结果表明,ECDG算法能通过计算迁移自动为数据源选择最优的机器学习模型,从而有效降低机器学习的延迟,提高边缘设备的能效,满足IoT应用的服务质量(QoS)要求。

关键词: 移动边缘计算, 计算迁移, 机器学习, 联邦学习, 延迟敏感

Abstract: Concerning the challenges of the diversity of data sources, non-independent and identical distribution of data and the heterogeneity of both computing capabilities and energy consumption of edge devices in Internet of Things (IoT), a computation offloading policy in Mobile Edge Computing (MEC) network that deploys both centralized learning and federated learning was proposed. Firstly, a system model of computation offloading related to both centralized learning and federated learning was built, considering the network transmission delay, computation delay and energy consumption of centralized learning and federated learning models. Then, with the system delay minimization as optimization object, considering the constraints of energy consumption and the training times based on machine learning accuracy, a computation offloading optimization model for machine learning was constructed. After that, the game for this computation offloading was formulated and analyzed. Based on the analysis results, an Energy-Constrained Delay-Greedy (ECDG) algorithm was proposed, which found the optimal solutions for the model via a two-stage policy of greedy decision and energy-constrained decision updating. Compared to the centralized-greedy and Federated Learning with Client Selection (FedCS) algorithms, ECDG algorithm has the lowest average learning delay, which is 1/10 of that in the centralized-greedy algorithm, and 1/5 of that in the FedCS algorithm. The experimental results show that, ECDG algorithms can automatically select the optimal machine learning models by computation offloading so that it can efficiently reduce the average machine learning delay, improve the energy efficiency of edge devices and satisfy the Quality of Service (QoS) requirements of IoT applications.

Key words: Mobile Edge Computing (MEC), computation offloading, machine learning, federated learning, delay sensitive

中图分类号: