Trial Averaging for Deep EEG Classification

Abstract

Many signals, particularly of biological origin, suffer from a signal-to-noise ratio sufficiently low that it can be difficult to classify individual examples reliably, even with relatively sophisticated machine-learning techniques such as deep learning. In some cases, the noise can be high enough that it is even difficult to achieve convergence during training. We considered this problem for one data type that often suffers from such difficulties, namely electroencephalography (EEG) data from cognitive neuroscience studies in humans. One solution to increase signal-to-noise is, of course, to perform averaging among trials, which has been employed before in other studies of human neuroscience but not, to our knowledge, investigated rigorously, particularly not in deep learning. Here, we parametrically studied the effects of different amounts of trial averaging during training and/or testing in a human EEG dataset, and compared the results to that of a related algorithm, Mixup. Broadly, we found that even a small amount of averaging could significantly improve classification, particularly when both training and testing data were subjected to averaging. Simple averaging clearly outperformed Mixup, although the benefits of averaging differed across classification categories. Overall, our results confirm the value of averaging during training and testing when single-trial classification is not strictly necessary for the application in question. HighlightsO_LIAveraging trials can dramatically improve performance in classification of EEG data C_LIO_LIThe benefits can be seen when averaging on both training and test datasets C_LIO_LISimple trial averaging outperformed a popular related algorithm, Mixup C_LIO_LIHowever, effects of averaging differed across different stimulus categories C_LI