Multi-Task Learning for Generalized Automatic Modulation Classification Under Non-Gaussian Noise with Varying SNR Conditions

Yu Wang, Guan Gui, Tomoaki Ohtsuki, Fumiyuki Adachi

Research output: Contribution to journalArticlepeer-review

Abstract

Automatic modulation classification (AMC) is a critical algorithm for the identification of modulation types so as to enable more accurate demodulation in the non-cooperative scenarios. Deep learning (DL)-based AMC is believed as one of the most promising methods with great classification accuracy. However, the conventional CNN-based methods are lack of generality capabilities under time-varying signal-to-noise ratio (SNR) conditions, because these methods are merely trained on specific datasets and can only work under the corresponding condition. In this paper, a novel multi-task learning (MTL)-based generalized AMC method is proposed, and a more realistic scenario is considered, including white non-Gaussian noise and synchronization error. Its generalization capability stems from knowledge-sharing-based MTL in varying noise scenarios. In detail, multiple CNN models with the same structure are trained for multiple SNR conditions, but they share their knowledge (e.g. model weight) with each other. Thus, MTL can extract the general features from datasets in different noise scenarios. Simulation results show that our proposed architecture can achieve higher robustness and generalization than the conventional ones.

Original languageEnglish
JournalIEEE Transactions on Wireless Communications
DOIs
Publication statusAccepted/In press - 2021

Keywords

  • Automatic modulation classification (AMC)
  • Computational modeling
  • Estimation
  • Feature extraction
  • Modulation
  • Signal to noise ratio
  • Training
  • Wireless communication
  • convolutional neural network (CNN)
  • generalization
  • multi-task learning (MTL)
  • white non-Gaussian noise

ASJC Scopus subject areas

  • Computer Science Applications
  • Electrical and Electronic Engineering
  • Applied Mathematics

Fingerprint Dive into the research topics of 'Multi-Task Learning for Generalized Automatic Modulation Classification Under Non-Gaussian Noise with Varying SNR Conditions'. Together they form a unique fingerprint.

Cite this