A Robust Framework for Severity Detection of Knee Osteoarthritis Using an Efficient Deep Learning Model

Abstract

With the changing lifestyle, a large population suffers from a bone disease known as an osteoarthritis affecting the knee, spine, and hip. Therefore, timely detection and classification of the disease are necessary to minimize the loss, however, it is a time-consuming task and requires various tests and physicians’ in-depth analysis. Thus, an accurate automated technique, timely detection and classification are needed to cope with the aforementioned challenges. This study proposes a technique based on an efficient DenseNet that uses the knee image’ features to identify the Knee Osteoarthritis (KOA) and determine its severity level according to the KL grading system such as Grade-I, Grade-II, Grade-III, and Grade-IV. We introduced the reweighted cross-entropy loss function which makes our proposed algorithm more robust as the training data is imbalanced. The dense connections of efficient DenseNet with regularization power help to reduce the overfitting during the training of small knee sample training sets. The proposed algorithm is an efficient approach that can identify the early symptoms of KOA and classify the severity level of the disease for better decision making by orthopedics. The algorithm is a pre-trained network that does not require a huge training set, therefore, the existing dataset i.e. Mendeley VI has been utilized for the training and testing. Additionally, cross-validation has been employed using the OAI dataset to assess the performance of the proposed model. The algorithm achieved 98.22% accuracy over the testing set and 98.08% accuracy over cross-validation. Various experiments have been performed to confirm that our proposed algorithm is more consistent and capable of detecting and classifying the KOA disease than existing state of the art.