Data-Limited Modulation Classification with a CVAE-Enhanced Learning Model

Xuefei Ji, Jue Wang, Ye Li, Qiang Sun, Shi Jin, Tony Q.S. Quek

Research output: Contribution to journalArticlepeer-review

3 Citations (Scopus)


The modulation classification problem can be effectively handled by modern machine learning techniques, however, the available data for training is usually limited due to costly measurements in practice. Concerning the problem of insufficient training data, we propose a conditional variational auto-encoder (CVAE)-enhanced learning model for modulation classification. Different from the conventional data augmentation approaches where the data is independently generated, we introduce a feedback unit from the classifier to the CVAE generative network, to ensure that the generated data helps improving the classification accuracy. A two-stage generating-training algorithm is proposed. Via both simulation and practical implementation on a universal software radio peripheral (USRP) platform, it is shown that the proposed method effectively improves the classification accuracy in realistic propagation environment.

Original languageEnglish
Article number9125974
Pages (from-to)2191-2195
Number of pages5
JournalIEEE Communications Letters
Issue number10
Publication statusPublished - 2020 Oct

All Science Journal Classification (ASJC) codes

  • Modelling and Simulation
  • Computer Science Applications
  • Electrical and Electronic Engineering


Dive into the research topics of 'Data-Limited Modulation Classification with a CVAE-Enhanced Learning Model'. Together they form a unique fingerprint.

Cite this