Comparative analysis of machine learning and deep learning algorithms for knee arthritis detection using YOLOv8 models

Abstract

Knee arthritis is a prevalent joint condition that affects many people worldwide. Early detection and appropriate treatment are essential to slow the disease's progression and enhance patients’ quality of life. In this study, various machine learning and deep learning algorithms were used to detect knee arthritis. The machine learning models included k-NN, SVM, and GBM, while DenseNet, EfficientNet, and InceptionV3 were used as deep learning models. Additionally, YOLOv8 classification models (YOLOv8n-cls, YOLOv8s-cls, YOLOv8m-cls, YOLOv8l-cls, and YOLOv8x-cls) were employed. The “Annotated Dataset for Knee Arthritis Detection” with five classes (Normal, Doubtful, Mild, Moderate, Severe) and 1650 images were divided into 80% training, 10% validation, and 10% testing using the Hold-Out method. YOLOv8 models outperformed both machine learning and deep learning algorithms. k-NN, SVM, and GBM achieved success rates of 63.61%, 64.14%, and 67.36%, respectively. Among deep learning models, DenseNet, EfficientNet, and InceptionV3 achieved 62.35%, 70.59%, and 79.41%. The highest success was seen in the YOLOv8x-cls model at 86.96%, followed by YOLOv8l-cls at 86.79%, YOLOv8m-cls at 83.65%, YOLOv8s-cls at 80.37%, and YOLOv8n-cls at 77.91%.