Investigation of Machine Learning and Deep Learning Approaches for Detection of Mild Traumatic Brain Injury from Human Sleep Electroencephalogram.

Traumatic Brain Injury (TBI) is a highly prevalent and serious public health concern. Most cases of TBI are mild in nature, yet some individuals may develop following-up persistent disability. The pathophysiologic causes for those with persistent postconcussive symptoms are most likely multifactorial and the underlying mechanism is not well understood, although it is clear that sleep disturbances feature prominently in those with persistent disability. The sleep electroencephalogram (EEG) provides a direct window into neuronal activity during an otherwise highly stereotyped behavioral state, and represents a promising quantitative measure for TBI diagnosis and prognosis. With the ever-evolving domain of machine learning, deep convolutional neural networks, and the development of better architectures, these approaches hold promise to solve some of the long entrenched challenges of personalized medicine for uses in recommendation systems and/or in health monitoring systems. In particular, advanced EEG analysis to identify putative EEG biomarkers of neurological disease could be highly relevant in the prognostication of mild TBI, an otherwise heterogeneous disorder with a wide range of affected phenotypes and disability levels. In this work, we investigate the use of various machine learning techniques and deep neural network architectures on a cohort of human subjects with sleep EEG recordings from overnight, in-lab, diagnostic polysomnography (PSG). An optimal scheme is explored for the classification of TBI versus non-TBI control subjects. The results were promising with an accuracy of ∼95% in random sampling arrangement and ∼70% in independent validation arrangement when appropriate parameters were used using a small number of subjects (10 mTBI subjects and 9 age- and sex-matched controls). We are thus confident that, with additional data and further studies, we would be able to build a generalized model to detect TBI accurately, not only via attended, in-lab PSG recordings, but also in practical scenarios such as EEG data obtained from simple wearables in daily life.

Classification of Electroencephalogram in a Mouse Model of Traumatic Brain Injury Using Machine Learning Approaches.

Traumatic Brain Injury (TBI) is highly prevalent, affecting ~1% of the U.S. population, with lifetime economic costs estimated to be over $75 billion. In the U.S., there are about 50,000 deaths annually related to TBI, and many others are permanently disabled. However, it is currently unknown which individuals will develop persistent disability following TBI and what brain mechanisms underlie these distinct populations. The pathophysiologic causes for those are most likely multifactorial. Electroencephalogram (EEG) has been used as a promising quantitative measure for TBI diagnosis and prognosis. The recent rise of advanced data science approaches such as machine learning and deep learning holds promise to further analyze EEG data, looking for EEG biomarkers of neurological disease, including TBI. In this work, we investigated various machine learning approaches on our unique 24-hour recording dataset of a mouse TBI model, in order to look for an optimal scheme in classification of TBI and control subjects. The epoch lengths were 1 and 2 minutes. The results were promising with accuracy of ~80-90% when appropriate features and parameters were used using a small number of subjects (5 shams and 4 TBIs). We are thus confident that, with more data and studies, we would be able to detect TBI accurately, not only via long-term recordings but also in practical scenarios, with EEG data obtained from simple wearables in the daily life.

Investigation of Machine Learning Approaches for Traumatic Brain Injury Classification via EEG Assessment in Mice.

Due to the difficulties and complications in the quantitative assessment of traumatic brain injury (TBI) and its increasing relevance in today's world, robust detection of TBI has become more significant than ever. In this work, we investigate several machine learning approaches to assess their performance in classifying electroencephalogram (EEG) data of TBI in a mouse model. Algorithms such as decision trees (DT), random forest (RF), neural network (NN), support vector machine (SVM), K-nearest neighbors (KNN) and convolutional neural network (CNN) were analyzed based on their performance to classify mild TBI (mTBI) data from those of the control group in wake stages for different epoch lengths. Average power in different frequency sub-bands and alpha:theta power ratio in EEG were used as input features for machine learning approaches. Results in this mouse model were promising, suggesting similar approaches may be applicable to detect TBI in humans in practical scenarios.