Deep Learning for Automatic Knee Osteoarthritis Severity Grading and Classification

Abstract

Introduction

Knee osteoarthritis (OA) is a prevalent condition that significantly impacts the quality of life, often leading to the need for knee replacement surgery. Accurate and timely identification of knee degeneration is crucial for effective treatment and management. Traditional methods of diagnosing OA rely heavily on radiological assessments, which can be time-consuming and subjective. This study aims to address these challenges by developing a deep learning-based method to predict the likelihood of knee replacement and the Kellgren–Lawrence (KL) grade of knee OA from X-ray images.

Methodology

We employed the Osteoarthritis Initiative (OAI) dataset and utilized a transfer learning approach with the Inception V3 architecture to enhance the accuracy of OA detection. Our approach involved training 14 different models—Xception, VGG16, VGG19, ResNet50, ResNet101, ResNet152, ResNet50V2, ResNet101V2, ResNet152V2, Inception V3, Inception, ResNetV2, DenseNet121, DenseNet169, DenseNet201—and comparing their performance.

Results

The study incorporated pixel ratio computation and picture pre-processing, alongside a decision tree model for prediction. Our experiments revealed that the Inception V3 model achieved the highest training accuracy of 91% and testing accuracy of 67%, with notable performance in both training and validation phases. This model effectively identified the presence and severity of OA, correlating with the Kellgren–Lawrence scale and facilitating the assessment of knee replacement needs.

Conclusion

By integrating advanced deep learning techniques with radiological diagnostics, our methodology supports radiologists in making more accurate and prompt decisions regarding knee degeneration. The Inception V3 model stands out as the optimal choice for knee X-ray analysis, contributing to more efficient and timely healthcare delivery for patients with knee osteoarthritis.