SemiAMR: Semi-Supervised Automatic Modulation Recognition With Corrected Pseudo-Label and Consistency Regularization
Yuanpu Guo, Dan Zhong, Haixin Sun, Zhiye Jiang, Linmei Ye, Zhenmiao Deng, Hui Liu
Abstract
In modern communication systems, automatic modulation recognition (AMR) has broad application prospects but a large amount of training data is usually required. Due to the high cost of adding annotations to signals, it is of great significance to study AMR under the condition of few labeled samples. In this paper, an end-to-end semi-supervised radio signal recognition algorithm is proposed. In view of the poor anti-noise performance of the original IQ data, short-time Fourier transform (STFT) is introduced to mine the stable modulation characteristics. In order to overcome the noise problem caused by pseudo-labels, a mixed sample data augmentation (MSDA) method PFMix is proposed, and the consistency constraint of the model disturbance based on mean teacher (MT) framework and data disturbance based on PFMix is designed. Meanwhile, in order to provide pseudo-labels with high confidence, entropy regularization is constructed by taking the prediction variance of MT framework as correction factor. The experimental results using the open source RML2016.10A dataset show the superior performance of the proposed method. The average recognition accuracy with 1% labeled data is 88.82% at 0 dB, which is a significant improvement compared with state of the art in semi-supervised AMR. Code available at <uri xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">https://github.com/zd2021fj/SemiAMR/tree/master</uri>