Browse > Article
http://dx.doi.org/10.7742/jksr.2022.16.1.25

Performance Evaluation of YOLOv5s for Brain Hemorrhage Detection Using Computed Tomography Images  

Kim, Sungmin (Department of Radiological Science, Konyang University)
Lee, Seungwan (Department of Radiological Science, Konyang University)
Publication Information
Journal of the Korean Society of Radiology / v.16, no.1, 2022 , pp. 25-34 More about this Journal
Abstract
Brain computed tomography (CT) is useful for brain lesion diagnosis, such as brain hemorrhage, due to non-invasive methodology, 3-dimensional image provision, low radiation dose. However, there has been numerous misdiagnosis owing to a lack of radiologist and heavy workload. Recently, object detection technologies based on artificial intelligence have been developed in order to overcome the limitations of traditional diagnosis. In this study, the applicability of a deep learning-based YOLOv5s model was evaluated for brain hemorrhage detection using brain CT images. Also, the effect of hyperparameters in the trained YOLOv5s model was analyzed. The YOLOv5s model consisted of backbone, neck and output modules. The trained model was able to detect a region of brain hemorrhage and provide the information of the region. The YOLOv5s model was trained with various activation functions, optimizer functions, loss functions and epochs, and the performance of the trained model was evaluated in terms of brain hemorrhage detection accuracy and training time. The results showed that the trained YOLOv5s model is able to provide a bounding box for a region of brain hemorrhage and the accuracy of the corresponding box. The performance of the YOLOv5s model was improved by using the mish activation function, the stochastic gradient descent (SGD) optimizer function and the completed intersection over union (CIoU) loss function. Also, the accuracy and training time of the YOLOv5s model increased with the number of epochs. Therefore, the YOLOv5s model is suitable for brain hemorrhage detection using brain CT images, and the performance of the model can be maximized by using appropriate hyperparameters.
Keywords
YOLOv5s; Brain hemorrhage detection; Hyperparameter; Computed tomography image;
Citations & Related Records
Times Cited By KSCI : 1  (Citation Analysis)
연도 인용수 순위
1 Statistics Korea, Mortality Trend by Cause of Death, 2021.
2 H. Eskandary, M. Sabba, F. Khajehpour, M. Eskandari, "Incidental findings in brain computed tomography scans of 3000 head trauma patients", Surgical Neurology, Vol. 63, No. 6, pp. 550-553, 2005. http://dx.doi.org/10.1016/j.surneu.2004.07.049   DOI
3 M. H. Choi, H. Eo, S. E. Jung, et al., "Teleradiology of Koreain 2017: A Questionnaireto Members of The Korean Society of Radiology", Journal of the Korean Society of Radiology, Vol. 80, No. 4, pp. 684-803, 2019. https://doi.org/10.3348/jksr.2019.80.4.684   DOI
4 2016~2019 Number of Radiation Examinations and Exposure Dose for Diagnosis, Korean Disease Control and Prevention Agency, 2021.
5 2020 Health and Welfare Statistical Yearbook, Ministry of Health and Welfare, 2020.
6 A. R. Pathak, M. Pandey, S. Rautaray, "Application of Deep Learning for Object Detection", Procedia Computer Science, Vol. 132, pp. 1706-1717, 2018. http://dx.doi.org/10.1016/j.procs.2018.05.144   DOI
7 J. Redmon, S. Divvala, R. Girshick, A. Farhadi, "You Only Look Once: Unified, Real-Time Object Detection", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 779-788, arXiv:1506.02640v5, 2016.
8 J. Bergstra, Y. Bengio, "Random search for Hyper-Parameter Optimization", Journal of Machine Learning Research, Vol. 13, No. 2, pp. 281-305, 2012.
9 J. Revaud, J. Almazan, R. S. Rezende, et al., "Learning with Average Precision: Training Image Retrieval with a Listwise Loss", Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 5107-5116, 2019.
10 D. Wang, D. He, "Channel pruned YOLO V5s-based deep learning approach for rapid and accurate apple fruitlet detection before fruit thinning", Biosystems Engineering, Vol. 210, pp. 271-281, 2021. http://dx.doi.org/10.1016/j.biosystemseng.2021.08.015   DOI
11 A. Mohiyuddin, A. Basharat, U. Ghani, et al., "Breast Tumor Detection and Classification in Mammogram Images Using Modified YOLOv5 Network", Computational and Mathematical Methods in Medicine, Vol. 2022, 2022. http://dx.doi.org/10.1155/2022/1359019   DOI
12 A. E. Flanders, L. M. Prevedello, G. Shih, et al., "Construction of a Machine Learning Dataset through Collaboration: The RSNA 2019 Brain CT Hemorrhage Challenge", Radiology: Artificial Intelligence, Vol. 2, No. 2, pp. e190211, 2020. https://doi.org/10.1148/ryai.2020190211   DOI
13 L. Yang, A. Shami, "On Hyperparam eter Optimization of Machine Learning Algorithms: Theory and Practice", Neurocomputing, Vol. 415, No. 20, pp. 295-316, 2020.   DOI
14 C. Shorten, T. M. Khoshgoftaar, "A survey on Image Data Augmentation for Deep Learning", Journal of Big Data, Vol. 6, No. 1, pp. 1-48, 2019. http://dx.doi.org/10.1186/s40537-019-0197-0   DOI
15 A. Howard, M. Sandler, G. Chu, et al., "Searching for MobileNetV3", Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 1314-1324, 2019.
16 S. Ruder, "An overview of gradient descent optimization algorithms", arXiv:1609.04747, 2016.
17 D. Kingma, J. Ba, "Adam: A method for stochastic optimization", Proceedings of the 3rd International Conference for Learning Representations, arXiv:1412.6980, 2014.
18 S. Elfwing, E. Uchibe, K. Doya, "Sigmoid-Weighted Linear Units for Neural Network Function", Neural Networks, Vol. 107, pp. 3-11, 2018.   DOI
19 D. Misra, "Mish: A Self Regularized Non-Monotonic Neural Activation Function", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, arXiv:1908.08681v2, 2020.
20 Z. Zheng, P. Wang, W. Liu, et al., "Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression", Proceedings of the AAAI Conference on Artificial Intelligence. Vol. 34, No. 7, pp. 12993-13000, 2020.
21 Y. Li, X. He, "COVID-19 Detection in Chest Radiograph Based on YOLO v5", 2021 IEEE International Conference on Computer Science, Electronic Information Engineering and Intelligent Control Technology, pp. 344-347, 2021.
22 Y.-H. Lee, Y. Kim "Comparison of CNN and YOLO for Object Detection", Journal of the Semiconductor & Display Technology, Vol. 19, No. 1, pp. 85-92, 2020.
23 A. Khan, A. Sohail, U. Zahoora, A. S. Qureshi, "A Survey of the Recent Architectures of Deep Convolutional Neural Networks", Artificial Intelligence Review, Vol. 53, No. 8, pp. 5455-5516, 2020. http://dx.doi.org/10.1007/s10462-020-09825-6   DOI
24 A. Bochkovskiy, C. Y. Wang, H. Y. M. Liao, "YOLOv4: Optimal Speed and Accuracy of Object Detection", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, arXiv:2004.10934, 2020.