Feature Fusion Based on Graph Convolution Network for Modulation Classification in Underwater Communication

Abstract

Automatic modulation classification (AMC) of underwater acoustic communication signals is of great significance in national defense and marine military. Accurate modulation classification methods can make great contributions to accurately grasping the parameters and characteristics of enemy communication systems. While a poor underwater acoustic channel makes it difficult to classify the modulation types correctly. Feature extraction and deep learning methods have proven to be effective methods for the modulation classification of underwater acoustic communication signals, but their performance is still limited by the complex underwater communication environment. Graph convolution networks (GCN) can learn the graph structured information of the data, making it an effective method for processing structured data. To improve the stability and robustness of AMC in underwater channels, we combined the feature extraction and deep learning methods by fusing the multi-domain features and deep features using GCN. The proposed method takes the relationships among the different multi-domain features and deep features into account. Firstly, a feature graph was built using the properties of the features. Secondly, multi-domain features were extracted from the received signals and deep features were extracted from the signals using a deep neural network. Thirdly, we constructed the input of GCN using these features and the graph. Then, the multi-domain features and deep features were fused by the GCN. Finally, we classified the modulation types using the output of GCN by way of a softmax layer. We conducted the experiments on a simulated dataset and a real-world dataset, respectively. The results show that the AMC based on GCN can achieve a significant improvement in performance compared to the current state-of-the-art methods. Our approach is robust in underwater acoustic channels.