iVAE - An Improved Deep Learning Structure for EEG Signal Characterization and Reconstruction.

Zheng Chen,Naoaki Ono,Md. Altaf-Ul-Amin,Shigehiko Kanaya,Ming Huang
DOI: https://doi.org/10.1109/BIBM49941.2020.9313107
2020-01-01
Abstract:Due to the inherent variability such as inter-users anatomical variability and the inter-systems differences, the design of new EEG-based index and a reliable model for sleep stages classification is still the main topic in sleep science. The unsupervised deep learning framework—variational autoencoder (VAE) which can capture the major characteristics of the input by imposing a Gaussian prior distribution on the latent features is suitable in EEG characterization and reconstruction. Although vanilla VAE and convolutional autoencoder (CAE) have been tried, it has yet been discussed that whether a deep structure or a multi-scale structure is more appropriate. In this paper, we constructed a shallow iVAE model, which will capture the multi-scale features of the spectrogram of EEG by replacing the main structure in encoder and decoder with the inception-like structure. By comparing with the vanilla VAE and the CAE, a more accurate reconstruction and a better classification using the latent features of the iVAE can be confirmed.
What problem does this paper attempt to address?