journal article·2023-07-28Open access

Spatial Attention Mechanism and Cascade Feature Extraction in a U-Net Model for Enhancing Breast Tumor Segmentation

Payam Zarbakhsh

Applied Sciences· Volume 13· Issue 15

Abstract

In the field of medical imaging, the accurate segmentation of breast tumors is a critical task for the diagnosis and treatment of breast cancer. To address the challenges posed by fuzzy boundaries, vague tumor shapes, variation in tumor size, and illumination variation, we propose a new approach that combines a U-Net model with a spatial attention mechanism. Our method utilizes a cascade feature extraction technique to enhance the subtle features of breast tumors, thereby improving segmentation accuracy. In addition, our model incorporates a spatial attention mechanism to enable the network to focus on important regions of the image while suppressing irrelevant areas. This combination of techniques leads to significant improvements in segmentation accuracy, particularly in challenging cases where tumors have fuzzy boundaries or vague shapes. We evaluate our suggested technique on the Mini-MIAS dataset and demonstrate state-of-the-art performance, surpassing existing methods in terms of accuracy, sensitivity, and specificity. Specifically, our method achieves an overall accuracy of 91%, a sensitivity of 91%, and a specificity of 93%, demonstrating its effectiveness in accurately identifying breast tumors.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: AI in cancer detection · Advanced Neural Network Applications · Radiomics and Machine Learning in Medical Imaging

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

Affiliations

Electrical and Electronic Engineering Department, Cyprus International University, Via Mersin 10, 99258 Nicosia, Turkey

Electrical and Electronic Engineering Department, Eastern Mediterranean University, Via Mersin 10, 99628 Famagusta, Turkey

References (69)

Show source-provided references
1. Deep and machine learning techniques for medical imaging-based breast cancer: A comprehensive review10.1016/j.eswa.2020.114161 ↗
2. Rapid review: Radiomics and breast cancer10.1007/s10549-018-4675-4 ↗
3. An extended approach to the diagnosis of tumour location in breast cancer using deep learning10.1007/s12652-021-03613-y ↗
4. Ranjbarzadeh, R., Sarshar, N.T., Ghoushchi, S.J., Esfahani, M.S., Parhizkar, M., Pourasad, Y., Anari, S., and Bendechache, M. (2022). MRFE-CNN: Multi-route feature extraction model for breast tumor segmentation in Mammograms using a convolutional neural network. Ann. Oper. Res., 1–22.10.1007/s10479-022-04755-8 ↗
5. Hierarchical Convolutional Neural Networks for Segmentation of Breast Tumors in MRI with Application to Radiogenomics10.1109/TMI.2018.2865671 ↗
6. A Gaussian Multivariate Hidden Markov Model for Breast Tumor Diagnosis10.4236/am.2021.128048 ↗
7. Breast tumor localization and segmentation using machine learning techniques: Overview of datasets, findings, and methods10.1016/j.compbiomed.2022.106443 ↗
8. Wang, S., Sun, K., Wang, L., Qu, L., Yan, F., Wang, Q., and Shen, D. (2021). Breast Tumor Segmentation in DCE-MRI With Tumor Sensitive Synthesis. IEEE Trans. Neural Netw. Learn. Syst., 1–12.10.1109/TNNLS.2021.3129781 ↗
9. Le, P.T., Pham, B.-T., Chang, C.-C., Hsu, Y.-C., Tai, T.-C., Li, Y.-H., and Wang, J.-C. (2023). Anti-Aliasing Attention U-net Model for Skin Lesion Segmentation. Diagnostics, 13.10.3390/diagnostics13081460 ↗
10. Le, P.T., Pham, T., Hsu, Y.-C., and Wang, J.-C. (2022). Convolutional Blur Attention Network for Cell Nuclei Segmentation. Sensors, 22.10.3390/s22041586 ↗
11. Ranjbarzadeh, R., Zarbakhsh, P., Caputo, A., Tirkolaee, E.B., and Bendechache, M. (2022). Brain Tumor Segmentation based on an Optimized Convolutional Neural Network and an Improved Chimp Optimization Algorithm, preprint.10.21203/rs.3.rs-2203596/v1 ↗
12. Attention-Enriched Deep Learning Model for Breast Tumor Segmentation in Ultrasound Images10.1016/j.ultrasmedbio.2020.06.015 ↗
13. Accurate segmentation of breast tumors using AE U-net with HDC model in ultrasound images10.1016/j.bspc.2021.103299 ↗
14. A review on image-based approaches for breast cancer detection, segmentation, and classification10.1016/j.eswa.2021.115204 ↗
15. Breast tumor segmentation in digital mammograms using spiculated regions10.1016/j.bspc.2022.103652 ↗
16. A Discriminative Level Set Method with Deep Supervision for Breast Tumor Segmentation10.1016/j.compbiomed.2022.105995 ↗
17. Deep Learning Based Capsule Neural Network Model for Breast Cancer Diagnosis Using Mammogram Images10.1007/s12539-021-00467-y ↗
18. Zebari, D.A., Ibrahim, D.A., and Al-Zebari, A. (2022, January 15–17). Suspicious Region Segmentation Using Deep Features in Breast Cancer Mammogram Images. Proceedings of the 2nd 2022 International Conference on Computer Science and Software Engineering, CSASE, Duhok, Iraq.10.1109/CSASE51777.2022.9759633 ↗
19. A Novel Hybrid Image Segmentation Method for Detection of Suspicious Regions in Mammograms Based on Adaptive Multi-Thresholding (HCOW)10.1109/ACCESS.2021.3089077 ↗
20. Brain tumor segmentation of MRI images: A comprehensive review on the application of artificial intelligence tools10.1016/j.compbiomed.2022.106405 ↗
21. Deep Refinement: Capsule network with attention mechanism-based system for text classification10.1007/s00521-019-04620-z ↗
22. Attention mechanisms in computer vision: A survey10.1007/s41095-022-0271-y ↗
23. Unsupervised Pansharpening Based on Self-Attention Mechanism10.1109/TGRS.2020.3009207 ↗
24. Brain tumor segmentation based on deep learning and an attention mechanism using MRI multi-modalities brain images10.1038/s41598-021-90428-8 ↗
25. Shuang, K., Ren, X., Guo, H., Loo, J., and Xu, P. (2018). Advances in Intelligent Systems and Computing, Springer.
26. Yao, H., Zhang, X., Zhou, X., and Liu, S. (2019). Parallel Structure Deep Neural Network Using CNN and RNN with an Attention Mechanism for Breast Cancer Histology Image Classification. Cancers, 11.10.3390/cancers11121901 ↗
27. Zhou, T., Ruan, S., Guo, Y., and Canu, S. (2020, January 3–7). A Multi-Modality Fusion Network Based on Attention Mechanism for Brain Tumor Segmentation. Proceedings of the International Symposium on Biomedical Imaging, IEEE Computer Society, Iowa City, IA, USA.10.1109/ISBI45749.2020.9098392 ↗
28. Multiscale lightweight 3D segmentation algorithm with attention mechanism: Brain tumor image segmentation10.1016/j.eswa.2022.119166 ↗
29. Ronneberger, O., Fischer, P., and Brox, T. (2015). U-Net: Convolutional Networks for Biomedical Image Segmentation, Springer. Lecture Notes in Computer Science (Including Subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics).10.1007/978-3-319-24574-4_28 ↗
30. DSU-Net: Distraction-Sensitive U-Net for 3D lung tumor segmentation10.1016/j.engappai.2021.104649 ↗
31. Diagnostic and Gradation Model of Osteoporosis Based on Improved Deep U-Net Network10.1007/s10916-019-1502-3 ↗
32. dResU-Net: 3D deep residual U-Net based brain tumor segmentation from multimodal MRI10.1016/j.bspc.2022.103861 ↗
33. Liver CT sequence segmentation based with improved U-Net and graph cut10.1016/j.eswa.2019.01.055 ↗
34. A U-Net Ensemble for breast lesion segmentation in DCE MRI10.1016/j.compbiomed.2021.105093 ↗
35. Lee, B., Yamanakkanavar, N., and Choi, J.Y. (2020). Automatic segmentation of brain MRI using a novel patch-wise U-net deep architecture. PLoS ONE, 15.10.1371/journal.pone.0236493 ↗
36. An Intelligent Segmentation and Diagnosis Method for Diabetic Retinopathy Based on Improved U-NET Network10.1007/s10916-019-1432-0 ↗
37. Mo, S., Cho, M., and Shin, J. (2020). Freeze the Discriminator: A Simple Baseline for Fine-Tuning GANs. arXiv.
38. Weld image deep learning-based on-line defects detection using convolutional neural networks for Al alloy in robotic arc welding10.1016/j.jmapro.2019.06.023 ↗
39. Deep learning and machine learning techniques to improve hand movement classification in myoelectric control system10.1016/j.bbe.2021.03.006 ↗
40. A novel pipeline framework for multi oriented scene text image detection and recognition10.1016/j.eswa.2020.114549 ↗
41. A Deep Learning Approach for Robust, Multi-oriented, and Curved Text Detection
42. Recognizing the Damaged Surface Parts of Cars in the Real Scene Using a Deep Learning Framework10.1155/2022/5004129 ↗
43. Sau, P.C. (2022). Retinal Blood Vessel Segmentation Using Attention Module and Tversky Loss Function, Springer.10.1007/978-981-19-0976-4_41 ↗
44. A scale-aware UNet++ model combined with attentional context supervision and adaptive Tversky loss for accurate airway segmentation10.1007/s10489-022-04380-9 ↗
45. Skin Lesion Segmentation Based on Edge Attention Vnet with Balanced Focal Tversky Loss10.1155/2022/4677044 ↗
46. Kasgari, A.B., Safavi, S., Nouri, M., Hou, J., Sarshar, N.T., and Ranjbarzadeh, R. (2023). Point-of-Interest Preference Model Using an Attention Mechanism in a Convolutional Neural Network. Bioengineering, 10.10.3390/bioengineering10040495 ↗
47. Ranjbarzadeh, R., Ghoushchi, S.J., Sarshar, N.T., Tirkolaee, E.B., Ali, S.S., Kumar, T., and Bendechache, M. (2023). ME-CCNN: Multi-encoded images and a cascade convolutional neural network for breast tumor segmentation and recognition. Artif. Intell. Rev., 1–38.10.1007/s10462-023-10426-2 ↗
48. Presentation of a Segmentation Method for a Diabetic Retinopathy Patient’s Fundus Region Detection Using a Convolutional Neural Network10.1155/2021/7714351 ↗
49. An Extended Approach to Predict Retinopathy in Diabetic Patients Using the Genetic Algorithm and Fuzzy C-Means10.1155/2021/5597222 ↗
50. Nerve optic segmentation in CT images using a deep learning model and a texture descriptor10.1007/s40747-022-00694-w ↗
51. (2022, October 08). The Mini-MIAS Database of Mammograms. Available online: http://peipa.essex.ac.uk/info/mias.html.
52. Tsochatzidis, L., Costaridou, L., and Pratikakis, I. (2019). Deep Learning for Breast Cancer Diagnosis from Mammograms—A Comparative Study. J. Imaging, 5.10.3390/jimaging5030037 ↗
53. Breast tumor segmentation in 3D automatic breast ultrasound using Mask scoring R-CNN10.1002/mp.14569 ↗
54. A novel wavelet decomposition and transformation convolutional neural network with data augmentation for breast cancer detection using digital mammogram10.1038/s41598-022-09905-3 ↗
55. Data Augmentation for Brain-Tumor Segmentation: A Review10.3389/fncom.2019.00083 ↗
56. Wang, F., Zhong, S.-H., Peng, J., Jiang, J., and Liu, Y. (2018). Data Augmentation for EEG-Based Emotion Recognition with Deep Convolutional Neural Networks, Springer. Lecture Notes in Computer Science (Including Subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics).10.1007/978-3-319-73600-6_8 ↗
57. A survey on Image Data Augmentation for Deep Learning10.1186/s40537-019-0197-0 ↗
58. Segmentation of Masses on Mammograms Using Data Augmentation and Deep Learning10.1007/s10278-020-00330-4 ↗
59. On the Importance of Visual Context for Data Augmentation in Scene Understanding10.1109/TPAMI.2019.2961896 ↗
60. Singh, A., Ranjbarzadeh, R., Raj, K., Kumar, T., and Roy, A.M. (2023). Understanding EEG signals for subject-wise Definition of Armoni Activities. arXiv.
61. Time-Frequency Analysis of EEG Signals and GLCM Features for Depth of Anesthesia Monitoring10.1155/2021/8430565 ↗
62. HECON: Weight assessment of the product loyalty criteria considering the customer decision's halo effect using the convolutional neural networks10.1016/j.ins.2022.12.027 ↗
63. LMA-Net: A lesion morphology aware network for medical image segmentation towards breast tumors10.1016/j.compbiomed.2022.105685 ↗
64. A breast tumors segmentation and elimination of pectoral muscle based on hidden markov and region growing10.1007/s11042-018-6089-z ↗
65. Safavi, S., and Jalali, M. (2021). RecPOID: POI Recommendation with Friendship Aware and Deep CNN. Futur. Internet, 13.10.3390/fi13030079 ↗
66. DeePOF: A hybrid approach of deep convolutional neural network and friendship to Point-of-Interest (POI) recommendation system in location-based social networks10.1002/cpe.6981 ↗
67. Corrigendum to “Automated liver and tumor segmentation based on concave and convex points using fuzzy c-means and mean shift clustering” [Measurement 150 (2020) 107086]10.1016/j.measurement.2019.107230 ↗
68. Investigation of Effectiveness of Shuffled Frog-Leaping Optimizer in Training a Convolution Neural Network
69. Review of Deep Learning Approaches for Thyroid Cancer Diagnosis10.1155/2022/5052435 ↗