Electroencephalograph (EEG) based classification of mental arithmetic using explainable machine learning

Abstract

Mental arithmetic can be helpful for the evaluation of neurodevelopmental disorders arising from atypical development of the brain. We propose a novel explainable machine learning method for classifying mental arithmetic calculation tasks from resting brain states and good from bad calculations using Electroencephalography. Empirical mode decomposition features are extracted from intrinsic mode functions of the average signals of all trials. Most relevant features to the mental arithmetic tasks are ranked by a random forest-based recursive feature elimination method. These features identify the changes in frequency bands of the brain rhythms, such as delta, theta, and alpha, during mental tasks for the first time in literature. These unique explainable features are also used to identify brain areas such as frontal, temporal, and occipital lobes involved in mental arithmetic tasks. Moreover, our approach describes the memory regions and that bad calculations excite the brain areas, mostly related to emotions such as frustration and anxiety due to stressful mental arithmetic. Using a random forest classifier, beating the state-of-the-art, this method achieved classification accuracies of 99.30 % and 98.33 % for resting vs calculation and good vs bad calculation brain tasks, respectively. Also, our method outperformed the state of art in handling the inter-subject variability and achieved 98.17 ± 0.47 % and 97.19 ± 0.95 % classification accuracies for resting vs calculation and good vs bad calculation tasks, respectively.