Transfer Learning-Based Smart Features Engineering for Osteoarthritis Diagnosis From Knee X-Ray Images

Abstract

Osteoarthritis is a deteriorating joint disease affecting millions worldwide. Osteoarthritis is a chronic condition that develops over time due to joint wear and tears. The degeneration of joint cartilage is the underlying cause of osteoarthritis, resulting in bone-to-bone contact and contributing to stiffness, discomfort, and restricted movement. People with osteoarthritis struggle to perform simple tasks such as walking, standing, or climbing stairs. Moreover, osteoarthritis can also cause psychological distress, including depression and anxiety, due to the chronic pain and disability associated with the condition. Improving the quality of life requires the development of efficient methods for early detection. Our study aims to create a model that can effectively diagnose osteoarthritis in knee X-ray images at an early stage. The advanced deep learning-based Convolutional Neural Network (CNN) and several machine learning-based techniques are applied in comparison. A novel transfer learning-based feature engineering technique CRK (CNN Random forest K-neighbors) is proposed to detect osteoarthritis with high performance. Using a 2D-CNN, the proposed CRK smartly extracts the spatial features from the X-ray images. The spatial features are input to the random forest and k-neighbors techniques, creating a probabilistic feature set. The probabilistic feature set is utilized to build the applied machine learning-based techniques. Extensive study experiments demonstrate that the proposed model outperformed with a 99% accuracy score for predicting osteoarthritis. The performance of each applied model is validated using hyperparameter optimization and k-fold-based cross-validation. The proposed study has the potential to revolutionize the prediction of osteoarthritis from X-ray images with high-performance scores.

Keywords

• Osteoarthritis
• X-ray images
• knee X-ray
• transfer learning
• smart feature engineering
• deep learning