A quantitative study in the performance of YOLOv8 versus Vision Transformers for breast cancer classification
2025 (English)Independent thesis Basic level (degree of Bachelor), 10 credits / 15 HE credits
Student thesisAlternative title
En kvantitativ studie om prestandan hos YOLOv8 jämfört med Vision Transformers för bröstcancer klassificering (Swedish)
Abstract [en]
This paper explores two different paradigms in evolving AI technologies, those being Vision Transformers(ViT) and YOLOv8, in the context of breast cancer detection of mammography images. Although Convolutional Neural Networks (CNNs) have seen widespread usage in the field of Computer Aided Diagnosis (CAD), whilst the ViT models being compared with YOLOv8 models has largely gone unexplored. To gather results and analyze them we utilize several python libraries such as Ultralytics and Transformer to either train or fine-tune our models, we employ a dataset with a validation split of 20% and 80% training split. The results are gathered through a confusion matrix and results graphs demonstrating the models loss over epochs. This paper fills a gap in current research by directly comparing YOLOv8 and ViT’s, offering insights for future CAD systems in breast cancer detection. This thesis find that the Vision transformer outperform YOLOv8 in terms of overall accuracy, with a score 84.0% and the YOLOv8 model had an overall accuracy of 82.7%, but that the YOLOv8 has more consistent performance across the three different classes.
Place, publisher, year, edition, pages
2025. , p. 31
Keywords [en]
Vision Transformers, YOLOv8, Breast Cancer Classification, Computer Aided Diagnosis, CAD, ViT, AI
National Category
Computer Systems
Identifiers
URN: urn:nbn:se:hv:diva-23654Local ID: EHD500OAI: oai:DiVA.org:hv-23654DiVA, id: diva2:1978264
Subject / course
Computer engineering
Educational program
Datateknik - högskoleingenjör
Supervisors
Examiners
2025-07-222025-06-272025-09-30Bibliographically approved