Ensemble Computational Intelligent for Insomnia Sleep Stage Detection via the Sleep ECG Signal

Abstract

Insomnia is a common sleep disorder in which patients cannot sleep properly. Accurate detection of insomnia disorder is a crucial step for disease analysis in the early stages. The disruption in getting quality sleep is one of the big sources of cardiovascular syndromes such as blood pressure and stroke. The traditional insomnia detection methods are time-consuming, cumbersome, and more expensive because they demand a long time from a trained neurophysiologist, and they are prone to human error, hence, the accuracy of diagnosis gets compromised. Therefore, the automatic insomnia diagnosis from the electrocardiogram (ECG) records is vital for timely detection and cure. In this paper, a novel hybrid approach based on the power spectral density (PSD) of the heart rate variability (HRV) is proposed to detect insomnia in three classification scenarios: (1) subject-based classification scenario (normal Vs. insomnia), (2) sleep stage-based classification (REM Vs. W. stage), and (3) the combined classification scenario using both subject-based and sleep stage-based features. The ensemble learning of random forest (RF) and decision tree (DT) classifiers are used to perform the first and second classification scenarios, while the linear discriminant analysis (LDA) classifier is used to perform the third combined scenario. The proposed framework includes data collection, investigation of the ECG signals, extraction of the signal HRV, estimation of the PSD, and AI-based classification via hybrid machine learning classifiers. The proposed framework is fine-tuned and evaluated using the free public Physio Net dataset over fivefold trails cross-validation. For the subject-based classification scenario, the detection performance in terms of sensitivity, specificity, and accuracy is recorded to be 96.0%, 94.0%, and 96.0%, respectively. For the sleep stage-based classification scenario, the detection evaluation results are recorded equally with 96.0% for ceiling level accuracy, sensitivity, and specificity. For the combined classification scenario, the LDA classifier have achieved the best insomnia detection accuracy of 99.0% of the three cases as discussed. In future, the proposed approach could be applicable for mobile observation schemes to automatically detect insomnia disorder.