An Adaptive Device-Edge Co-Inference Framework Based on Soft Actor-Critic

• Main Authors: Niu, Tao, Teng, Yinglei, Han, Zhu, Zou, Panpan
• Format: Conference Proceeding
• Language: English
• Subjects: Adaptation models; Computational modeling; Deep learning; Deep Neural Network (DNN); Edge Computing; Image edge detection; Inference; Internet of Things (IoTs); Neural networks; Performance evaluation; Reinforcement Learning; Wireless communication
• ISSN: 1558-2612
• DOI: 10.1109/WCNC51071.2022.9771550

Recently, the applications of deep neural network (DNN) have been very prominent in many fields due to its superior feature extraction performance. However, the high-dimension parameter model and large-scale mathematical calculation restrict the execution efficiency, especially for the Internet of Things (IoT) devices. Different from the previous cloud/edge-only pattern that brings significant pressure for uplink communication and device-only fashion that undertakes unaffordable calculation strength, we highlight the collaborative computation between the device and edge for DNN models, which can achieve a good balance between the communication load and execution accuracy. Specifically, a systematic on-demand co-inference framework is proposed to exploit the multi-branch structure, in which the pre-trained Alexnet is right-sized through early-exit and partitioned at an intermediate DNN layer. The integer quantization is enforced to further compress transmission bits. As a result, we establish a new Deep Reinforcement Learning (DRL) optimizer-Soft Actor Critic for discrete (SAC-d), which generates the exit point, partition point, and compressing bits by soft policy iterations. Based on the latency and accuracy aware reward design, such an optimizer can well adapt to the complex environment like dynamic wireless channel and arbitrary CPU processing, and is capable of supporting the 5G URLLC. Real-world experiment on Raspberry Pi 4 and PC shows the effective performance of the proposed solution.