Mental Task Classification Using Electroencephalogram Signal

Abstract

This paper studies the classification problem on electroencephalogram (EEG) data of mental tasks, using standard architecture of three-layer CNN, stacked LSTM, stacked GRU. We further propose a novel classifier - a mixed LSTM model with a CNN decoder. A hyperparameter optimization on CNN shows validation accuracy of 72% and testing accuracy of 62%. The stacked LSTM and GRU models with FFT preprocessing and downsampling on data achieve 55% and 51% testing accuracy respectively. As for the mixed LSTM model with CNN decoder, validation accuracy of 75% and testing accuracy of 70% are obtained. We believe the mixed model is more robust and accurate than both CNN and LSTM individually, by using the CNN layer as a decoder for following LSTM layers. The code is completed in the framework of Pytorch and Keras. Results and code can be found at https://github.com/theyou21/BigProject.

6 pages, 5 figures

References in corpus (7)

• LSTM: A Search Space Odyssey
• Deep learning with convolutional neural networks for EEG decoding and visualization
• Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation
• How to Construct Deep Recurrent Neural Networks
• Independently Recurrent Neural Network (IndRNN): Building A Longer and Deeper RNN
• Identify Statistical Similarities and Differences Between the Deadliest Cancer Types Through Gene Expression
• Dynamic Characterization of Arrows through Stochastic Perturbation

Cited by in corpus (3)

• Structural Vibration Signal Denoising Using Stacking Ensemble of Hybrid CNN-RNN
• Light-Weight 1-D Convolutional Neural Network Architecture for Mental Task Identification and Classification Based on Single-Channel EEG
• A Multi-Variate Triple-Regression Forecasting Algorithm for Long-Term Customized Allergy Season Prediction