journal article·2023-10-16Open access

RMAU-Net: Breast Tumor Segmentation Network Based on Residual Depthwise Separable Convolution and Multiscale Channel Attention Gates

Sheng Yuan, Zhao Qiu, Peipei Li, Yuqi Hong

Applied Sciences· Volume 13· Issue 20

Abstract

Breast cancer is one of the most common female diseases, posing a great threat to women’s health, and breast ultrasound imaging is a common method for breast cancer diagnosis. In recent years, U-Net and its variants have dominated the medical image segmentation field with their excellent performance. However, the existing U-type segmentation networks have the following problems: (1) the design of the feature extractor is complicated, and the calculation difficulty is increased; (2) the skip connection operation simply combines the features of the encoder and the decoder, without considering both spatial and channel dimensions; (3) during the downsampling phase, the pooling operation results in the loss of feature information. To address the above deficiencies, this paper proposes a breast tumor segmentation network, RMAU-Net, that combines residual depthwise separable convolution and a multi-scale channel attention gate. Specifically, we designed the RDw block, which has a simple structure and a larger sensory field, to overcome the localization problem of convolutional operations. Meanwhile, the MCAG module is designed to correct the low-level features in both spatial and channel dimensions and assist the high-level features to recover the up-sampling and pinpoint non-regular breast tumor features. In addition, this paper used the Patch Merging operation instead of the pooling method to prevent the loss of breast ultrasound image information. Experiments were conducted on two breast ultrasound datasets, Dataset B and BUSI, and the results show that the method in this paper has superior segmentation performance and better generalization.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: AI in cancer detection · Brain Tumor Detection and Classification · Radiomics and Machine Learning in Medical Imaging

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

School of Computer Science and Technology, Hainan University, Haikou 570228, China

Funding

Education Department of Hainan Province

References (47)

Show source-provided references
1. Radiological findings of breast involvement in benign and malignant systemic diseases10.1590/0100-3984.2016.0125 ↗
2. Cancer statistics, 201510.3322/caac.21254 ↗
3. Computer-aided detection of clustered microcalcifications in digital breast tomosynthesis: A 3D approach10.1118/1.3662072 ↗
4. Automated breast cancer detection and classification using ultrasound images: A survey10.1016/j.patcog.2009.05.012 ↗
5. Ultrasound image segmentation: A survey10.1109/TMI.2006.877092 ↗
6. Speckle in ultrasonic imaging10.1016/0041-624X(81)90007-X ↗
7. Segmentation of ultrasound B-mode images with intensity inhomogeneity correction10.1109/42.981233 ↗
8. Liu, Y., Ren, L., Cao, X., and Tong, Y. (2020). Breast tumors recognition based on edge feature extraction using support vector machine. Biomed. Signal Process. Control, 58.10.1016/j.bspc.2019.101825 ↗
9. Inoue, K., Yamanaka, C., Kawasaki, A., Koshimizu, K., Sasaki, T., and Doi, T. (2017). Computer aided detection of breast cancer on ultrasound imaging using deep learning. Ultrasound Med. Biol., 43.10.1016/j.ultrasmedbio.2017.08.1002 ↗
10. Computerized lesion detection on breast ultrasound10.1118/1.1485995 ↗
11. Automatic segmentation of breast lesions on ultrasound10.1118/1.1386426 ↗
12. Tumor detection in automated breast ultrasound images using quantitative tissue clustering10.1118/1.4869264 ↗
13. A novel segmentation method for breast ultrasound images based on neutrosophic l-means clustering10.1118/1.4747271 ↗
14. Mitra, A., De, A., and Bhattacharjee, A.K. (2014, January 14–15). MRI Skull Bone Lesion Segmentation Using Distance Based Watershed Segmentation. Proceedings of the 3rd International Conference on Frontiers of Intelligent Computing: Theory and Applications (FICTA), Bhubaneswar, Odisha, India.10.1007/978-3-319-12012-6_28 ↗
15. Semi-automatic breast ultrasound image segmentation based on mean shift and graph cuts10.1177/0161734614524735 ↗
16. A robust graph-based segmentation method for breast tumors in ultrasound images10.1016/j.ultras.2011.08.011 ↗
17. Supervised and unsupervised algorithms for bioinformatics and data science10.1016/j.pbiomolbio.2019.11.012 ↗
18. Sohail, A., Younas, M., Bhatti, Y., Li, Z., Tunç, S., and Abid, M. (2019). Analysis of trabecular bone mechanics using machine learning. Evol. Bioinform., 15.10.1177/1176934318825084 ↗
19. Artificial intelligence to link environmental endocrine disruptors (EEDs) with bone diseases10.1142/S1793962322500192 ↗
20. Adaptive kernel-based fuzzy c-means clustering with spatial constraints for image segmentation10.1142/S021800141954003X ↗
21. Xu, M., Huang, K., Chen, Q., and Qi, X. (2021, January 13–16). Mssa-net: Multi-scale self-attention network for breast ultrasound image segmentation. Proceedings of the 2021 IEEE 18th International Symposium on Biomedical Imaging (ISBI), Nice, France.10.1109/ISBI48211.2021.9433899 ↗
22. Mannem, R., Ca, V., and Ghosh, P.K. (2019, January 20–23). A SegNet based image enhancement technique for air-tissue boundary segmentation in real-time magnetic resonance imaging video. Proceedings of the 2019 National Conference on Communications (NCC), Bangalore, India.10.1109/NCC.2019.8732257 ↗
23. Cross-modality synthesis from CT to PET using FCN and GAN networks for improved automated lesion detection10.1016/j.engappai.2018.11.013 ↗
24. Amiri, M., Brooks, R., and Rivaz, H. (2019). Domain Adaptation and Representation Transfer and Medical Image Learning with Less Labels and Imperfect Data, Springer.
25. Zhou, Z., Rahman Siddiquee, M.M., Tajbakhsh, N., and Liang, J. (2018). Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support: 4th International Workshop, DLMIA 2018, and 8th International Workshop, ML-CDS 2018, Held in Conjunction with MICCAI 2018, Granada, Spain, 20 September 2018, Springer International Publishing.
26. Oktay, O., Schlemper, J., Folgoc, L.L., Lee, M., Heinrich, M., Misawa, K., Mori, K., McDonagh, S., Hammerla, N.Y., and Kainz, B. (2018). Attention u-net: Learning where to look for the pancreas. arXiv.
27. Xiao, X., Lian, S., Luo, Z., and Li, S. (2018, January 19–21). Weighted res-unet for high-quality retina vessel segmentation. Proceedings of the 2018 9th International Conference on Information Technology in Medicine and Education (ITME), Hangzhou, China.10.1109/ITME.2018.00080 ↗
28. Ibtehaz, N., and Sohel Rahman, M.M. (2019). Rethinking the U-Net architecture for multimodal biomedical image segmentation. arXiv.10.1016/j.neunet.2019.08.025 ↗
29. Huang, H., Lin, L., Tong, R., Hu, H., Zhang, Q., Iwamoto, Y., Han, X., Chen, Y.-W., and Wu, J. (2020, January 4–8). Unet 3+: A full-scale connected unet for medical image segmentation. Proceedings of the ICASSP 2020-2020 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Barcelona, Spain.10.1109/ICASSP40776.2020.9053405 ↗
30. Valanarasu, J.M.J., and Patel, V.M. (2022, January 18–22). Unext: Mlp-based rapid medical image segmentation network. Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, Singapore.10.1007/978-3-031-16443-9_3 ↗
31. Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A.N., Kaiser, Ł., and Polosukhin, I. (2017). Attention is all you need. Adv. Neural Inf. Process. Syst., 30.
32. Chen, J., Lu, Y., Yu, Q., Luo, X., Adeli, E., Wang, Y., Lu, L., Yuille, A.L., and Zhou, Y. (2021). Transunet: Transformers make strong encoders for medical image segmentation. arXiv.
33. Valanarasu, J.M.J., Oza, P., Hacihaliloglu, I., and Patel, V.M. (October, January 27). Medical transformer: Gated axial-attention for medical image segmentation. Proceedings of the Medical Image Computing and Computer Assisted Intervention–MICCAI 2021: 24th International Conference, Strasbourg, France.10.1007/978-3-030-87193-2_4 ↗
34. Wang, W., Chen, C., Ding, M., Yu, H., Zha, S., and Li, J. (October, January 27). Transbts: Multimodal brain tumor segmentation using transformer. Proceedings of the Medical Image Computing and Computer Assisted Intervention–MICCAI 2021: 24th International Conference, Strasbourg, France.10.1007/978-3-030-87193-2_11 ↗
35. Shareef, B., Xian, M., and Vakanski, A. (2020, January 3–7). Stan: Small tumor-aware network for breast ultrasound image segmentation. Proceedings of the 2020 IEEE 17th International Symposium on Biomedical Imaging (ISBI), Iowa City, IA, USA.10.1109/ISBI45749.2020.9098691 ↗
36. Segmentation of breast anatomy for automated whole breast ultrasound images with boundary regularized convolutional encoder–decoder network10.1016/j.neucom.2018.09.043 ↗
37. Global guidance network for breast lesion segmentation in ultrasound images10.1016/j.media.2021.101989 ↗
38. Boundary-rendering network for breast lesion segmentation in ultrasound images10.1016/j.media.2022.102478 ↗
39. Tong, Y., Liu, Y., Zhao, M., Meng, L., and Zhang, J. (2021). Improved U-net MALF model for lesion segmentation in breast ultrasound images. Biomed. Signal Process. Control, 68.10.1016/j.bspc.2021.102721 ↗
40. Zhuang, Z., Li, N., Joseph Raj, A.N., Mahesh, V.G., and Qiu, S. (2019). An RDAU-NET model for lesion segmentation in breast ultrasound images. PLoS ONE, 14.10.1371/journal.pone.0221535 ↗
41. Deep Learning-based Multi-stage segmentation method using ultrasound images for breast cancer diagnosis
42. Chollet, F. (2020, January 13–19). Xception: Deep learning with depthwise separable convolutions. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.
43. Dar, M.F., and Ganivada, A. (2023). EfficientU-Net: A Novel Deep Learning Method for Breast Tumor Segmentation and Classification in Ultrasound Images. Neural Process. Lett., 1–24.10.1007/s11063-023-11333-x ↗
44. Hu, J., Shen, L., and Sun, G. (2020, January 13–19). Squeeze-and-excitation networks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.
45. Liu, Z., Lin, Y., Cao, Y., Hu, H., Wei, Y., Zhang, Z., Lin, S., and Guo, B. (2021, January 10–17). Swin transformer: Hierarchical vision transformer using shifted windows. Proceedings of the IEEE/CVF International Conference on Computer Vision, Montreal, QC, Canada.10.1109/ICCV48922.2021.00986 ↗
46. Breast ultrasound region of interest detection and lesion localisation10.1016/j.artmed.2020.101880 ↗
47. Dataset of breast ultrasound images10.1016/j.dib.2019.104863 ↗