Uncertainty-Aware Deep Learning Classification of Adamantinomatous Craniopharyngioma from Preoperative MRI

Eric W Prince1,2,3, Debashis Ghosh2, Carsten Görg2

  • 1Department of Neurosurgery, University of Colorado School of Medicine, Aurora, CO 80045, USA.

Insights

Deep learning models can now estimate uncertainty for non-invasive adamantinomatous craniopharyngioma (ACP) diagnosis using MRI. This approach improves accuracy by allowing the model to abstain from uncertain classifications.

Area of Science:

  • Neuroimaging
  • Artificial Intelligence
  • Oncology

Background:

  • Adamantinomatous craniopharyngioma (ACP) diagnosis typically requires invasive neurosurgical biopsies.
  • Current deep learning (DL) models for ACP diagnosis lack predictive uncertainty representation, limiting clinical implementation.
  • Radiologists achieve 86% accuracy in distinguishing ACP from other tumors via MRI, with 9% diagnosed non-invasively.

Purpose of the Study:

  • To develop and validate a Bayesian deep learning approach for non-invasive ACP diagnosis using MRI.
  • To incorporate predictive uncertainty estimation into DL classification for improved diagnostic reliability.
  • To establish a classification abstention mechanism based on model uncertainty.

Main Methods:

  • Trained and tested a DL classifier on preoperative MRI scans from 86 suprasellar tumor patients.
  • Applied a Bayesian DL approach to calibrate the existing ACP classifier, focusing on predictive distributions.
  • Developed an algorithm integrating predicted values and uncertainty for a classification abstention mechanism.

Main Results:

  • The calibrated DL model achieved 95.5% accuracy with a 34.2% abstention rate, significantly improving upon the initial 80.8% accuracy.
  • Predictive distribution variance, not mean values, effectively indicated model uncertainty.
  • The abstention mechanism enhanced diagnostic performance by deferring uncertain cases.

Conclusions:

  • Bayesian DL calibration effectively estimates predictive uncertainty in neuroimaging AI models.
  • Incorporating uncertainty estimation and abstention mechanisms can significantly improve AI diagnostic accuracy for brain tumors.
  • This approach holds promise for the clinical translation of AI in non-invasive tumor diagnosis.