Deep learning-based personalised outcome prediction after acute ischaemic stroke

Abstract

BackgroundWhether deep learning models using clinical data and brain imaging can predict the long-term risk of major adverse cerebro/cardiovascular events (MACE) after acute ischaemic stroke (AIS) at the individual level has not yet been studied.MethodsA total of 8590 patients with AIS admitted within 5 days of symptom onset were enrolled. The primary outcome was the occurrence of MACEs (a composite of stroke, acute myocardial infarction or death) over 12 months. The performance of deep learning models (DeepSurv and Deep-Survival-Machines (DeepSM)) and traditional survival models (Cox proportional hazards (CoxPH) and random survival forest (RSF)) were compared using the time-dependent concordance index (Ctdindex).ResultsGiven the top 1 to all 60 clinical factors according to feature importance, CoxPH and RSF yieldedCtdindex of 0.7236–0.8222 and 0.7279–0.8335, respectively. Adding image features improved the performance of deep learning models and traditional models assisted by deep learning models. DeepSurv and DeepSM yielded the bestCtdindex of 0.8496 and 0.8531 when images were added to all 39 relevant clinical factors, respectively. In feature importance, brain image was consistently ranked highly. Deep learning models automatically extracted the image features directly from personalised brain images and predicted the risk and date of future MACEs at the individual level.ConclusionsDeep learning models using clinical data and brain images could improve the prediction of MACEs and provide personalised outcome prediction for patients with AIS. Deep learning models will allow us to develop more accurate and tailored prognostic prediction systems that outperform traditional models.