DARU-Net: A dual attention residual U-Net for uterine fibroids segmentation on MRI.

Abstract

Uterine fibroid is the most common benign tumor in female reproductive organs. In order to guide the treatment, it is crucial to detect the location, shape, and size of the tumor. This study proposed a deep learning approach based on attention mechanisms to segment uterine fibroids automatically on preoperative Magnetic Resonance (MR) images. The proposed method is based on U-Net architecture and integrates two attention mechanisms: channel attention of squeeze-and-excitation (SE) blocks with residual connections, spatial attention of pyramid pooling module (PPM). We did the ablation study to verify the performance of these two attention mechanisms module and compared DARU-Net with other deep learning methods. All experiments were performed on a clinical dataset consisting of 150 cases collected from our hospital. Among them, 120 cases were used as the training set, and 30 cases are used as the test set. After preprocessing and data augmentation, we trained the network and tested it on the test dataset. We evaluated segmentation performance through the Dice similarity coefficient (DSC), precision, recall, and Jaccard index (JI). The average DSC, precision, recall, and JI of DARU-Net reached 0.8066 ± 0.0956, 0.8233 ± 0.1255, 0.7913 ± 0.1304, and 0.6743 ± 0.1317. Compared with U-Net and other deep learning methods, DARU-Net was more accurate and stable. This work proposed an optimized U-Net with channel and spatial attention mechanisms to segment uterine fibroids on preoperative MR images. Results showed that DARU-Net was able to accurately segment uterine fibroids from MR images.