DeepKneeExplainer: Explainable Knee Osteoarthritis Diagnosis From Radiographs and Magnetic Resonance Imaging

Osteoarthritis (OA) is a degenerative joint disease, which significantly affects middle-aged and elderly people. Although primarily identified via hyaline cartilage change based on medical images, technical bottlenecks like noise, artifacts, and modality impose an enormous challenge on high-precision, objective, and efficient early quantification of OA. Owing to recent advancements, approaches based on neural networks (DNNs) have shown outstanding success in this application domain. However, due to nested non-linear and complex structures, DNNs are mostly opaque and perceived as black-box methods, which raises numerous legal and ethical concerns. Moreover, these approaches do not have the ability to provide the reasoning behind diagnosis decisions in the way humans would do, which poses an additional risk in the clinical setting. In this paper, we propose a novel explainable method for knee OA diagnosis based on radiographs and magnetic resonance imaging (MRI), which we called DeepKneeExplainer. First, we comprehensively preprocess MRIs and radiographs through the deep-stacked transformation technique against possible noises and artifacts that could contain unseen images for domain generalization. Then, we extract the region of interests (ROIs) by employing U-Net architecture with ResNet backbone. To classify the cohorts, we train DenseNet and VGG architectures on the extracted ROIs. Finally, we highlight class-discriminating regions using gradient-guided class activation maps (Grad-CAM++) and layer-wise relevance propagation (LRP), followed by providing human-interpretable explanations of the predictions. Comprehensive experiments based on the multicenter osteoarthritis study (MOST) cohorts, our approach yields up to 91% classification accuracy, outperforming comparable state-of-the-art approaches. We hope that our results will encourage medical researchers and developers to adopt explainable methods and DNN-based analytic pipelines towards an increasing acceptance and adoption of AI-assisted applications in the clinical practice for improved knee OA diagnoses.

[1]  Till Döhmen,et al.  DeepCOVIDExplainer: Explainable COVID-19 Diagnosis Based on Chest X-ray Images , 2020 .

[2]  Daguang Xu,et al.  Generalizing Deep Learning for Medical Image Segmentation to Unseen Domains via Deep Stacked Transformation , 2020, IEEE Transactions on Medical Imaging.

[3]  Gyogwon Koo,et al.  Extending Class Activation Mapping Using Gaussian Receptive Field , 2020, Comput. Vis. Image Underst..

[4]  Stefan Decker,et al.  A snapshot neural ensemble method for cancer-type prediction based on copy number variations , 2019, Neural Computing and Applications.

[5]  Brian Kenji Iwana,et al.  Explaining Convolutional Neural Networks using Softmax Gradient Layer-wise Relevance Propagation , 2019, 2019 IEEE/CVF International Conference on Computer Vision Workshop (ICCVW).

[6]  Simo Saarakkala,et al.  Multimodal Machine Learning-based Knee Osteoarthritis Progression Prediction from Plain Radiographs and Clinical Data , 2019, Scientific Reports.

[7]  Michael W. Mahoney,et al.  Traditional and Heavy-Tailed Self Regularization in Neural Network Models , 2019, ICML.

[8]  Mohammad Mansouri,et al.  An explainable deep-learning algorithm for the detection of acute intracranial haemorrhage from small datasets , 2018, Nature Biomedical Engineering.

[9]  Volker Tresp,et al.  Understanding Individual Decisions of CNNs via Contrastive Backpropagation , 2018, ACCV.

[10]  Thomas M. Link,et al.  Applying Densely Connected Convolutional Neural Networks for Staging Osteoarthritis Severity from Plain Radiographs , 2018, Journal of Digital Imaging.

[11]  Jianhua Yin,et al.  Fourier transform infrared spectroscopic imaging application for multi-stage discrimination in cartilage degeneration , 2018, Infrared Physics & Technology.

[12]  Yang Zhang,et al.  A Theoretical Explanation for Perplexing Behaviors of Backpropagation-based Visualizations , 2018, ICML.

[13]  S Zachow,et al.  Knee menisci segmentation using convolutional neural networks: data from the Osteoarthritis Initiative. , 2018, Osteoarthritis and cartilage.

[14]  Abdelbasset Brahim,et al.  Texture analysis using complex wavelet decomposition for knee osteoarthritis detection: Data from the osteoarthritis initiative , 2018, Comput. Electr. Eng..

[15]  S. Delp,et al.  Automated staging of knee osteoarthritis severity using deep neural networks , 2018 .

[16]  A. A. Gatti,et al.  NEURALSEG: state-of-the-art cartilage segmentation using deep learning – analyses of data from the osteoarthritis initiative , 2018 .

[17]  N. Hafezi-Nejad,et al.  Osteoarthritis year in review 2017: updates on imaging advancements. , 2018, Osteoarthritis and cartilage.

[18]  Dawid Baczkowicz,et al.  Analysis and multiclass classification of pathological knee joints using vibroarthrographic signals , 2018, Comput. Methods Programs Biomed..

[19]  Mauridhi Hery Purnomo,et al.  Osteoarthritis Severity Determination using Self Organizing Map Based Gabor Kernel , 2018 .

[20]  Anand A. Joshi,et al.  WRIST: A WRist Image Segmentation Toolkit for carpal bone delineation from MRI , 2017, Comput. Medical Imaging Graph..

[21]  Chee Peng Lim,et al.  Medical image analysis using wavelet transform and deep belief networks , 2017, Expert Syst. Appl..

[22]  Andrew Y. Ng,et al.  CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning , 2017, ArXiv.

[23]  Artjoms Suponenkovs,et al.  Knee-joint tissue recognition in magnetic resonance imaging , 2017, 2017 IEEE 30th Neumann Colloquium (NC).

[24]  Ming Zhang,et al.  Knee osteoarthritis prediction on MR images using cartilage damage index and machine learning methods , 2017, 2017 IEEE International Conference on Bioinformatics and Biomedicine (BIBM).

[25]  Anirban Sarkar,et al.  Grad-CAM++: Generalized Gradient-Based Visual Explanations for Deep Convolutional Networks , 2017, 2018 IEEE Winter Conference on Applications of Computer Vision (WACV).

[26]  Simo Saarakkala,et al.  Automatic Knee Osteoarthritis Diagnosis from Plain Radiographs: A Deep Learning-Based Approach , 2017, Scientific Reports.

[27]  Shadpour Demehri,et al.  New imaging modalities to predict and evaluate osteoarthritis progression. , 2017, Best practice & research. Clinical rheumatology.

[28]  Le Lu,et al.  ChestX-Ray8: Hospital-Scale Chest X-Ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseases , 2017, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[29]  Kilian Q. Weinberger,et al.  Snapshot Ensembles: Train 1, get M for free , 2017, ICLR.

[30]  Telajala Venkata Mahendra,et al.  Image enhancement with the application of local and global enhancement methods for dark images , 2017, 2017 International Conference on Innovations in Electronics, Signal Processing and Communication (IESC).

[31]  Noel E. O'Connor,et al.  Automatic Detection of Knee Joints and Quantification of Knee Osteoarthritis Severity Using Convolutional Neural Networks , 2017, MLDM.

[32]  D. P. Acharjya,et al.  Clustering Algorithm in Possibilistic Exponential Fuzzy C-Mean Segmenting Medical Images , 2017 .

[33]  Timothy F. Cootes,et al.  Fully automated shape analysis for detection of Osteoarthritis from lateral knee radiographs , 2016, 2016 23rd International Conference on Pattern Recognition (ICPR).

[34]  Zhuowen Tu,et al.  Aggregated Residual Transformations for Deep Neural Networks , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[35]  Ashutosh Kumar Singh,et al.  Global, regional, and national incidence, prevalence, and years lived with disability for 310 diseases and injuries, 1990–2015: a systematic analysis for the Global Burden of Disease Study 2015 , 2016, Lancet.

[36]  Ramprasaath R. Selvaraju,et al.  Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization , 2016, International Journal of Computer Vision.

[37]  Sanjay Talbar,et al.  A generalized contrast enhancement approach for knee MR images , 2016, 2016 International Conference on Signal and Information Processing (IConSIP).

[38]  Joseph Antony,et al.  Quantifying radiographic knee osteoarthritis severity using deep convolutional neural networks , 2016, 2016 23rd International Conference on Pattern Recognition (ICPR).

[39]  Kilian Q. Weinberger,et al.  Densely Connected Convolutional Networks , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[40]  Tao Zhang,et al.  Bootstrapping Face Detection with Hard Negative Examples , 2016, ArXiv.

[41]  A. Iagnocco,et al.  Imaging of osteoarthritis (OA): What is new? , 2016, Best practice & research. Clinical rheumatology.

[42]  Simo Saarakkala,et al.  Association between subchondral bone structure and osteoarthritis histopathological grade , 2016, Journal of orthopaedic research : official publication of the Orthopaedic Research Society.

[43]  D. Robertson,et al.  Machine learning based analytics of micro-MRI trabecular bone microarchitecture and texture in type 1 Gaucher disease. , 2016, Journal of biomechanics.

[44]  A. K. Jayanthy,et al.  Classification of MRI images in 2D coronal view and measurement of articular cartilage thickness for early detection of knee osteoarthritis , 2016, 2016 IEEE International Conference on Recent Trends in Electronics, Information & Communication Technology (RTEICT).

[45]  Stuart Crozier,et al.  Automated segmentation and T2-mapping of the posterior cruciate ligament from MRI of the knee: Data from the osteoarthritis initiative , 2016, 2016 IEEE 13th International Symposium on Biomedical Imaging (ISBI).

[46]  Phi Vu Tran,et al.  A Fully Convolutional Neural Network for Cardiac Segmentation in Short-Axis MRI , 2016, ArXiv.

[47]  A. Sassoon,et al.  Classifications in Brief: Kellgren-Lawrence Classification of Osteoarthritis , 2016, Clinical orthopaedics and related research.

[48]  Bolei Zhou,et al.  Learning Deep Features for Discriminative Localization , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[49]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[50]  K. Müller,et al.  Explaining nonlinear classification decisions with deep Taylor decomposition , 2015, Pattern Recognit..

[51]  Kouki Nagamune,et al.  Computer-Aided Segmentation System of Posterior Cruciate Ligament in Knee Joint from CT and MRI Using Anatomical Information: A Pilot Study of System Configuration , 2015, 2015 IEEE International Conference on Systems, Man, and Cybernetics.

[52]  Alexander Binder,et al.  On Pixel-Wise Explanations for Non-Linear Classifier Decisions by Layer-Wise Relevance Propagation , 2015, PloS one.

[53]  Kaiming He,et al.  Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[54]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[55]  Dimitris N. Metaxas,et al.  Automated anatomical landmark detection ondistal femur surface using convolutional neural network , 2015, 2015 IEEE 12th International Symposium on Biomedical Imaging (ISBI).

[56]  Jaume Puig-Junoy,et al.  Socio-economic costs of osteoarthritis: a systematic review of cost-of-illness studies. , 2015, Seminars in arthritis and rheumatism.

[57]  Hongtu Zhu,et al.  Diseased Region Detection of Longitudinal Knee Magnetic Resonance Imaging Data , 2015, IEEE Transactions on Medical Imaging.

[58]  Prashant Dahiwale,et al.  A combined effect of local and global method for contrast image enhancement , 2015, 2015 IEEE International Conference on Engineering and Technology (ICETECH).

[59]  Trevor Darrell,et al.  Fully convolutional networks for semantic segmentation , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[60]  Rati Khandelwal,et al.  Various Image Enhancement Techniques - A Critical Review , 2014 .

[61]  P. Conaghan,et al.  The role of imaging modalities in the diagnosis, differential diagnosis and clinical assessment of peripheral joint osteoarthritis. , 2014, Osteoarthritis and cartilage.

[62]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[63]  Pooja P. Kawathekar,et al.  Severity analysis of Osteoarthritis of knee joint from X-ray images: A Literature review , 2014, 2014 International Conference on Signal Propagation and Computer Technology (ICSPCT 2014).

[64]  Jian Sun,et al.  Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition , 2014, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[65]  Subir Singh Lamba,et al.  Modified Histogram based contrast enhancement using Homomorphic Filtering for medical images , 2014, 2014 IEEE International Advance Computing Conference (IACC).

[66]  Neil A Segal,et al.  The Multicenter Osteoarthritis Study: Opportunities for Rehabilitation Research , 2013, PM & R : the journal of injury, function, and rehabilitation.

[67]  Deok Won Kim,et al.  Interpretation of movement during stair ascent for predicting severity and prognosis of knee osteoarthritis in elderly women using support vector machine , 2013, 2013 35th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[68]  V. Rajini,et al.  Local Binary Pattern Approach to the Classification of Osteoarthritis in Knee X-ray Images , 2013 .

[69]  I Ketut Eddy Purnama,et al.  Osteoarthritis Classification Using Self Organizing Map Based on Gabor Kernel and Contrast-Limited Adaptive Histogram Equalization , 2013, The open biomedical engineering journal.

[70]  S. Kubakaddi,et al.  Measurement of cartilage thickness for early detection of knee osteoarthritis (KOA) , 2013, 2013 IEEE Point-of-Care Healthcare Technologies (PHT).

[71]  Garry E Gold,et al.  Diagnosis of osteoarthritis: imaging. , 2012, Bone.

[72]  N. Theumann,et al.  [Imaging in osteoarthritis]. , 2012, Revue medicale suisse.

[73]  A. Hani,et al.  Development of a non-invasive diagnostic tool for early detection of knee osteoarhritis , 2011, 2011 National Postgraduate Conference.

[74]  M. Nevitt,et al.  Comparison of BLOKS and WORMS scoring systems part I. Cross sectional comparison of methods to assess cartilage morphology, meniscal damage and bone marrow lesions on knee MRI: data from the osteoarthritis initiative. , 2010, Osteoarthritis and cartilage.

[75]  Scott D. Martin,et al.  In vivo micron scale arthroscopic imaging of human knee osteoarthritis with OCT: Comparison with MRI and arthroscopy , 2009, 2009 Conference on Lasers and Electro-Optics and 2009 Conference on Quantum electronics and Laser Science Conference.

[76]  Abd. Rahman Ramli,et al.  Contrast enhancement using recursive mean-separate histogram equalization for scalable brightness preservation , 2003, IEEE Trans. Consumer Electron..

[77]  Charles G. Peterfy,et al.  MR findings in knee osteoarthritis , 2003, European Radiology.

[78]  Giovanni Ramponi,et al.  Image enhancement via adaptive unsharp masking , 2000, IEEE Trans. Image Process..

[79]  P. Wolf,et al.  The effects of specific medical conditions on the functional limitations of elders in the Framingham Study. , 1994, American journal of public health.

[80]  T D Cooke,et al.  Radiographic assessment of progression in osteoarthritis. , 1987, Arthritis and rheumatism.

[81]  Patrick Thiam,et al.  Multimodal Deep Denoising Convolutional Autoencoders for Pain Intensity Classification based on Physiological Signals , 2020, ICPRAM.

[82]  D. P. Acharjya,et al.  Segmentation and Feature Extraction in Medical Imaging: A Systematic Review , 2020 .

[83]  P. Victer Paul,et al.  A Survey on Predictive Models of Learning Analytics , 2020 .

[84]  V. Kamalaveni,et al.  Image Denoising Using Variations of Perona-Malik Model with Different Edge Stopping Functions☆ , 2015 .

[85]  Linda G. Shapiro IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, VOL. 12. NO. 7. JULY 1990 Scale-Space and Edge Detection Using Anisotropic Diffusion , 2022 .