Universal Joint Feature Extraction for P300 EEG Classification using Multi-task Autoencoder

Abstract

The process of recording Electroencephalography (EEG) signals is onerous and requires massive storage to store signals at an applicable frequency rate. In this work, we propose the EventRelated Potential Encoder Network (ERPENet); a multi-task autoencoder-based model, that can be applied to any ERP-related tasks. The strength of ERPENet lies in its capability to handle various kinds of ERP datasets and its robustness across multiple recording setups, enabling joint training across datasets. ERPENet incorporates Convolutional Neural Networks (CNNs) and Long Short-Term Memory (LSTM), in an autoencoder setup, which tries to simultaneously compress the input EEG signal and extract related P300 features into a latent vector. Here, we can infer the process for generating the latent vector as universal joint feature extraction. The network also includes a classification part for attended and unattended events classification as an auxiliary task. We experimented on six different P300 datasets. The results show that the latent vector exhibits better compression capability than the previous state-of-the-art semi-supervised autoencoder model. For attended and unattended events classification, pre-trained weights are adopted as initial weights and tested on unseen P300 datasets to evaluate the adaptability of the model, which shortens the training process as compared to using random Xavier weight initialization. At the compression rate of 6.84, the classification accuracy outperforms conventional P300 classification models: XdawnLDA, DeepConvNet, and EEGNet achieving 79.37% - 88.52% classification accuracy depending on the dataset.

References in corpus (11)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
• Deep learning with convolutional neural networks for EEG decoding and visualization
• Convolutional Neural Networks for Sentence Classification
• Extension of SBL Algorithms for the Recovery of Block Sparse Signals with Intra-Block Correlation
• Compressed Sensing for Energy-Efficient Wireless Telemonitoring of Noninvasive Fetal ECG via Block Sparse Bayesian Learning
• Compressed Sensing of EEG for Wireless Telemonitoring with Low Energy Consumption and Inexpensive Hardware
• Compact Convolutional Neural Networks for Classification of Asynchronous Steady-state Visual Evoked Potentials
• Lossy Image Compression with Compressive Autoencoders
• Spatiotemporal Sparse Bayesian Learning with Applications to Compressed Sensing of Multichannel Physiological Signals
• Deep Models for Engagement Assessment With Scarce Label Information

Cited by in corpus (9)

• MIN2Net: End-to-End Multi-Task Learning for Subject-Independent Motor Imagery EEG Classification
• Decoding EEG Rhythms During Action Observation, Motor Imagery, and Execution for Standing and Sitting
• Deep Learning in EEG: Advance of the Last Ten-Year Critical Period
• MetaSleepLearner: A Pilot Study on Fast Adaptation of Bio-signals-Based Sleep Stage Classifier to New Individual Subject Using Meta-Learning
• SleepPoseNet: Multi-View Learning for Sleep Postural Transition Recognition Using UWB
• Consumer Grade Brain Sensing for Emotion Recognition
• Fuzzy temporal convolutional neural networks in P300-based Brain-computer interface for smart home interaction
• Deep Convolutional Autoencoders as Generic Feature Extractors in Seismological Applications
• Review of Deep Representation Learning Techniques for Brain-Computer Interfaces and Recommendations