FRBNet: Feedback refinement boundary network for semantic segmentation in breast ultrasound images

Abstract

Ultrasound breast tumour segmentation is a key step in computer-aided diagnosis and provides an important basis for clinical diagnosis and analysis. Accurate segmentation of breast tumours from ultrasound images is a challenging task due to the characteristics of black shadows, blurred boundaries and uneven colour intensity variations between classes. Currently, most breast tumour segmentation methods focus on extracting multi-scale information and fusing contextual information while underestimating the importance of feature information that can assist in identifying object boundaries in segmentation tasks. The loss of boundary feature information can easily lead to discontinuity or inaccuracy of the target boundary when the network generates the final prediction map. To address this problem, we propose a new feedback refinement boundary network (FRBNet) for accurate segmentation of breast tumour regions in ultrasound images, which mainly consists of a channel calibration module (CCM), boundary detection (BD) module, and feedback refinement module (FRM). Specifically, before fusing low-level feature maps with high-level feature maps, CCM first adopts the method of redistributing feature channel responses to enhance the channels carrying key target information and suppress the noisy channels in low-level feature maps. The BD module then improves the quality of the boundaries in the segmentation results by additionally learning the boundaries of breast tumours to provide accurate boundary feature information for subsequent prediction. The FRM employs a feedback mechanism that complementarily fuses the coarse prediction map and the feature map containing the target boundary feature information, thus achieving the best prediction results before generating the final prediction map. Experimental results on a public ultrasound breast dataset show that our network outperforms other medical image segmentation methods.