Mental Arithmetic Task Recognition Using Effective Connectivity and Hierarchical Feature Selection From EEG Signals

Mental arithmetic analysis based on Electroencephalogram (EEG) signals can help understand disorders, such as attention-deficit hyperactivity, dyscalculia, or autism spectrum disorder where the difficulty in learning or understanding the arithmetic exists. Most mental arithmetic recognition systems rely on features of a single channel of EEG; however, the relationships between EEG channels in the form of effective brain connectivity analysis can contain valuable information. This study aims to find distinctive, effective brain connectivity features and create a hierarchical feature selection for effectively classifying mental arithmetic and baseline tasks. Methods: We estimated effective connectivity using Directed Transfer Function (DTF), direct DTF (dDTF) and Generalized Partial Directed Coherence (GPDC) methods. These measures determine the causal relationship between different brain areas. A hierarchical feature subset selection method selects the most significant effective connectivity features. Initially, Kruskal– Wallis test was performed. Consequently, five feature selection algorithms, namely, Support Vector Machine (SVM) method based on Recursive Feature Elimination, Fisher score, mutual information, minimum Redundancy Maximum Relevance (RMR), and concave minimization and SVM are used to select the best discriminative features. Finally, the SVM method was used for classification. Results: The obtained results indicated that the best EEG classification performance in 29 participants and 60 trials is obtained using GPDC and feature selection via concave minimization method in Beta2 (15-22Hz) frequency band with 89% accuracy. Conclusion: This new hierarchical automated system could be helpful in the discrimination of mental arithmetic and baseline tasks from EEG signals effectively.