Automatic sleep staging model based on single channel electroencephalogram signal / 生物医学工程学杂志

Sleep staging is the basis for solving sleep problems. There's an upper limit for the classification accuracy of sleep staging models based on single-channel electroencephalogram (EEG) data and features. To address this problem, this paper proposed an automatic sleep staging model that mixes deep convolutional neural network (DCNN) and bi-directional long short-term memory network (BiLSTM). The model used DCNN to automatically learn the time-frequency domain features of EEG signals, and used BiLSTM to extract the temporal features between the data, fully exploiting the feature information contained in the data to improve the accuracy of automatic sleep staging. At the same time, noise reduction techniques and adaptive synthetic sampling were used to reduce the impact of signal noise and unbalanced data sets on model performance. In this paper, experiments were conducted using the Sleep-European Data Format Database Expanded and the Shanghai Mental Health Center Sleep Database, and achieved an overall accuracy rate of 86.9% and 88.9% respectively. When compared with the basic network model, all the experimental results outperformed the basic network, further demonstrating the validity of this paper's model, which can provide a reference for the construction of a home sleep monitoring system based on single-channel EEG signals.

Automatic sleep staging algorithm for stochastic depth residual networks based on transfer learning / 生物医学工程学杂志

The existing automatic sleep staging algorithms have the problems of too many model parameters and long training time, which in turn results in poor sleep staging efficiency. Using a single channel electroencephalogram (EEG) signal, this paper proposed an automatic sleep staging algorithm for stochastic depth residual networks based on transfer learning (TL-SDResNet). Firstly, a total of 30 single-channel (Fpz-Cz) EEG signals from 16 individuals were selected, and after preserving the effective sleep segments, the raw EEG signals were pre-processed using Butterworth filter and continuous wavelet transform to obtain two-dimensional images containing its time-frequency joint features as the input data for the staging model. Then, a ResNet50 pre-trained model trained on a publicly available dataset, the sleep database extension stored in European data format (Sleep-EDFx) was constructed, using a stochastic depth strategy and modifying the output layer to optimize the model structure. Finally, transfer learning was applied to the human sleep process throughout the night. The algorithm in this paper achieved a model staging accuracy of 87.95% after conducting several experiments. Experiments show that TL-SDResNet50 can accomplish fast training of a small amount of EEG data, and the overall effect is better than other staging algorithms and classical algorithms in recent years, which has certain practical value.

Automatic sleep staging based on power spectral density and random forest / 生物医学工程学杂志

The method of using deep learning technology to realize automatic sleep staging needs a lot of data support, and its computational complexity is also high. In this paper, an automatic sleep staging method based on power spectral density (PSD) and random forest is proposed. Firstly, the PSDs of six characteristic waves (K complex wave, δ wave, θ wave, α wave, spindle wave, β wave) in electroencephalogram (EEG) signals were extracted as the classification features, and then five sleep states (W, N1, N2, N3, REM) were automatically classified by random forest classifier. The whole night sleep EEG data of healthy subjects in the Sleep-EDF database were used as experimental data. The effects of using different EEG signals (Fpz-Cz single channel, Pz-Oz single channel, Fpz-Cz + Pz-Oz dual channel), different classifiers (random forest, adaptive boost, gradient boost, Gaussian naïve Bayes, decision tree, K-nearest neighbor), and different training and test set divisions (2-fold cross-validation, 5-fold cross-validation, 10-fold cross-validation, single subject) on the classification effect were compared. The experimental results showed that the effect was the best when the input was Pz-Oz single-channel EEG signal and the random forest classifier was used, no matter how the training set and test set were transformed, the classification accuracy was above 90.79%. The overall classification accuracy, macro average F1 value, and Kappa coefficient could reach 91.94%, 73.2% and 0.845 respectively at the highest, which proved that this method was effective and not susceptible to data volume, and had good stability. Compared with the existing research, our method is more accurate and simpler, and is suitable for automation.