Adaptive automatic sleep stage classification under covariate shift

Abstract

Current automatic sleep stage classification (ASSC) methods that rely on polysomnographic (PSG) signals suffer from inter-subject differences that make them unreliable in facing with new and different subjects. A novel adaptive sleep scoring method based on unsupervised domain adaptation, aiming to be robust to inter-subject variability, is proposed. We assume that the sleep quality variants follow a covariate shift model, where only the sleep features distribution change in the training and test phases. The maximum overlap discrete wavelet transform (MODWT) is applied to extract relevant features from EEG, EOG and EMG signals. A set of significant features are selected by minimum-redundancy maximum-relevance (mRMR) which is a powerful feature selection method. Finally, an instance-weighting method, namely the importance weighted kernel logistic regression (IWKLR) is applied for the purpose of obtaining adaptation in classification. The classification results using leave one out cross-validation (LOOCV), show that the proposed method performs at the state-of-the art in the field of ASSC.