Competition convolutional neural network for sleep stage classification

摘要

Although convolutional neural network (CNN) has become very popular, and has been applied to the sleep stage classification problem, almost all existing studies on sleep stage classification require a lot of labeled data. Obtaining labeled data is a subjective process and difficulty task. At the same time, due to different knowledge backgrounds, the sleep stage labels scored by different experts will be different. Therefore, a new unsupervised competition convolutional neural network (C-CNN) is proposed in this study. It consists of alternating layers containing a convolution operator, competitive operator, and pooling operator. The convolution operator is used to extract features from EEG signals. The competitive layer iteratively adjusts the weight vectors of the winning neurons according to the competition learning rules. By this way, the learned weight vectors can reflect the distribution of input samples. To evaluate the C-CNN model, two common datasets (UCD and Sleep-EDF) are used. The proposed model obtains a classification performance of 77.2% and 83.4% on UCD and Sleep-EDF datasets, respectively. The experimental results also show that our method outperforms the base models by 4.3% and 9.47%, respectively. This work provides avenues for further studies of unsupervised deep learning models.