Multi-modal signal integration for enhanced sleep stage classification: Leveraging EOG and 2-channel EEG data with advanced feature extraction

Abstract

This paper introduces an innovative approach to sleep stage classification, leveraging a multi-modal signal integration framework encompassing Electrooculography (EOG) and two-channel electroencephalography (EEG) data. We explore the utility of various feature extraction techniques, including Short-Time Fourier Transform (STFT), Wavelet Transform, and raw signal processing, alongside the utilization of neural networks as feature extractors. This unique combination allows us to harness the benefits of traditional feature extraction methods while capitalizing on the power of neural networks to enhance classification performance. Our comprehensive classifier evaluation encompasses a range of models, including Long Short-Term Memory (LSTM) networks and XGBoost. Remarkably, our results reveal exceptional performance with the XGBoost classifier, achieving an overall accuracy of 84.57 % and a macro-F1 score of 78.21 % on the Sleep-EDF expanded dataset, and an overall accuracy of 86.02 % and a macro-F1 score of 81.96 % on the ISRUC-Sleep dataset. Class-specific accuracies highlight its proficiency, particularly in detecting wake and N2 stages, solidifying its superiority among the classifiers tested. This amalgamation of feature sets, complemented by Principal Component Analysis (PCA) for dimensionality reduction, underscores its significance in yielding top-tier classification outcomes. The integration of traditional feature extraction methods with neural networks as feature extractors creates a robust and comprehensive system for sleep stage classification, offering the advantages of both approaches to enhance the accuracy and reliability of the results.