XIAORAN CAI, XIAOPENG MO, JIE XU. D2D computation task offloading for efficient federated learning. [J]. Chinese journal on internet of things, 2019, 3(4): 82-90.
DOI:
XIAORAN CAI, XIAOPENG MO, JIE XU. D2D computation task offloading for efficient federated learning. [J]. Chinese journal on internet of things, 2019, 3(4): 82-90. DOI: 10.11959/j.issn.2096-3750.2019.00135.
D2D computation task offloading for efficient federated learning
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.
关键词
联合学习移动边缘计算任务卸载D2D通信
Keywords
federated learningmobile edge computingtask offloadingdevice-to-device communication
references
MAO Y Y, YOU C S, ZHANG J ,et al. A survey on mobile edge computing:the communication perspective[J]. IEEE Communications Surveys & Tutorials, 2017,19(4): 2322-2358.
CAO X W, WANG F, XU J ,et al. Joint computation and communication cooperation for energy-efficient mobile edge computing[J]. IEEE Internet of Things Journal, 2019,6(3): 4188-4200.
WANG F, XU J, WANG X ,et al. Joint offloading and computing optimization in wireless powered mobile-edge computing systems[J]. IEEE Transactions on Wireless Communications, 2018,17(3): 1784-1797.
LETAIEF B K, CHEN W, SHI Y M ,et al. The roadmap to 6G:AI empowered wireless networks[J]. IEEE Communications Magazine, 2019,57(8): 84-90.
ELGAZAR A, HARRAS K, AAZAM M ,et al. Towards intelligent edge storage management:determining and predicting mobile file popularity[C]// IEEE International Conference on Mobile Cloud Computing,Services,and Engineering. IEEE, 2018.
KONECNY J, BRENDAN M, RAMAGE D ,et al. Federated machine learning:concept and applications[J]. ACM Transactions on Intelligent Systems and Technology, 2019,10(2):12.
ZHANG J L, TU H D, REN Y J ,et al. An adaptive synchronous parallel strategy for distributed machine learning[J]. IEEE Access, 2018(6): 19222-19230.
SPRAGUE M R, JALALIRAD A, SCAVUZZO M ,et al. Asynchronous federated learning for geospatial applications[C]// Joint European Conference on Machine Learning and Knowledge Discovery in Databases. Springer, 2018.
CHEN J M, MONGA R, BENGIO S ,et al. Revisiting distributed synchronous SGD[C]// International Conference on Learning Representations Workshop Track. Arxiv, 2016.
WANG S Q, TUOR T, SALONIDIS T ,et al. Adaptive federated learning in resource constrained edge computing systems[J]. IEEE Journal on Selected Areas in Communications, 2019,37(6): 1205-1221.
WEN W, XU C, YAN F ,et al. TernGrad:ternary gradients to reduce communication in distributed deep learning[C]// Advances in Neural Information Processing Systems. IEEE, 2017.
LI X Y, ZHU G X, GONG Y ,et al. Wirelessly powered data aggregation for IoT via over-the-air function computation:beamforming and power control[J]. IEEE Transactions on Wireless Communications, 2019,18(7): 3437-3452.
HU Y C, PATEL M, SABELLA D ,et al. Mobile edge computing—a key technology towards 5G[J]. ETSI White Paper, 2015,11(11): 1-16.
XU J, CHEN L X, LIU K ,et al. Designing security-aware incentives for computation offloading via device-to-device communication[J]. IEEE Transactions on Wireless Communications, 2018,17(9): 6053-6066.
LEE S J, TCHA Y, SEO S Y ,et al. Efficient use of multicast and unicast channels for multicast service transmission[J]. IEEE Transactions on Communications, 2011,59(5): 1264-1267.
HUNGER R . Floating point operations in matrix-vector calculus[M]. Munich: Institute for Circuit Theory and Signal ProcessingPress, 2005.
LECUN Y, BOTTOU L, BENGIO Y ,et al. Gradient-based learning applied to document recognition[J]. IEEE, 1998,86(11): 2278-2324.