Optimisation of Real-time EEG Signal Classification Accuracy

Authors

  • Junhan Zhang

DOI:

https://doi.org/10.61173/eckf4q49

Keywords:

EEG, CNN, LSTM

Abstract

This study proposes a hybrid model based on the combination of a convolutional neural network (CNN) and a long short-term memory network (LSTM) for the classification of electroencephalographic (EEG) signals, with a particular focus on the motor imagery task in the BCI Competition IV Dataset 2a. The usability of the data is enhanced by band-pass filtering and independent component analysis (ICA), which effectively removes artefacts and noise from the signal, thus improving the quality of the data. In the process of feature extraction, time-domain statistical features are employed, while high-dimensional features are reduced in dimension through principal component analysis (PCA), thus enhancing the computational efficiency and classification performance of the model. This paper experimental results show that the CNN-LSTM model achieves 100% accuracy on the training set, but only 72.41% on the test set. These findings suggest that the model demonstrates robust classification capabilities when processing training data, but exhibits some limitations in its ability to generalise to previously unseen data. This paper also discusses the limitations of the model and makes suggestions for improvements to further optimise its generalisation performance and enhance the results of its applications.

References

[1] Nunez P L, Srinivasan R. Electric fields of the brain: the neurophysics of EEG. Oxford University Press, 2006.

[2] Luck S J. An introduction to the event-related potential technique. MIT Press, 2014.

[3] Müller K R, Blankertz B. Toward noninvasive braincomputer interfaces. IEEE Signal Processing Magazine, 2006, 23(5): 125-131.

[4] Schirrmeister R T, Springenberg J T, Fiederer L D, Glasstetter M, Eggensperger K, Tangermann M, et al. Deep learning with convolutional neural networks for EEG decoding and visualization. Human Brain Mapping, 2017, 38(11): 5391- 5420.

[5] Makeig S, Onton J. ERP features and EEG dynamics: An ICA perspective. In: Luck S J, Kappenman E S, eds. The Oxford handbook of event-related potential components. Oxford University Press, 2012.

[6] Roy Y, Banville H, Albuquerque I, Gramfort A, Falk T H, Faubert J. Deep learning-based electroencephalography analysis: A systematic review. Journal of Neural Engineering, 2019, 16(5): 051001.

[7] Cohen M X. Analyzing neural time series data: Theory and practice. MIT Press, 2014.

[8] Delorme A, Makeig S. EEGLAB: An open source toolbox for analysis of single-trial EEG dynamics including independent component analysis. Journal of Neuroscience Methods, 2004, 134(1): 9-21.

[9] Wolpaw J R, Birbaumer N, McFarland D J, Pfurtscheller G, Vaughan T M. Brain-computer interfaces for communication and control. Clinical Neurophysiology, 2002, 113(6): 767-791.

[10] Mason S G, Birch G E. A general framework for braincomputer interface design. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 2003, 11(1): 70-85.

Downloads

Published

2024-12-31