Signal modulation classification based on the transformer network

In this work, the Transformer Network (TRN) is applied to the automatic modulation classification (AMC) problem for the first time. Different from the other deep networks, the TRN can incorporate the global information of each sample sequence and exploit the information that is semantically relevant for classification. In order to illustrate the performance of the proposed model, it is compared with four other deep models and two traditional methods. Simulation results show that the proposed one has a higher classification accuracy especially at low signal to noise ratios (SNRs), and the number of training parameters of the proposed model is less than those of the other deep models, which makes it more suitable for practical applications.