A Novel Approach to Learning Models on EEG Data Using Graph Theory Features—A Comparative Study

Abstract

Brain connectivity is studied as a functionally connected network using statistical methods such as measuring correlation or covariance. The non-invasive neuroimaging techniques such as Electroencephalography (EEG) signals are converted to networks by transforming the signals into a Correlation Matrix and analyzing the resulting networks. Here, four learning models, namely, Logistic Regression, Random Forest, Support Vector Machine, and Recurrent Neural Networks (RNN), are implemented on two different types of correlation matrices: Correlation Matrix (static connectivity) and Time-resolved Correlation Matrix (dynamic connectivity), to classify them either on their psychometric assessment or the effect of therapy. These correlation matrices are different from traditional learning techniques in the sense that they incorporate theory-based graph features into the learning models, thus providing novelty to this study. The EEG data used in this study is trail-based/event-related from five different experimental paradigms, of which can be broadly classified as working memory tasks and assessment of emotional states (depression, anxiety, and stress). The classifications based on RNN provided higher accuracy (74–88%) than the other three models (50–78%). Instead of using individual graph features, a Correlation Matrix provides an initial test of the data. When compared with the Time-resolved Correlation Matrix, it offered a 4–5% higher accuracy. The Time-resolved Correlation Matrix is better suited for dynamic studies here; it provides lower accuracy when compared to the Correlation Matrix, a static feature.

Application of Machine Learning to Electroencephalography for the Diagnosis of Primary Progressive Aphasia: A Pilot Study

“…Interestingly, features derived from network analysis obtained the best classification, emphasizing the role of graph theory in the analysis of EEG data [32]. These findings are consistent with recent investigations that are exploiting this new area of analysis [43,44].…”

EEG Channel Selection Techniques in Motor Imagery Applications: A Review and New Perspectives

“…Researchers have proposed various techniques to handle these difficulties successfully. Signal preprocessing, feature extraction, and selecting an appropriate classifier, for example, can increase a BCI’s classification accuracy even though multi-channel EEG has a wide range of applications [ 17 ], specific low computation complexity, and wearable applications. When building a real-time system, researchers frequently overlook the channel selection phase.…”