Enhanced Breast Cancer Tumor Classification using MobileNetV2: A Detailed Exploration on Image Intensity, Error Mitigation, and Streamlit-driven Real-time Deployment
Surya, Aaditya, Shah, Aditya, Kabore, Jarnell, Sasikumar, Subash
–arXiv.org Artificial Intelligence
This research introduces a sophisticated transfer learning model based on Google's MobileNetV2 for breast cancer tumor classification into normal, benign, and malignant categories, utilizing a dataset of 1576 ultrasound images (265 normal, 891 benign, 420 malignant). The model achieves an accuracy of 0.82, precision of 0.83, recall of 0.81, ROC-AUC of 0.94, PR-AUC of 0.88, and MCC of 0.74. It examines image intensity distributions and misclassification errors, offering improvements for future applications. Addressing dataset imbalances, the study ensures a generalizable model. This work, using a dataset from Baheya Hospital, Cairo, Egypt, compiled by Walid Al-Dhabyani et al., emphasizes MobileNetV2's potential in medical imaging, aiming to improve diagnostic precision in oncology. Additionally, the paper explores Streamlit-based deployment for real-time tumor classification, demonstrating MobileNetV2's applicability in medical imaging and setting a benchmark for future research in oncology diagnostics.
arXiv.org Artificial Intelligence
Jan-6-2024
- Country:
- Africa > Middle East
- Egypt > Cairo Governorate > Cairo (0.24)
- North America > United States
- New York > Clinton County (0.14)
- Africa > Middle East
- Genre:
- Research Report > Experimental Study (0.46)
- Industry:
- Health & Medicine > Therapeutic Area > Oncology > Breast Cancer (0.86)
- Technology: