AUTHOR=Muller Jennifer J. , Wang Ruixuan , Milddleton Devon , Alizadeh Mahdi , Kang Ki Chang , Hryczyk Ryan , Zabrecky George , Hriso Chloe , Navarreto Emily , Wintering Nancy , Bazzan Anthony J. , Wu Chengyuan , Monti Daniel A. , Jiao Xun , Wu Qianhong , Newberg Andrew B. , Mohamed Feroze B. TITLE=Machine learning-based classification of chronic traumatic brain injury using hybrid diffusion imaging JOURNAL=Frontiers in Neuroscience VOLUME=17 YEAR=2023 URL=https://www.frontiersin.org/journals/neuroscience/articles/10.3389/fnins.2023.1182509 DOI=10.3389/fnins.2023.1182509 ISSN=1662-453X ABSTRACT=Background and purpose

Traumatic 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 methods

A 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.

Results

Compared 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%.

Conclusion

The 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.