Few-Shot Modulation Classification Method Based on Feature Dimension Reduction and Pseudo-Label Training

Abstract

In modulation classification domain, handcrafted feature based method can fit well from a few labeled samples, while deep learning based method require a large amount of samples to achieve the superior classification performance. In order to improve the modulation classification accuracy under the constraint of limited labeled samples, this paper proposes a few-shot modulation classification method based on feature dimension reduction and pseudo-label training (FDRPLT), which combines handcrafted feature based method with deep learning based method. First, an optimal low-dimensional feature subset is created by the combination of the handcrafted features and autoencoder-extracted features post-processed by a feature selection algorithm. Then, a fully connected network (FCN), trained on a small number of labeled signals, is designed for the automated annotating, where unlabeled samples can be annotated and used for the later convolution neural network (CNN) training. The simulation results show that the classification accuracy of eight kinds of modulation types can reach to 98.3% when the SNR is 20dB.