Machine learning-based classification of chronic traumatic brain injury using hybrid diffusion imaging

Abstract

Background and purposeTraumatic brain injury (TBI) can cause progressive neuropathology that leads to chronic impairments, creating a need for biomarkers to detect and monitor this condition to improve outcomes. This study aimed to analyze the ability of data-driven analysis of diffusion tensor imaging (DTI) and neurite orientation dispersion imaging (NODDI) to develop biomarkers to infer symptom severity and determine whether they outperform conventional T1-weighted imaging.Materials and methodsA machine learning-based model was developed using a dataset of hybrid diffusion imaging of patients with chronic traumatic brain injury. We first extracted the useful features from the hybrid diffusion imaging (HYDI) data and then used supervised learning algorithms to classify the outcome of TBI. We developed three models based on DTI, NODDI, and T1-weighted imaging, and we compared the accuracy results across different models.ResultsCompared with the conventional T1-weighted imaging-based classification with an accuracy of 51.7–56.8%, our machine learning-based models achieved significantly better results with DTI-based models at 58.7–73.0% accuracy and NODDI with an accuracy of 64.0–72.3%.ConclusionThe machine learning-based feature selection and classification algorithm based on hybrid diffusion features significantly outperform conventional T1-weighted imaging. The results suggest that advanced algorithms can be developed for inferring symptoms of chronic brain injury using feature selection and diffusion-weighted imaging.