中国邮电高校学报(英文) ›› 2022, Vol. 29 ›› Issue (5): 51-61.doi: 10.19682/j.cnki.1005-8885.2022.0009

• Artificial Intelligence • 上一篇    下一篇

Automatic modulation classification based on AlexNet with data augmentation

张承畅,徐余,杨建鹏,李晓梦   

  1. 重庆邮电大学
  • 收稿日期:2021-05-20 修回日期:2021-09-12 出版日期:2022-10-31 发布日期:2022-10-28
  • 通讯作者: 徐余 E-mail:2105173658@qq.com
  • 基金资助:
    重庆邮电大学教学改革项目;重庆市教委科学技术研究项目;教育部产学合作协同育人项目

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.

关键词: automatic modulation classification (AMC)| data augmentation| random erasing| CutMix| rotation| deep learning (DL)

Abstract:

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)

中图分类号: