BG-UISNet: A Novel Boundary-Guided Network for Robust and Precise Underwater Image Segmentation

Abstract

The exploration of marine environments is a key focus in marine biology, ecological conservation, and underwater robotics, driving the demand for precise underwater image segmentation to advance related scientific research. Yet, the inherent complexity of underwater settings, marked by light attenuation, turbidity, and the camouflage traits of marine organisms, severely hampers segmentation performance, posing challenges to meet the practical needs of marine applications. To address this, we propose BG-UISNet, a novel boundary-guided underwater image segmentation network built upon recent advances in powerful large-scale pre-trained models. This network incorporates a Boundary-Aware Module (BAM) to sharpen edge detection, a Boundary-Guided Feature Enhancement Module (BFEM) to refine edge-related features, and a Multi-Scale Context Aggregation Module (MCAM) to capture global contextual details. Experiments show that BG-UISNet outperforms existing methods across four classic underwater datasets. Ablation studies on the MAS3K and RMAS datasets further reveal average performance gains of 3.25%, 1.65%, 3.60%, and 1.00% in mIoU, S<sub xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">α</sub>, F<sup xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">w</sup><sub xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">β</sub>, and mE<sub xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">ϕ</sub>, respectively. Moreover, BG-UISNet demonstrates remarkable generalization ability. It also performs well on two general object segmentation tasks: camouflaged object detection and salient object detection. These results confirm its ability to effectively fuse low-level edge details with high-level global semantics. By overcoming the shortcomings of traditional convolutional neural networks and Transformer models in underwater contexts, BG-UISNet offers a robust segmentation tool for marine vision tasks, paving new technical avenues for future underwater environmental studies.