Automated feature learning using deep convolutional auto-encoder neural network for clustering electroencephalograms into sleep stages

Published

Conference Paper

© 2019 IEEE. Deep neural networks have emerged as popular machine learning tools due to their ability to automatically learn feature representations from raw input data. An auto-encoder neural network is a special network that can be trained in an unsupervised manner for automated feature learning. Unsupervised analysis of EEG signals is highly desirable since supervised analysis requires manual labeling of EEG signals which can be labor intensive and time consuming given the large amount of EEG data collected. We present a deep convolutional auto-encoder neural network to automatically learn feature representations from raw EEG signals in an unsupervised manner. We use the features extracted from the auto-encoder neural network for clustering EEG signals into sleep stages. For clustering, we test two algorithms: K-means - which is a single-membership model, and the latent Dirichlet allocation (LDA) topic model - which is a mixed membership model. Results are presented demonstrating an improvement in clustering performance using auto-encoder features compared to standard manually extracted features.

Full Text

Duke Authors

Cited Authors

  • Prabhudesai, KS; Collins, LM; Mainsah, BO

Published Date

  • May 16, 2019

Published In

Volume / Issue

  • 2019-March /

Start / End Page

  • 937 - 940

Electronic International Standard Serial Number (EISSN)

  • 1948-3554

International Standard Serial Number (ISSN)

  • 1948-3546

International Standard Book Number 13 (ISBN-13)

  • 9781538679210

Digital Object Identifier (DOI)

  • 10.1109/NER.2019.8716996

Citation Source

  • Scopus