A comparative study of YOLO models and a transformer-based YOLOv5 model for mass detection in mammograms

Abstract

Breast cancer is a prevalent form of cancer across the globe, and if it is not diagnosed at an early stage it can be life-threatening. In order to aid in its diagnosis, detection, and classification, computer-aided detection (CAD) systems are employed. You Only Look Once (YOLO)-based CAD algorithms have become very popular owing to their highly accurate results for object detection tasks in recent years. Therefore, the most popular YOLO models are implemented to compare the performance in mass detection with various experiments on the INbreast dataset. In addition, a YOLO model with an integrated Swin Transformer in its backbone is proposed for mass detection in mammography images within the study. The performance of YOLOv5 models and a transformer-based YOLO model is compared to that of each other and YOLOv3 and YOLOv4 models using images with different sizes on the INbreast dataset. The best results are obtained by the transformer-based YOLO model of YOLOv5 for 832 × 832 image size. In another experiment, we compared the default anchors against the anchors provided by the YOLOv5 autoanchor function before training and saw that the anchors generated by the YOLOv5 autoanchor increased the success rates. Furthermore, various experiments were conducted to observe how data augmentation affects performance. Although a small amount of data was used in the study, high performance was obtained by YOLO algorithms, which are promising tools for cancer detection.