FS-YOLOv9: A Frequency and Spatial Feature-Based YOLOv9 for Real-time Breast Cancer Detection

Haitian Gui,Tao Su,Xinhua Jiang,Li Li,Lang Xiong,Ji Zhou,Zhiyong Pang
DOI: https://doi.org/10.1016/j.acra.2024.09.048
2024-10-14
Abstract:Rationale and objectives: Breast cancer screening is critical for reducing mortality rates. YOLOv9, a new real-time object-detection model, is ideal for cancer screening. A customized YOLOv9 model with enhancements for detecting breast cancer on the basis of species and morphological diversity has potential clinical significance. Materials and methods: The internal dataset consisted of 687 cases split 3:1 for cross-validation. Additionally, 98 cases from external datasets were used for testing. We developed an FS-YOLOv9 model customized for breast cancer detection that incorporated an extra max-pooling layer before the Conv1 of the Adown to enhance high-brightness features. The Adown of the P3 in the backbone was replaced with a high-frequency Haar wavelet convolution kernel, which ignored the low-frequency components during down-sampling to enhance morphology and texture features. The reliability and robustness of our model was determined by measuring the F1 score, the area under curve of free-response receiver operating characteristic (FAUC), mean average precision (mAP), recall, and precision, and comparing them with the findings for the official YOLOv9, YOLOv8, YOLOv5 models. Results: In comparison with the official YOLOv9 model, FS-YOLOv9 showed a higher average F1 score (0.700 vs. 0.669), FAUC (0.695 vs. 0.662), and mAP50 (0.713 vs. 0.679) in the internal dataset; in the external testing dataset, the FS-YOLOv9 improved the average F1 score, FAUC, and mAP50 by 4.58%, 5.78%, and 4.41% respectively. Conclusion: Our FS-YOLOv9 model showed significantly improved performance in detecting breast cancer, making it more practical for high-risk breast cancer diagnosis.
What problem does this paper attempt to address?