A Deep Knowledge Distillation framework for EEG assisted enhancement of single-lead ECG based sleep staging

12/14/2021
by   Vaibhav Joshi, et al.
0

Automatic Sleep Staging study is presently done with the help of Electroencephalogram (EEG) signals. Recently, Deep Learning (DL) based approaches have enabled significant progress in this area, allowing for near-human accuracy in automated sleep staging. However, EEG based sleep staging requires an extensive as well as an expensive clinical setup. Moreover, the requirement of an expert for setup and the added inconvenience to the subject under study renders it unfavourable in a point of care context. Electrocardiogram (ECG), an unobtrusive alternative to EEG, is more suitable, but its performance, unsurprisingly, remains sub-par compared to EEG-based sleep staging. Naturally, it would be helpful to transfer knowledge from EEG to ECG, ultimately enhancing the model's performance on ECG based inputs. Knowledge Distillation (KD) is a renowned concept in DL that looks to transfer knowledge from a better but potentially more cumbersome teacher model to a compact student model. Building on this concept, we propose a cross-modal KD framework to improve ECG-based sleep staging performance with assistance from features learned through models trained on EEG. Additionally, we also conducted multiple experiments on the individual components of the proposed model to get better insight into the distillation approach. Data of 200 subjects from the Montreal Archive of Sleep Studies (MASS) was utilized for our study. The proposed model showed a 14.3% and 13.4% increase in weighted-F1-score in 4-class and 3-class sleep staging, respectively. This demonstrates the viability of KD for performance improvement of single-channel ECG based sleep staging in 4-class(W-L-D-R) and 3-class(W-N-R) classification.

READ FULL TEXT
research
11/18/2022

EEG aided boosting of single-lead ECG based sleep staging with Deep Knowledge Distillation

An electroencephalogram (EEG) signal is currently accepted as a standard...
research
11/20/2022

A CNN-Transformer Deep Learning Model for Real-time Sleep Stage Classification in an Energy-Constrained Wireless Device

This paper proposes a deep learning (DL) model for automatic sleep stage...
research
11/10/2021

Recognition of Patient Groups with Sleep Related Disorders using Bio-signal Processing and Deep Learning

Accurately diagnosing sleep disorders is essential for clinical assessme...
research
03/20/2023

Towards Domain Generalization for ECG and EEG Classification: Algorithms and Benchmarks

Despite their immense success in numerous fields, machine and deep learn...
research
08/25/2020

Using Deep Networks for Scientific Discovery in Physiological Signals

Deep neural networks (DNN) have shown remarkable success in the classifi...
research
03/05/2021

SCRIB: Set-classifier with Class-specific Risk Bounds for Blackbox Models

Despite deep learning (DL) success in classification problems, DL classi...
research
01/07/2021

MRNet: a Multi-scale Residual Network for EEG-based Sleep Staging

Sleep staging based on electroencephalogram (EEG) plays an important rol...

Please sign up or login with your details

Forgot password? Click here to reset