Explainable artificial intelligence-based prediction of poor neurological outcome from head computed tomography in the immediate post-resuscitation phase

Abstract

Predicting poor neurological outcomes after resuscitation is important for planning treatment strategies. We constructed an explainable artificial intelligence-based prognostic model using head computed tomography (CT) scans taken immediately within 3 h of resuscitation from cardiac arrest and compared its predictive accuracy with that of previous methods using gray-to-white matter ratio (GWR). We included 321 consecutive patients admitted to our institution after resuscitation for out-of-hospital cardiopulmonary arrest with circulation resumption over 6 years. A machine learning model using head CT images with transfer learning was used to predict the neurological outcomes at 1 month. These predictions were compared with the predictions of GWR for multiple regions of interest in head CT using receiver operating characteristic (ROC)-area under curve (AUC) and precision recall (PR)-AUC. The regions of focus were visualized using a heatmap. Both methods had similar ROC-AUCs, but the machine learning model had a higher PR-AUC (0.73 vs. 0.58). The machine learning-focused area of interest for classification was the boundary between gray and white matter, which overlapped with the area of focus when diagnosing hypoxic– ischemic brain injury. The machine learning model for predicting poor outcomes had superior accuracy to conventional methods and could help optimize treatment.