Attention-based CNN for KL Grade Classification: Data from the Osteoarthritis Initiative

Bofei Zhang, Jimin Tan, Kyunghyun Cho, Gregory Chang, Cem M. Deniz

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Knee osteoarthritis (OA) is a chronic degenerative disorder of joints and it is the most common reason leading to total knee joint replacement. Diagnosis of OA involves subjective judgment on symptoms, medical history, and radiographic readings using Kellgren-Lawrence grade (KL-grade). Deep learning-based methods such as Convolution Neural Networks (CNN) have recently been applied to automatically diagnose radiographic knee OA. In this study, we applied Residual Neural Network (ResNet) to first detect knee joint from radiographs and later combine ResNet with Convolutional Block Attention Module (CBAM) to make a prediction of the KL-grade automatically. The proposed model achieved a multi-class average accuracy of 74.81%, mean squared error of 0.36, and quadratic Kappa score of 0.88, which demonstrates a significant improvement over the published results. The attention maps were analyzed to provide insights on the decision process of the proposed model11Code is available at https://github.com/denizlab/OAI-KL-Grade-Classification.

Original languageEnglish (US)
Title of host publicationISBI 2020 - 2020 IEEE International Symposium on Biomedical Imaging
PublisherIEEE Computer Society
Pages731-735
Number of pages5
ISBN (Electronic)9781538693308
DOIs
StatePublished - Apr 2020
Event17th IEEE International Symposium on Biomedical Imaging, ISBI 2020 - Iowa City, United States
Duration: Apr 3 2020Apr 7 2020

Publication series

NameProceedings - International Symposium on Biomedical Imaging
Volume2020-April
ISSN (Print)1945-7928
ISSN (Electronic)1945-8452

Conference

Conference17th IEEE International Symposium on Biomedical Imaging, ISBI 2020
Country/TerritoryUnited States
CityIowa City
Period4/3/204/7/20

Keywords

  • convolutional neural networks
  • deep learning
  • knee
  • osteoarthritis
  • radiography

ASJC Scopus subject areas

  • Biomedical Engineering
  • Radiology Nuclear Medicine and imaging

Fingerprint

Dive into the research topics of 'Attention-based CNN for KL Grade Classification: Data from the Osteoarthritis Initiative'. Together they form a unique fingerprint.

Cite this