Prediction of Recovery from Traumatic Brain Injury with EEG Power Spectrum in Combination of Independent Component Analysis and RUSBoost Model

The computational electroencephalogram (EEG) is recently garnering significant attention in examining whether the quantitative EEG (qEEG) features can be used as new predictors for the prediction of recovery in moderate traumatic brain injury (TBI). However, the brain’s recorded electrical activity has always been contaminated with artifacts, which in turn further impede the subsequent processing steps. As a result, it is crucial to devise a strategy for meticulously flagging and extracting clean EEG data to retrieve high-quality discriminative features for successful model development. This work proposed the use of multiple artifact rejection algorithms (MARA), which is an independent component analysis (ICA)-based algorithm, to eliminate artifacts automatically, and explored their effects on the predictive performance of the random undersampling boosting (RUSBoost) model. Continuous EEG were acquired using 64 electrodes from 27 moderate TBI patients at four weeks to one-year post-accident. The MARA incorporates an artifact removal stage based on ICA prior to RUSBoost, SVM, DT, and k-NN classification. The area under the curve (AUC) of RUSBoost was higher in absolute power spectral density (PSD) in AUCδ = 0.75, AUC α = 0.73 and AUCθ = 0.71 bands than SVM, DT, and k-NN. The MARA has provided a good generalization performance of the RUSBoost prediction model.