Published in:
01-12-2020 | Artificial Intelligence | Imaging Informatics and Artificial Intelligence
Automated identification of chest radiographs with referable abnormality with deep learning: need for recalibration
Authors:
Eui Jin Hwang, Hyungjin Kim, Jong Hyuk Lee, Jin Mo Goo, Chang Min Park
Published in:
European Radiology
|
Issue 12/2020
Login to get access
Abstract
Objectives
To evaluate the calibration of a deep learning (DL) model in a diagnostic cohort and to improve model’s calibration through recalibration procedures.
Methods
Chest radiographs (CRs) from 1135 consecutive patients (M:F = 582:553; mean age, 52.6 years) who visited our emergency department were included. A commercialized DL model was utilized to identify abnormal CRs, with a continuous probability score for each CR. After evaluation of the model calibration, eight different methods were used to recalibrate the original model based on the probability score. The original model outputs were recalibrated using 681 randomly sampled CRs and validated using the remaining 454 CRs. The Brier score for overall performance, average and maximum calibration error, absolute Spiegelhalter’s Z for calibration, and area under the receiver operating characteristic curve (AUROC) for discrimination were evaluated in 1000-times repeated, randomly split datasets.
Results
The original model tended to overestimate the likelihood for the presence of abnormalities, exhibiting average and maximum calibration error of 0.069 and 0.179, respectively; an absolute Spiegelhalter’s Z value of 2.349; and an AUROC of 0.949. After recalibration, significant improvements in the average (range, 0.015–0.036) and maximum (range, 0.057–0.172) calibration errors were observed in eight and five methods, respectively. Significant improvement in absolute Spiegelhalter’s Z (range, 0.809–4.439) was observed in only one method (the recalibration constant). Discriminations were preserved in six methods (AUROC, 0.909–0.949).
Conclusion
The calibration of DL algorithm can be augmented through simple recalibration procedures. Improved calibration may enhance the interpretability and credibility of the model for users.
Key Points
• A deep learning model tended to overestimate the likelihood of the presence of abnormalities in chest radiographs.
• Simple recalibration of the deep learning model using output scores could improve the calibration of model while maintaining discrimination.
• Improved calibration of a deep learning model may enhance the interpretability and the credibility of the model for users.