Advancing Breast Cancer-AI Diagnostics: An Explainable Deep Learning Model Using 2D Grayscale Ultrasound Imaging
Abbas, G. H.; Rehman, S. U.; Bargshady, G.
Show abstract
BackgroundBreast cancer stands as the primary reason for fatality in female patients from cancer worldwide. The diagnostic precision of ultrasound imaging depends on operator skills because it lacks invasive procedures but remains accessible through wide availability. The medical field needs automated systems which provide explainable results to help doctors achieve better breast lesion diagnosis accuracy. ObjectivesThis research presents a new deep learning diagnostic system based on MobileNetV2 architecture which achieves optimal results for analyzing 2D grayscale (B-mode) ultrasound images. The system combines Grad-CAM explainability tools with clinical-grade functionality to generate interpretable results for both correct and incorrect predictions. MethodsThe BUSI dataset served as the basis for model training and evaluation because it contained a total of 780 ultrasound images that contains 437 benign images, 210 malignant images and 133 normal images. We divided the data into training and validation sets which made up 80% and 20% of the total data respectively. We then standardized and normalized all images by resizing them to 192x192 pixels and converting them into 3-channel images. The model design included frozen base layers, global average pooling, dropout regularization and a fully connected classification head. The model received 7 training runs with a total of 150 epochs with 16 samples per batch using adam optimizer and categorical cross-entropy as the loss function. ResultsThe proposed model achieved 90.1% validation accuracy, 0.90 precision, 0.90 recall, 0.90 F1-score and 0.88 macro-averaged AUC. The Grad-CAM visualizations showed precise localization of features within the lesion area. The t-SNE projection of learned features displayed three separate lusters which corresponded to malignant, benign and normal tissue types. The system produced most of its errors when it failed to distinguish between benign and malignant lesions that appeared very similar to each other. ConclusionThe system provides both high diagnostic precision and clinical understanding which represents a major breakthrough for ultrasound-based breast cancer screening systems. Further research will direct upcoming work toward validating the system externally and implementing real-time functionality, obtaining necessary regulatory approvals as well as inclusion for multi-modality in deep learning.
Matching journals
The top 8 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Automated and Manual Quantification of Tumour Cellularity in Digital Slides for Tumour Burden Assessment 95%
- Automated detection of the HER2 gene amplification status in Fluorescence in situ hybridization images for the diagnostics of cancer tissues 94%
- Segmentation of Pancreatic Ductal Adenocarcinoma (PDAC) and surrounding vessels in CT images using deep convolutional neural networks and Texture Descriptors 94%
Similar papers in this journal
- A Machine Learning Ensemble Based on Radiomics to Predict BI-RADS Category and Reduce the Biopsy Rate of Ultrasound-Detected Suspicious Breast Masses 96%
- Demarcation line determination for diagnosis of gastric cancer disease range using unsupervised machine learning in magnifying narrow-band imaging 93%
- Deep neural frameworks improve the accuracy of general practitioners in the classification of pigmented skin lesions 91%
Similar papers in this journal
- The mathematics of erythema: Development of machine learning models for artificial intelligence assisted measurement and severity scoring of radiation induced dermatitis 94%
- Deep learning ensemble for abdominal aortic calcification scoring from lumbar spine X-ray and DXA images 93%
- Two-Step Machine Learning to Diagnose and Predict Involvement of Lungs in COVID-19 and Pneumonia using CT Radiomics 92%
Similar papers in this journal
- Fibre tracing in biomedical images: An objective comparison between seven algorithms 94%
- Automated Detection of COVID-19 through Convolutional Neural Network using Chest x-ray images 94%
- Enhancing Breast Ultrasound Segmentation through Fine-tuning and Optimization Techniques: Sharp Attention UNet 93%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.