Eui Jin Hwang1, Hyungjin Kim1, Jong Hyuk Lee1,2, Jin Mo Goo1, Chang Min Park3. 1. Department of Radiology and Institute of Radiation Medicine, Seoul National University College of Medicine, 101 Daehak-ro, Jongno-gu, Seoul, 03080, South Korea. 2. Department of Radiology, Armed Forces Seoul Hospital, Seoul, South Korea. 3. Department of Radiology and Institute of Radiation Medicine, Seoul National University College of Medicine, 101 Daehak-ro, Jongno-gu, Seoul, 03080, South Korea. cmpark.morphius@gmail.com.
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.
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.
Entities:
Keywords:
Artificial intelligence; Calibration; Deep learning; Thoracic radiography
Authors: Eui Jin Hwang; Jong Hyuk Lee; Jae Hyun Kim; Woo Hyeon Lim; Jin Mo Goo; Chang Min Park Journal: BMC Pulm Med Date: 2021-12-07 Impact factor: 3.317