A Hierarchical Classification Head Based Convolutional Gated Deep Neural Network for Automatic Modulation Classification

Automatic modulation classification (AMC) identifies a received signal’s modulation scheme without prior knowledge of the intercepted signal, which enables significant applications in both the military and civilian domains. Inspired by the great success of deep learning (DL), lots of neural networks are introduced into AMC. To further improve classification performance, various complementary cues including in-phase/quadrature (I/Q), amplitude/phase (A/P), constellation, and other formats are used together to enhance the discrimination of the DL model, where only outputs of the last layer are used. In this paper, we find that different layers’ outputs in the DL model are also complementary to each other. As a result, a hierarchical classification head based convolutional gated deep neural network (HCGDNN) is proposed by utilizing different layers’ output, which only uses the I/Q cue. The proposed HCGDNN consists of three groups of convolutional neural networks (CNN) blocks, two groups of bidirectional gated recurrent units (BiGRU), and a hierarchical classification head. Compared to the long short-term memory (LSTM), the BiGRU has a smaller computational complexity and also releases the gradient dispersion and explosion in the training phase. With the help of the hierarchical classification head, three groups of modulation predictions are made for a received I/Q signal. After that, a novel nonlinear optimization fusion method is derived to generate fusion weights to fuse different groups, then a final classification decision is made. Compared to AMC methods using various cues, the proposed HCGDNN only uses I/Q cue and has low computational overhead. Numerical results suggest that the newly developed HCGDNN achieves superior performance on the public benchmark.To help other researchers, the source code will be uploaded to the github as long as the paper is published.