Wireless data acquisition for edge learning: data-importance aware retransmission

Liu, D. , Zhu, G., Zeng, Q., Zhang, J. and Huang, K. (2021) Wireless data acquisition for edge learning: data-importance aware retransmission. IEEE Transactions on Wireless Communications, 20(1), pp. 406-420. (doi: 10.1109/TWC.2020.3024980)

[img] Text
320649.pdf - Accepted Version

5MB

Abstract

By deploying machine-learning algorithms at the network edge, edge learning can leverage the enormous real-time data generated by billions of mobile devices to train AI models, which enable intelligent mobile applications. In this emerging research area, one key direction is to efficiently utilize radio resources for wireless data acquisition to minimize the latency of executing a learning task at an edge server. Along this direction, we consider the specific problem of retransmission decision in each communication round to ensure both reliability and quantity of those training data for accelerating model convergence. To solve the problem, a new retransmission protocol called data-importance aware automatic-repeat-request (importance ARQ) is proposed. Unlike the classic ARQ focusing merely on reliability, importance ARQ selectively retransmits a data sample based on its uncertainty which helps learning and can be measured using the model under training. Underpinning the proposed protocol is a derived elegant communication-learning relation between two corresponding metrics, i.e., signal-to-noise ratio (SNR) and data uncertainty. This relation facilitates the design of a simple threshold based policy for importance ARQ. The policy is first derived based on the classic classifier model of support vector machine (SVM), where the uncertainty of a data sample is measured by its distance to the decision boundary. The policy is then extended to the more complex model of convolutional neural networks (CNN) where data uncertainty is measured by entropy. Extensive experiments have been conducted for both the SVM and CNN using real datasets with balanced and imbalanced distributions. Experimental results demonstrate that importance ARQ effectively copes with channel fading and noise in wireless data acquisition to achieve faster model convergence than the conventional channel-aware ARQ. The gain is more significant when the dataset is imbalanced.

Item Type:Articles
Status:Published
Refereed:Yes
Glasgow Author(s) Enlighten ID:Liu, Dr Dongzhu
Authors: Liu, D., Zhu, G., Zeng, Q., Zhang, J., and Huang, K.
College/School:College of Science and Engineering > School of Computing Science
Journal Name:IEEE Transactions on Wireless Communications
Publisher:IEEE
ISSN:1536-1276
ISSN (Online):1558-2248
Published Online:25 September 2020
Copyright Holders:Copyright © 2021 IEEE
First Published:First published in IEEE Transactions on Wireless Communications 20(1):406-420
Publisher Policy:Reproduced in accordance with the publisher copyright policy

University Staff: Request a correction | Enlighten Editors: Update this record