Distributed Learning on Mobile Devices: A New Approach to Data Mining in the Internet of Things
Xiongtao Zhang,Xiaomin Zhu,Weidong Bao,Laurence T. Yang,Ji Wang,Hui Yan,Huangke Chen
DOI: https://doi.org/10.1109/jiot.2020.3030783
IF: 10.6
2021-07-01
IEEE Internet of Things Journal
Abstract:It is well known that deep learning is one of the most important methods for data mining. With the development of the fifth-generation mobile networks (5G) and the Internet of Things (IoT), the large volume of data collected in IoTs provides a new way to improve the capability of deep learning. Due to privacy, bandwidth, and legal concerns, it is impractical to send the data to a server or the cloud. The computing power of mobile devices makes it possible to process the data. Therefore, this article focuses on training these models in mobile devices. To solve the challenges, including unreliable networks, constrained resources, and slow convergence, we let multiple mobile devices learn a shared model collaboratively. We propose a novel architecture, GREAT, where each node chooses partners to share local model parameters according to link reliability. To balance the constrained resources and learning effectiveness, an optimization problem is developed by taking the reliability threshold as the variable of controlling the resources' overhead. To implement this architecture, a dynamic control algorithm called Alpha-GossipSGD has been proposed. Its performance is evaluated by extensive experiments, which show that Alpha-GossipSGD can realize stable learning effectiveness over unreliable networks with constrained resources.
computer science, information systems,telecommunications,engineering, electrical & electronic