Version 1
: Received: 12 September 2024 / Approved: 12 September 2024 / Online: 12 September 2024 (11:34:10 CEST)
How to cite:
Shobayo, O.; Tanimola, O.; Popoola, O.; Okoyeigbo, O. Breast Cancer Classification Using Fine-Tuned SWIN Transformer Model on Mammographic Images. Preprints2024, 2024090982. https://doi.org/10.20944/preprints202409.0982.v1
Shobayo, O.; Tanimola, O.; Popoola, O.; Okoyeigbo, O. Breast Cancer Classification Using Fine-Tuned SWIN Transformer Model on Mammographic Images. Preprints 2024, 2024090982. https://doi.org/10.20944/preprints202409.0982.v1
Shobayo, O.; Tanimola, O.; Popoola, O.; Okoyeigbo, O. Breast Cancer Classification Using Fine-Tuned SWIN Transformer Model on Mammographic Images. Preprints2024, 2024090982. https://doi.org/10.20944/preprints202409.0982.v1
APA Style
Shobayo, O., Tanimola, O., Popoola, O., & Okoyeigbo, O. (2024). Breast Cancer Classification Using Fine-Tuned SWIN Transformer Model on Mammographic Images. Preprints. https://doi.org/10.20944/preprints202409.0982.v1
Chicago/Turabian Style
Shobayo, O., Olusogo Popoola and Obinna Okoyeigbo. 2024 "Breast Cancer Classification Using Fine-Tuned SWIN Transformer Model on Mammographic Images" Preprints. https://doi.org/10.20944/preprints202409.0982.v1
Abstract
Breast cancer is the most prevalent type of disease among women. It has become one of the foremost causes of death among women globally. Early detection plays a significant role in administering personalized treatment and improving patient outcomes. Mammography procedures are often used to detect early-stage cancer cells. This traditional method of mammography while valuable has limitations of potential for false positives and negatives, patient discomfort, and radiation exposure. Therefore, there is a probe for more accurate techniques required in detecting breast cancer, leading to exploring the potential of machine learning in the classification of diagnostic images due to its efficiency and accuracy. This study conducted a comparative analysis of pre-trained CNNs (ResNet50 and VGG16) and Vision Transformers (ViT-Base and SWIN Transformer) with the inclusion of ViT-Base trained from scratch model architectures to effectively classify breast cancer mammographic images into benign and malignant cases. The Swin transformer exhibits superior performance with 99.8% accuracy and a precision of 99.8%. These findings demonstrate the efficiency of deep learning to accurately classify breast cancer mammographic images for the diagnosis of breast cancer, leading to improvement in patient outcomes.
Keywords
Breast cancer; Deep Learning; Transformer models
Subject
Computer Science and Mathematics, Artificial Intelligence and Machine Learning
Copyright:
This is an open access article distributed under the Creative Commons Attribution License which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.