Unsupervised Hybrid Deep Feature Encoder for Robust Feature Learning from Resting-State EEG Data.
Journal:
Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference
PMID:
40039110
Abstract
EEG classification is a challenging task due to the nonstationary nature of EEG data and the covariance shift induced by cross-subject variance. Recently, various machine learning and deep learning models have been developed to learn robust features for inter-subject EEG classification tasks. However, current existing models are designed based on active task-related EEG, with a lack of investigation into learning robust feature representation from resting-state EEG data. Given the differences in the nature of brain activities captured by resting-state and active task-related EEG, existing models might not be applicable to resting-state EEG. This study proposed an unsupervised hybrid deep feature encoder to learn robust feature representation in resting-state EEG data. It involves using a Variational Autoencoder (VAE) to learn latent feature representation, followed by a further feature selection conducted through a non-task-related sample-level proximity classification using K-means clustering. We demonstrate the efficiency of our proposed model through significantly improved classification accuracies compared to benchmark models, as well as the high between-subject separability manifested by the learned feature representation.