Enhanced Lesion Classification Based on YOLO Architectures Using Thermal Breast Images on a Patient by Patient Basis


Creative Commons License

Çevik K. K., Çivilibal S., Bozkurt A., Dandıl E.

TRAITEMENT DU SIGNAL, vol.41, no.6, pp.2989-2999, 2024 (SCI-Expanded)

  • Publication Type: Article / Article
  • Volume: 41 Issue: 6
  • Publication Date: 2024
  • Doi Number: 10.18280/ts.410617
  • Journal Name: TRAITEMENT DU SIGNAL
  • Journal Indexes: Science Citation Index Expanded (SCI-EXPANDED), Scopus, PASCAL, Business Source Elite, Business Source Premier, Compendex, zbMATH
  • Page Numbers: pp.2989-2999
  • Akdeniz University Affiliated: Yes

Abstract

Breast cancer classification using deep learning architectures plays a crucial role in assisting clinicians with early-stage diagnosis. In this study, we present a comprehensive evaluation of YOLO architectures-YOLOv2, YOLOv3, YOLOv4, and YOLOv5-for the classification of breast lesions in thermal breast images. By employing these architectures, we enhanced the identification of relevant regions of interest (ROIs) for lesion contouring. The dataset for this research was sourced from a publicly available repository, and divided on a patient-by-patient basis. This patient-based split enhances the robustness and clinical relevance of the model’s performance compared to prior studies that relied on random data partitioning. Experimental results demonstrate that YOLOv5, trained with the Stochastic Gradient Descent with Momentum (SGDM) optimizer, achieved superior performance, with 0.83, 0.66, 0.97 and 0.79 for the key metrics of accuracy, precision, recall and F1-score, respectively. These results underscore the model’s potential for reliable breast lesion classification and emphasize the importance of robust dataset partitioning to enhance clinical applicability.