Comparative Study of Transfer Learning with VGG16 and VGG19 for Breast Cancer Classification
Breast cancer is a significant health challenge, particularly in resource-constrained regions like Pakistan, where early detection remains a major hurdle. This study evaluates the effectiveness of transfer learning using VGG16 and VGG19 models for mammogram analysis, employing the CBIS-DDSM dataset. Both models underwent 5-fold cross-validation to ensure robust performance. The VGG16 model achieved an accuracy of 72.78%, while VGG19 demonstrated a slightly lower accuracy of 70.79% with data augmentation. The findings underscore the potential of transfer learning to enhance diagnostic accuracy, particularly in areas with limited healthcare resources. This research highlights the importance of adopting AI-based diagnostic tools to improve breast cancer detection and contribute to reducing mortality rates associated with late-stage diagnoses.
Authors
- Sumra Khan (ORCID: https://orcid.org/0009-0004-9667-2262)
- Uzma Fatima
- Fatima Tuz Zahra (ORCID: https://orcid.org/0000-0001-5989-1117)
- Sheeraz Arif (ORCID: https://orcid.org/0000-0002-1005-7209)
- Amber Asif
Institutions
- Habib University (PK)
Publication Details
- Published
- 2024-12-30
- DOI
- https://doi.org/10.1109/inmic64792.2024.11004305
- Citations
- 2
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.69