CBAM-RIUnet: Breast Tumor Segmentation With Enhanced Breast Ultrasound and Test-Time Augmentation

Abstract

This study addresses the challenge of precise breast tumor segmentation in ultrasound images, crucial for effective Computer-Aided Diagnosis (CAD) in breast cancer. We introduce CBAM-RIUnet, a deep learning (DL) model for automated breast tumor segmentation in breast ultrasound (BUS) images. The model, featuring an efficient convolutional block attention module residual inception Unet, outperforms existing models, particularly excelling in Dice and IoU scores. CBAM-RIUnet follows the Unet structure with a residual inception depth-wise separable convolution, and incorporates a convolutional block attention module (CBAM) to eliminate irrelevant features and focus on the region of interest. Evaluation under three scenarios, including enhanced breast ultrasound (EBUS) and test-time augmentation (TTA), demonstrates impressive results. CBAM-RIUnet achieves Dice and IoU scores of 89.38% and 88.71%, respectively, showcasing significant improvements compared to state-of-the-art DL techniques. In conclusion, CBAM-RIUnet presents a highly effective and simplified DL model for breast tumor segmentation in BUS imaging.