MROD-CNN: A Miniature Re-parameterized One-Dimension Convolutional Neural Network for Underwater Source Localization

Yuhong Zheng,Jiayi Xu,Xingfei Li,Zhenwang Li
DOI: https://doi.org/10.1145/3641584.3641695
2024-01-01
Abstract:In the ocean where computing resources are scarce, underwater source localization is very dependent on the prediction speed of the algorithm. At present, most of the deep learning neural networks used for underwater source passive localization have the problem of slow inference speed, which makes it difficult to directly use them in unmanned underwater vehicle. To this end, we developed a miniature re-parameterized one-dimension convolutional neural network (MROD-CNN) for underwater source localization. Unlike traditional networks, the number of channels of MROD-CNN decreases layer by layer. The first layer of the network is convolved by the large convolution kernel. Several single-channel re-parameterized structures are connected to form a slender “tail”, which plays a significant role in improving the accuracy of the model. MROD-CNN uses the reparameterization technique to fit a large number of signal data with different signal-to-noise ratios well. Experiments show that the proposed model can significantly improve the localization accuracy and speed of KRAKEN acoustic tool simulation data, verifying the importance of reparameterization technique for underwater source localization tasks.
What problem does this paper attempt to address?