AMSP-UOD: When Vortex Convolution and Stochastic Perturbation Meet Underwater Object Detection

TL;DR

AMSP-UOD introduces a novel underwater object detection network combining vortex convolution and stochastic perturbation to improve accuracy and robustness in noisy, complex underwater environments.

Contribution

The paper proposes AMSP-UOD, integrating AMSP vortex convolution and a feature decoupling module to enhance detection performance under challenging underwater conditions.

Findings

Outperforms existing methods on URPC and RUOD datasets.

Improves noise immunity and detection accuracy in complex underwater scenes.

Reduces model parameters while maintaining high performance.

Abstract

In this paper, we present a novel Amplitude-Modulated Stochastic Perturbation and Vortex Convolutional Network, AMSP-UOD, designed for underwater object detection. AMSP-UOD specifically addresses the impact of non-ideal imaging factors on detection accuracy in complex underwater environments. To mitigate the influence of noise on object detection performance, we propose AMSP Vortex Convolution (AMSP-VConv) to disrupt the noise distribution, enhance feature extraction capabilities, effectively reduce parameters, and improve network robustness. We design the Feature Association Decoupling Cross Stage Partial (FAD-CSP) module, which strengthens the association of long and short range features, improving the network performance in complex underwater environments. Additionally, our sophisticated post-processing method, based on Non-Maximum Suppression (NMS) with aspect-ratio similarity thresholds, optimizes detection in dense scenes, such as waterweed and schools of fish, improving object detection accuracy. Extensive experiments on the URPC and RUOD datasets demonstrate that our method outperforms existing state-of-the-art methods in terms of accuracy and noise immunity. AMSP-UOD proposes an innovative solution with the potential for real-world applications. Our code is available at https://github.com/zhoujingchun03/AMSP-UOD.