用于水声目标识别的注意力机制特征增强
A feature attention enhancement network for underwater acoustic classification
-
摘要: 提出了一种基于注意力机制的特征增强网络(FEAN)用于水声目标的识别, 该方法以水声目标辐射噪声信号的线谱信息和调制信息为基础, 采用可学习的注意力模块, 一方面依照后端的分类任务, 对特征进行线性的自适应滤波操作, 保留对分类任务有效的信息, 并且滤除干扰; 另一方面利用线谱信息在时域上的相干性, 调制信息在频域上的相关性来优化时频谱图特征。在DeepShip和Shipsear公开数据集上的实验结果表明, 相比于以传统的短时傅里叶变换(STFT)和Fbank特征作为输入的识别模型, FEAN在两个数据集上分别获得了4.5%和1.2%的识别准确率提升。Abstract: A feature attention enhancement network (FEAN) is proposed for underwater acoustic classification. Based on the line spectrum information and modulation information of radiated noise signals from underwater acoustic targets, the FEAN employs a learnable attention module that performs linear adaptive filtering on features according to the classification task, retaining information relevant to the classification task while filtering out interference. Additionally, it optimizes time-frequency spectrogram features by utilizing the coherence of line spectrum information in the time domain and the correlation of modulation information in the frequency domain. The experimental results on the DeepShip and Shipsear public datasets show that compared with the recognition models with the traditional short-time Fourier transform (STFT) and Fbank features as input, the FEAN has achieved a 4.5% and 1.2% improvement in recognition accuracy on the two datasets, respectively.