Enhancing a deep learning model for pulmonary nodule malignancy risk estimation in chest CT with uncertainty estimation

Publikation: Bidrag til tidsskriftTidsskriftartikelForskningfagfællebedømt

  • Dré Peeters
  • Natália Alves
  • Kiran V. Venkadesh
  • Renate Dinnessen
  • Saghir, Zaigham
  • Ernst T. Scholten
  • Cornelia Schaefer-Prokop
  • Rozemarijn Vliegenthart
  • Mathias Prokop
  • Colin Jacobs
Objective
To investigate the effect of uncertainty estimation on the performance of a Deep Learning (DL) algorithm for estimating malignancy risk of pulmonary nodules.

Methods and materials
In this retrospective study, we integrated an uncertainty estimation method into a previously developed DL algorithm for nodule malignancy risk estimation. Uncertainty thresholds were developed using CT data from the Danish Lung Cancer Screening Trial (DLCST), containing 883 nodules (65 malignant) collected between 2004 and 2010. We used thresholds on the 90th and 95th percentiles of the uncertainty score distribution to categorize nodules into certain and uncertain groups. External validation was performed on clinical CT data from a tertiary academic center containing 374 nodules (207 malignant) collected between 2004 and 2012. DL performance was measured using area under the ROC curve (AUC) for the full set of nodules, for the certain cases and for the uncertain cases. Additionally, nodule characteristics were compared to identify trends for inducing uncertainty.

Results
The DL algorithm performed significantly worse in the uncertain group compared to the certain group of DLCST (AUC 0.62 (95% CI: 0.49, 0.76) vs 0.93 (95% CI: 0.88, 0.97); p < .001) and the clinical dataset (AUC 0.62 (95% CI: 0.50, 0.73) vs 0.90 (95% CI: 0.86, 0.94); p < .001). The uncertain group included larger benign nodules as well as more part-solid and non-solid nodules than the certain group.

Conclusion
The integrated uncertainty estimation showed excellent performance for identifying uncertain cases in which the DL-based nodule malignancy risk estimation algorithm had significantly worse performance.
OriginalsprogEngelsk
TidsskriftEuropean Radiology
ISSN0938-7994
DOI
StatusAccepteret/In press - 2024

Bibliografisk note

Funding Information:
RV is supported by an institutional research grant from Siemens Healthineers.

Funding Information:
This study has received funding from the Dutch Cancer Society (KWF Kankerbestrijding, project number 14113).

Publisher Copyright:
© The Author(s) 2024.

ID: 388778255