Automatic Modulation Recognition of Underwater Acoustic Signals Using a Two-Stream Transformer
Juan Li,Qingning Jia,Xuerong Cui,T. Aaron Gulliver,Bin Jiang,Shibao Li,Jungang Yang
DOI: https://doi.org/10.1109/jiot.2024.3367852
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:Automatic modulation recognition (AMR) of underwater acoustic (UWA) signals is incredibly challenging due to the complexity of UWA channels and the severity of ocean noise. In the presence of noise interference, single-modal features fail to fully represent the characteristics of different modulated signals. While the in-phase/quadrature (I/Q) and time-frequency maps can adequately represent the signal features in the time, frequency, and time-frequency domains, the direct integration of the two modalities is ineffective because of the variations in shape, information granularity, and noise manifestation. To address the low recognition rate caused by the above issues, we propose a two-stream transformer (TSTR) based network for AMR of UWA signals. First, the input pre-processing layer obtains the I/Q and time-frequency features from the received signals. Then, the feature capture layer extracts high-dimensional signal features in the time, frequency, and time-frequency domains. Finally, the classification layer estimates the modulation of the signals. A multi-head self-attention module with adaptive soft thresholding is used in the feature capture layer to provide noise reduction and redundant feature rejection while retaining context information. Moreover, multi-scale ghost convolution is employed to address the inability of the transformer to efficiently extract spatial characteristics from the signals. Results are presented using real UWA channels from the Watermark dataset for two different seas which show that the TSTR improves recognition by 1.2% and 5.9% over the best existing model. Further, it has better generalization capabilities and the model has a small number of parameters so the time complexity is low.
computer science, information systems,telecommunications,engineering, electrical & electronic