Study on the classification of sleep stages in EEG signals based on DoubleLinkSleepCLNet.

ABSTRACT

PURPOSE: The classification of sleep stages based on Electroencephalogram (EEG) changes has significant implications for evaluating sleep quality and sleep status. Most polysomnography (PSG) systems have a limited number of channels and do not achieve optimal classification performance due to a paucity of raw data. To leverage the data characteristics and enhance the classification accuracy, we propose and evaluate a novel dual-link deep neural network model, 'DoubleLinkSleepCLNet'. METHODS: The DoubleLinkSleepCLNet model performs feature extraction and efficient classification on both the raw EEG and the EEG processed with the Hilbert transform. It leverages the frequency domain and time domain feature modules, resulting in superior performance compared to other models. RESULTS: The DoubleLinkSleepCLNet model, using the 2 Raw/2 Hilbert data modes, achieved the highest classification performance with an accuracy of 88.47%. The average accuracy of the EEG was improved by approximately 4.08% after the application of the Hilbert transform. Additionally, Convolutional Neural Network (CNN) demonstrated superior performance in processing phase information, whereas Long Short-Term Memory (LSTM) excelled in handling time series data. CONCLUSION: The application of the Hilbert transform to EEG data, followed by processing it with a convolutional neural network, enhances the accuracy of the model. These findings introduce novel concepts for accelerating sleep stage prediction research, suggesting potential applications of these methods to other EEG analyses.