Multi-scale detection of underwater objects using attention mechanisms and normalized Wasserstein distance loss

Abstract

Underwater imaging faces challenges such as light attenuation, scattering, and water turbidity, which degrade image quality and hinder accurate organism recognition. The detecting underwater objects dataset, with resolutions from 586 x 482 to 3840 x 2160 pixels, highlights significant object scale variation, including a high proportion of small objects (27.38%). This study introduces the underwater attention-parallel residual bi-fusion feature pyramid network model, which improves detection accuracy for small- and medium-sized objects in complex underwater environments. The proposed model incorporates a spatial pyramid pooling module with attention mechanisms to enhance multi-scale feature representation and integrates the normalized Wasserstein distance into the loss function for better detection flexibility. Experimental results demonstrate that the model outperforms state-of-the-art methods, achieving a mean average precision at intersection over union threshold of 0.5 of 88.8% and a mean average precision at intersection over union threshold range of 0.5-0.95 of 68.3%, representing a 2.5-9% improvement over baseline models. Furthermore, the model achieved a precision of 85.5%, recall of 82.9%, and an F1-score of 0.8417. These results highlight the model's robustness and effectiveness, offering significant contributions to underwater biodiversity studies, environmental assessments, and marine ecosystem management. By addressing scale variability and achieving high accuracy even for rare species such as scallops, the proposed model supports practical applications in underwater monitoring and conservation.