Progressive feature transmission for split classification at the wireless edge
File(s)LPZGH_TWC22.pdf (2.91 MB)
Accepted version
Author(s)
Lan, Qiao
Zeng, Qunsong
Popovski, Petar
Gunduz, Deniz
Huang, Kaibin
Type
Journal Article
Abstract
We consider the scenario of inference at the wire-less edge , in which devices are connected to an edge server and ask the server to carry out remote classification, that is, classify data samples available at edge devices. This requires the edge devices to upload high-dimensional features of samples over resource-constrained wireless channels, which creates a communication bottleneck. The conventional feature pruning solution would require the device to have access to the inference model, which is not available in the current split inference scenario. To address this issue, we propose the progressive feature transmission (ProgressFTX) protocol, which minimizes the overhead by progressively transmitting features until a target confidence level is reached. A control policy is proposed to accelerate inference, comprising two key operations: importance-aware feature selection at the server and transmission-termination control . For the former, it is shown that selecting the most important features, characterized by the largest discriminant gains of the corresponding feature dimensions, achieves a sub-optimal performance. For the latter, the proposed policy is shown to exhibit a threshold structure. Specifically, the transmission is stopped when the incremental uncertainty reduction by further feature transmission is outweighed by its communication cost. The indices of the selected features and transmission decision are fed back to the device in each slot. The control policy is first derived for the tractable case of linear classification, and then extended to the more complex case of classification using a convolutional neural network . Both Gaussian and fading channels are considered. Experimental results are obtained for both a statistical data model and a real dataset. It is shown that ProgressFTX can substantially reduce the communication latency compared to conventional feature pruning and random feature transmission strategies.
Date Issued
2023-06-01
Date Acceptance
2022-11-01
Citation
IEEE Transactions on Wireless Communications, 2023, 22 (6), pp.3837-3852
ISSN
1536-1276
Publisher
Institute of Electrical and Electronics Engineers (IEEE)
Start Page
3837
End Page
3852
Journal / Book Title
IEEE Transactions on Wireless Communications
Volume
22
Issue
6
Copyright Statement
Copyright © 2022 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.
Identifier
https://ieeexplore.ieee.org/document/9955582
Publication Status
Published
Date Publish Online
2022-11-18