Explainable AI based hybrid DRM-Net transfer learning model for breast cancer detection and classification using ultrasound images.
Authors
Affiliations (7)
Affiliations (7)
- Department of Health Information Management and Technology, College of Applied Medical Sciences, King Faisal University, 31982, Al-Ahsa, Saudi Arabia. [email protected].
- Department of Computer Sciences, School of Mathematical and Computer Sciences, BGSB University, Rajouri, J&K UT, 185234, India.
- Department of Computer Science and Engineering, Model Institute of Engineering and Technology, Jammu, J&K UT, India.
- Department of Nursing, College of Applied Medical Sciences, King Faisal University, 31982, Al-Ahsa, Saudi Arabia.
- Department of Chemistry, College of Science, King Faisal University, 31982, Al-Ahsa, Saudi Arabia.
- Department of Basic Sciences, Preparatory Year, King Faisal University, 31982, Al-Ahsa, Saudi Arabia.
- Department of Public Health, College of Applied Medical Sciences, King Faisal University, 31982, Al-Ahsa, Saudi Arabia. [email protected].
Abstract
Breast cancer is a serious health concern and one of the leading causes of cancer-related deaths among women worldwide. The early diagnosis of breast cancer is crucial for successful treatment and improved patient outcomes. Advanced computational techniques, particularly deep transfer learning have gained considerable attention for their effectiveness in medical image computing. Initially, six transfer learning models with different specifications are trained, validated and tested on the breast ultrasound image dataset. Additionally, a novel hybrid model, DRM-Net, is developed by stacking the top three TL models based on concatenation and flattening of deep dense layers. Various techniques, including image preprocessing, image masking, data augmentation, and hyperparameter tuning, are incorporated to enhance the overall performance of the considered models. The models are thoroughly evaluated using various standard metrics and statistical methods. Furthermore, to ensure transparent decision-making, the proposed model is interpreted using XAI-based class activation mapping (CAM) method with different versions. The proposed DRM-Net model demonstrated superior predictive performance compared to the other six transfer learning models. Among all the models, DRM-Net achieved the highest overall performance, with an accuracy of 96.71%, precision of 96%, recall of 97%, F1-score of 97%, and an AUC value of 99%, respectively. It also outperformed state-of-the-art similar studies in the literature. The experimental results demonstrate the potential utility of the hybrid DRM-Net model in improving the accuracy of breast cancer diagnosis, which could facilitate informed clinical decision-making. The proposed model can also be applied to other diseases where ultrasound imaging is available.