改进通道注意力机制的时域水声信号识别网络
Underwater Time-domain Signal Recognition Network with Improved Channel Attention Mechanism
-
摘要: 为了提高时域水声信号识别网络的性能,本文在改进通道注意力机制的基础上提出了一种识别时域信号的卷积神经网络。该网络分别在原始时域信号和时域重构序列中提取特征,并在训练过程中随机丢弃输入中的数据点以防止网络训练的过拟合。同时,本文使用由多个卷积层或残差模块构造的多尺度卷积模块提取不同频率成分下的信号特征。针对时域信号特点,本文在通道注意力机制中分别引入样本特征通道能量信息,样本特征通道幅值信息以及样本特征通道与样本整体间的相关性求解特征通道权值,增强特征中的有效成分。最后,在损失函数中添加分类器权值范数的正则项,突出网络提取的有效特征。在ShipsEar和DeepShip数据库下的实验结果表明,当训练数据和测试数据具有相似分布时,本文改进的卷积神经网络可对时域目标信号进行有效识别。Abstract: In order to improve the performance of time-domain underwater acoustic signal recognition network, a convolutional neural network for time-domain signal recognition based on improved channel attention mechanism was proposed. The network extracted features from the original signal and the time-domain reconstruction sequence, and randomly discarded the input data points during the training process to prevent overfitting of the network training. Meanwhile, multi-scale convolution modules constructed by multiple convolution layers or residual blocks were used to extract signal features under different frequency components. According to the characteristics of time domain signals, this paper introduced the energy information of sample feature channels, the amplitude information of sample feature channels and the correlation between sample feature channels and the whole sample into the channel attention mechanism to solve the weights of feature channels and enhanced the effective components of features. Finally, the weight regularization term of the classifier was added to the loss function to highlight the effective features extracted by the network. Experimental results on ShipsEar and DeepShip databases showed that the proposed convolutional neural network can effectively identify the target signals in time-domain when the training data and test data have similar distribution.