Search papers, labs, and topics across Lattice.
The paper introduces PBE-UNet, a novel U-Net architecture for ultrasound image segmentation designed to address challenges posed by low contrast, blurry boundaries, and scale variations. PBE-UNet incorporates a scale-aware aggregation module (SAAM) for multi-scale contextual information and a boundary-guided feature enhancement (BGFE) module that progressively expands boundary predictions to cover wider segmentation error regions. Experiments on four ultrasound datasets demonstrate that PBE-UNet achieves state-of-the-art performance in ultrasound image segmentation.
By progressively expanding boundary predictions, PBE-UNet effectively focuses on challenging segmentation error regions in ultrasound images, leading to state-of-the-art performance.
Accurate lesion segmentation in ultrasound images is essential for preventive screening and clinical diagnosis, yet remains challenging due to low contrast, blurry boundaries, and significant scale variations. Although existing deep learning-based methods have achieved remarkable performance, these methods still struggle with scale variations and indistinct tumor boundaries. To address these challenges, we propose a progressive boundary enhanced U-Net (PBE-UNet). Specially, we first introduce a scale-aware aggregation module (SAAM) that dynamically adjusts its receptive field to capture robust multi-scale contextual information. Then, we propose a boundary-guided feature enhancement (BGFE) module to enhance the feature representations. We find that there are large gaps between the narrow boundary and the wide segmentation error areas. Unlike existing methods that treat boundaries as static masks, the BGFE module progressively expands the narrow boundary prediction into broader spatial attention maps. Thus, broader spatial attention maps could effectively cover the wider segmentation error regions and enhance the model's focus on these challenging areas. We conduct expensive experiments on four benchmark ultrasound datasets, BUSI, Dataset B, TN3K, and BP. The experimental results how that our proposed PBE-UNet outperforms state-of-the-art ultrasound image segmentation methods. The code is at https://github.com/cruelMouth/PBE-UNet.
