Multi-Site Mild Traumatic Brain Injury Classification with Machine Learning and Harmonization

Abstract

Traumatic brain injury (TBI) can drastically affect an individual's cognition, physical, emotional wellbeing, and behavior. Even patients with mild TBI (mTBI) may suffer from a variety of long-lasting symptoms, which motivates researchers to find better biomarkers. Machine learning algorithms have shown promising results in detecting mTBI from resting-state functional network connectivity (rsFNC) data. However, data collected at multiple sites introduces additional noise called site-effects, resulting in erroneous conclusions. Site errors are controlled through a process called harmonization, but its use in classifying neuroimaging data has been addressed lightly. With the ongoing need to improve mTBI detection, this study shows that harmonization should be integrated into the machine learning process when working with multi-site neuroimaging datasets.