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.
Sleep apnea (SA) detection method based on traditional machine learning needs a lot of efforts in feature engineering and classifier design. We constructed a one-dimensional convolutional neural network (CNN) model, which consists in four convolution layers, four pooling layers, two full connection layers and one classification layer. The automatic feature extraction and classification were realized by the structure of the proposed CNN model. The model was verified by the whole night single-channel sleep electrocardiogram (ECG) signals of 70 subjects from the Apnea-ECG dataset. Our results showed that the accuracy of per-segment SA detection was ranged from 80.1% to 88.0%, using the input signals of single-channel ECG signal, RR interval (RRI) sequence, R peak sequence and RRI sequence + R peak sequence respectively. These results indicated that the proposed CNN model was effective and can automatically extract and classify features from the original single-channel ECG signal or its derived signal RRI and R peak sequence. When the input signals were RRI sequence + R peak sequence, the CNN model achieved the best performance. The accuracy, sensitivity and specificity of per-segment SA detection were 88.0%, 85.1% and 89.9%, respectively. And the accuracy of per-recording SA diagnosis was 100%. These findings indicated that the proposed method can effectively improve the accuracy and robustness of SA detection and outperform the methods reported in recent years. The proposed CNN model can be applied to portable screening diagnosis equipment for SA with remote server.