Curriculum Learning-Driven YOLO for Tumor Detection in Ultrasound Using Hierarchically Zoomed-In Images
Source abstract
Abstract
Ultrasound imaging is widely employed for breast cancer detection; however, its diagnostic reliability is often constrained by operator dependence and subjective interpretation. Deep learning-based computer-aided diagnosis (CADx) systems offer potential to improve diagnostic consistency, yet their effectiveness is frequently limited by the scarcity of annotated medical images. This work introduces a training framework to enhance the performance and training stability of a YOLO-based object detection model for breast tumor localization, particularly in data-constrained scenarios. The proposed method integrates a detail-to-context curriculum learning scheme using hierarchically zoomed-in B-mode images, with progression difficulty determined by the tumor-to-background area ratio. A preprocessing step resizes all images to 640 × 640 pixels while preserving aspect ratio to improve intra-dataset consistency. Our evaluation indicates that aspect ratio-preserving resizing is associated with a 2.3% increase in recall and a reduction in the standard deviation of stability metrics by more than 20%. Moreover, the curriculum learning approach reached 97.2% of the final model performance using only 35% of the training data required by conventional methods, while achieving a more balanced precision–recall profile. These findings suggest that the proposed framework holds potential as an effective strategy for developing more robust and efficient tumor detection models, particularly for deployment in resource-limited clinical environments.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: COVID-19 diagnosis using AI · Brain Tumor Detection and Classification · Radiomics and Machine Learning in Medical Imaging
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Electrical Engineering, Kwangwoon University, Seoul 01897, Republic of Korea
Department of Mechanical Engineering, Clemson University, Clemson, SC 29634, USA
Funding
Institute of Information & communications Technology Planning & Evaluation (IITP)
References (25)
Show source-provided references
1. Global Cancer Statistics 2020: GLOBOCAN Estimates of Incidence and Mortality Worldwide for 36 Cancers in 185 Countries
2. Breast Cancer Statistics, 2022
3. Breast ultrasound: Recommendations for information to women and referring physicians by the European Society of Breast Imaging10.1007/s13244-018-0636-z ↗
4. Breast imaging reporting and data system lexicon for US: Interobserver agreement for assessment of breast masses10.1148/radiol.2523080670 ↗
5. Anniversary paper: History and status of CAD and quantitative image analysis: The role of Medical Physics and AAPM10.1118/1.3013555 ↗
6. Deep learning in medical image analysis10.1146/annurev-bioeng-071516-044442 ↗
7. Breast mass classification in sonography with transfer learning using a deep convolutional neural network and color conversion10.1002/mp.13361 ↗
8. Wei, K., Wang, B., and Saniie, J. (2020, January 10–21). Faster Region Convolutional Neural Networks Applied to Ultrasonic Images for Breast Lesion Detection and Classification. Proceedings of the 2020 IEEE International Symposium on Circuits and Systems (ISCAS), Seville, Spain.10.1109/EIT48999.2020.9208264 ↗
9. Deep learning-based breast cancer classification through medical imaging modalities: State of the art and research challenges
10. Bengio, Y., Louradour, J., Collobert, R., and Weston, J. (2009, January 14–18). Curriculum Learning. Proceedings of the 26th Annual International Conference on Machine Learning (ICML 2009), Montreal, QC, Canada.10.1145/1553374.1553380 ↗
11. A survey on curriculum learning
12. Jesson, A., Guizard, N., Ghalehjegh, S.H., Goblot, D., Soudan, F., and Chapados, N. (2017, January 10–14). CASED: Curriculum Adaptive Sampling for Extreme Data Imbalance. Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, Quebec City, QC, Canada.10.1007/978-3-319-66179-7_73 ↗
13. Curriculum learning for medical image analysis: A survey
14. Loizidou, K., Elia, R., and Pitris, C. (2023). Computer-Aided Breast Cancer Detection and Classification in Mammography: A Comprehensive Review. Comput. Biol. Med., 153.10.1016/j.compbiomed.2023.106554 ↗
15. Li, Y., Wu, Z., Chen, X., and Wang, L. (2023, January 18–22). From Easy to Hard: A Dual Curriculum Learning Framework for Context-Aware Document Layout Analysis. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada.
16. Redmon, J., Divvala, S., Girshick, R., and Farhadi, A. (2016, January 27–30). You Only Look Once: Unified, Real-Time Object Detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.10.1109/CVPR.2016.91 ↗
17. Liu, H., Sun, F., Gu, J., and Deng, L. (2022). SF-YOLOv5: A Lightweight Small Object Detection Algorithm Based on Improved Feature Fusion Mode. Sensors, 22.10.3390/s22155817 ↗
18. Li, C., Li, L., Jiang, H., Weng, K., Geng, Y., Li, L., Ke, Z., Li, Q., Cheng, M., and Nie, W. (2022). YOLOv6: A single-stage object detection framework for industrial applications. arXiv.
19. Wang, C.-Y., Bochkovskiy, A., and Liao, H.-Y.M. (2023, January 18–22). YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada.10.1109/CVPR52729.2023.00721 ↗
20. Terven, J., and Córdova-Esparza, D.-M. (2023). A Comprehensive Review of YOLO Architectures in Computer Vision: From YOLOv1 to YOLOv8 and YOLO-NAS. arXiv.10.3390/make5040083 ↗
21. Open access database of raw ultrasonic signals acquired from malignant and benign breast lesions10.1002/mp.12538 ↗
22. Weinshall, I.D., Cohen, G., and Amir, D. (2018, January 10–15). On the Power of Curriculum Learning in Training Deep Networks. Proceedings of the 35th International Conference on Machine Learning (ICML), Stockholm, Sweden.
23. Bochkovskiy, A., Wang, C.-Y., and Liao, H.-Y.M. (2020). YOLOv4: Optimal speed and accuracy of object detection. arXiv.
24. The PASCAL Visual Object Classes (VOC) Challenge10.1007/s11263-009-0275-4 ↗
25. Shen, L., Margolies, L.R., Rothstein, J.H., Fluder, E., McBride, R., and Sieh, W. (2019). Deep learning to improve breast cancer detection on screening mammography. Sci. Rep., 9.10.1038/s41598-019-48995-4 ↗