We found a match
Your institution may have rights to this item. Sign in to continue.
- Title
Revolutionizing breast ultrasound diagnostics with EfficientNet-B7 and Explainable AI.
- Authors
Latha, M.; Kumar, P. Santhosh; Chandrika, R. Roopa; Mahesh, T. R.; Kumar, V. Vinoth; Guluwadi, Suresh
- Abstract
Breast cancer is a leading cause of mortality among women globally, necessitating precise classification of breast ultrasound images for early diagnosis and treatment. Traditional methods using CNN architectures such as VGG, ResNet, and DenseNet, though somewhat effective, often struggle with class imbalances and subtle texture variations, leading to reduced accuracy for minority classes such as malignant tumors. To address these issues, we propose a methodology that leverages EfficientNet-B7, a scalable CNN architecture, combined with advanced data augmentation techniques to enhance minority class representation and improve model robustness. Our approach involves fine-tuning EfficientNet-B7 on the BUSI dataset, implementing RandomHorizontalFlip, RandomRotation, and ColorJitter to balance the dataset and improve model robustness. The training process includes early stopping to prevent overfitting and optimize performance metrics. Additionally, we integrate Explainable AI (XAI) techniques, such as Grad-CAM, to enhance the interpretability and transparency of the model's predictions, providing visual and quantitative insights into the features and regions of ultrasound images influencing classification outcomes. Our model achieves a classification accuracy of 99.14%, significantly outperforming existing CNN-based approaches in breast ultrasound image classification. The incorporation of XAI techniques enhances our understanding of the model's decision-making process, thereby increasing its reliability and facilitating clinical adoption. This comprehensive framework offers a robust and interpretable tool for the early detection and diagnosis of breast cancer, advancing the capabilities of automated diagnostic systems and supporting clinical decision-making processes.
- Subjects
IMAGE recognition (Computer vision); BREAST ultrasound; ULTRASONIC imaging; DATA augmentation; BREAST imaging
- Publication
BMC Medical Imaging, 2024, Vol 24, Issue 1, p1
- ISSN
1471-2342
- Publication type
Article
- DOI
10.1186/s12880-024-01404-3