Anatomy-specific Progression Classification in Chest Radiographs via Weakly Supervised Learning.
Ke YuShantanu GhoshZhexiong LiuChristopher DeibleClare B PoyntonKayhan BatmanghelichPublished in: Radiology. Artificial intelligence (2024)
"Just Accepted" papers have undergone full peer review and have been accepted for publication in Radiology: Artificial Intelligence . This article will undergo copyediting, layout, and proof review before it is published in its final version. Please note that during production of the final copyedited article, errors may be discovered which could affect the content. Purpose To develop a machine learning approach for classifying disease progression in chest radiographs using weak labels automatically derived from radiology reports. Materials and Methods In this retrospective study, a twin neural network was developed to classify anatomy-specific disease progression into four categories: improved, unchanged, worsened, and new. A two-step weakly-supervised learning approach was employed, pretraining the model on 243,008 frontal chest radiographs from 63,877 MIMIC-CXR patients (mean age 51.7 years; female 34,813), and fine-tuning it on the subset with progression labels derived from consecutive studies. Model performance was evaluated for six pathologic observations on test datasets of unseen MIMIC-CXR patients. Area under the receiver operating characteristic (AUC) analysis was used to evaluate classification performance. The algorithm is also capable of generating bounding-box predictions to localize areas of new progression. Recall, precision, and mean average precision (mAP) were used to evaluate the new progression localization. One-tailed paired t tests were used to assess statistical significance. Results The model outperformed most baselines in progression classification, achieving macro AUC scores of 0.72 ± 0.004 for atelectasis, 0.75 ± 0.007 for consolidation, 0.76 ± 0.017 for edema, 0.81 ± 0.006 for effusion, 0.7 ± 0.032 for pneumonia, and 0.69 ± 0.01 for pneumothorax. For new observation localization, the model achieved mAP scores of 0.25 ± 0.03 for atelectasis, 0.34 ± 0.03 for consolidation, 0.33 ± 0.03 for edema, and 0.31 ± 0.03 for pneumothorax. Conclusion Disease progression classification models were developed on a large chest radiograph dataset, which can be used to monitor interval changes and detect new pathologies on chest radiographs. ©RSNA, 2024.
Keyphrases
- machine learning
- artificial intelligence
- deep learning
- big data
- end stage renal disease
- newly diagnosed
- ejection fraction
- chronic kidney disease
- neural network
- air pollution
- radiation therapy
- prognostic factors
- transcription factor
- neoadjuvant chemotherapy
- randomized controlled trial
- quality improvement
- locally advanced
- lymph node
- intensive care unit
- data analysis
- adverse drug
- rectal cancer