A real-time constellation image classification method of wireless communication signals based on the lightweight network MobileViT

Qinghe Zheng,Sergio Saponara,Xinyu Tian,Zhiguo Yu,Abdussalam Elhanashi,Rui Yu
DOI: https://doi.org/10.1007/s11571-023-10015-7
Abstract:Automatic modulation classification (AMC) is a challenging topic in the development of cognitive radio, which can sense and learn surrounding electromagnetic environments and help to make corresponding decisions. In this paper, we propose to complete the real-time AMC through constructing a lightweight neural network MobileViT driven by the clustered constellation images. Firstly, the clustered constellation images are transformed from I/Q sequences to help extract robust and discriminative features. Then the lightweight neural network called MobileViT is developed for the real-time constellation image classification. Experimental results on the public dataset RadioML 2016.10a with edge computing platform demonstrate the superiority and efficiency of MobileViT. Furthermore, the extensive ablation tests prove the robustness of the proposed method to the learning rate and batch size. To the best of our knowledge, this is the first attempt to deploy the deep learning model to complete the real-time classification of modulation schemes of received signals at the edge.
What problem does this paper attempt to address?