ARX-based EEG data balancing for error potential BCI.

Abstract

<i>Objective.</i>Deep learning algorithms employed in brain computer interfaces (BCIs) need large electroencephalographic (EEG) datasets to be trained. These datasets are usually unbalanced, particularly when error potential (ErrP) experiment are considered, being ErrP's epochs much rarer than non-ErrP ones. To face the issue of unbalance of rare epochs, this paper presents a novel, data balancing methods based on ARX-modelling.<i>Approach.</i>AutoRegressive with eXogenous input (ARX)-models are identified on the EEG data of the 'Monitoring error-related potentials' dataset of the BNCI Horizon 2020 and then employed to generate new synthetic data of the minority class of ErrP epochs. The balanced dataset is used to train a classifier of non-ErrP vs. ErrP epochs based on EEGNet.<i>Main results.</i>Compared to classical techniques (e.g. class weights, CW) for data balancing, the new method outperforms the others in terms of resulting accuracy (i.e. ARX91.5%vs CW88.3%), F1-score (i.e. ARX78.3%vs CW73.7%) and balanced accuracy (i.e. ARX87.0%vs CW81.1%) and also reduces the number of false positive detection (i.e. ARX 51 vs CW 104). Moreover, the ARX-based method shows a better generalization capability of the whole model to classify and predict new data.<i>Significance.</i>The results obtained suggest that the proposed method can be used in BCI application for tackling the issue of data unbalance and obtain more reliable and robust performances.