The Journal of China Universities of Posts and Telecommunications ›› 2022, Vol. 29 ›› Issue (5): 51-61.doi: 10.19682/j.cnki.1005-8885.2022.0009

Previous Articles     Next Articles

Automatic modulation classification based on AlexNet with data augmentation

Zhang Chengchang, Xu Yu  Yang Jianpeng, Li Xiaomeng   

  • Received:2021-05-20 Revised:2021-09-12 Online:2022-10-31 Published:2022-10-28
  • Supported by:
    Teaching reform project of Chongqing University of Posts and Telecommunications


Deep learning (DL) requires massive volume of data to train the network. Insufficient training data will cause serious overfitting problem and degrade the classification accuracy. In order to solve this problem, a method for automatic modulation classification ( AMC) using AlexNet with data augmentation was proposed. Three data augmentation methods is considered, i. e. , random erasing, CutMix, and rotation. Firstly, modulated signals are converted into constellation representations. And all constellation representations are divided into training dataset and test dataset. Then, training dataset are augmented by three methods. Secondly, the optimal value of execution probability for random erasing and CutMix are determined. Simulation results show that both of them perform optimally when execution probability is 0.5. Thirdly, the performance of three data augmentation methods are evaluated. Simulation results demonstrate that all augmentation methods can improve the classification accuracy. Rotation improves the classification accuracy by 13.04% when signal noise ratio (SNR) is 2 dB. Among three methods, rotation outperforms random erasing and CutMix when SNR is greater than - 6 dB. Finally, compared with other classification algorithms, random erasing, CutMix, and rotation used in this paper achieved the performance significantly improved. It is worth mentioning that the classification accuracy can reach 90.5% with SNR at 10 dB.

Key words: automatic modulation classification (AMC)| data augmentation| random erasing| CutMix| rotation| deep learning (DL)

CLC Number: