Automatic Localization and Brand Detection of Cervical Spine Hardware on Radiographs Using Weakly Supervised Machine Learning

Raman Dutt, Dylan Mendonca, Huai Ming Phen, Samuel Broida, Marzyeh Ghassemi, Judy Gichoya, Imon Banerjee, Tim Yoon, Hari Trivedi

Research output: Contribution to journalArticlepeer-review

Abstract

Purpose: To develop an end-to-end pipeline to localize and identify cervical spine hardware brands on routine cervical spine radiographs. Materials and Methods: In this single-center retrospective study, patients who received cervical spine implants between 2014 and 2018 were identified. Information on the implant model was retrieved from the surgical notes. The dataset was filtered for implants present in at least three patients, which yielded five anterior and five posterior hardware models for classification. Images for training were manually annotated with bounding boxes for anterior and posterior hardware. An object detection model was trained and implemented to localize hardware on the remaining images. An image classification model was then trained to differentiate between five anterior and five posterior hardware models. Model performance was evaluated on a holdout test set with 1000 iterations of bootstrapping. Results: A total of 984 patients (mean age, 62 years 6 12 [standard deviation]; 525 women) were included for model training, validation, and testing. The hardware localization model achieved an intersection over union of 86.8% and an F1 score of 94.9%. For brand classification, an F1 score, sensitivity, and specificity of 98.7% 6 0.5, 98.7% 6 0.5, and 99.2% 6 0.3, respectively, were attained for anterior hardware, with values of 93.5% 6 2.0, 92.6% 6 2.0, and 96.1% 6 2.0, respectively, attained for posterior hardware. Conclusion: The developed pipeline was able to accurately localize and classify brands of hardware implants using a weakly supervised learning framework.

Original languageEnglish (US)
Article numbere210099
JournalRadiology: Artificial Intelligence
Volume4
Issue number2
DOIs
StatePublished - Mar 2022

Keywords

  • Convolutional Neural Network
  • Deep Learning Algorithms
  • Machine Learning Algorithms
  • Prostheses
  • Semisupervised Learning
  • Spine

ASJC Scopus subject areas

  • Radiology Nuclear Medicine and imaging
  • Artificial Intelligence
  • Radiological and Ultrasound Technology

Fingerprint

Dive into the research topics of 'Automatic Localization and Brand Detection of Cervical Spine Hardware on Radiographs Using Weakly Supervised Machine Learning'. Together they form a unique fingerprint.

Cite this