Skip to content

Bettycxh/Toward-Sleep-Apnea-Detection-with-Lightweight-Multi-scaled-Fusion-Network

Repository files navigation

Toward sleep apnea detection with lightweight multi-scaled fusion network

Abstract

Sleep apnea (SA) is a common sleep disorder that occurs during sleep, leading to the decrease of oxygen saturation in the blood. It would develop a variety of complications like diabetes, chronic kidney disease, depression, cardiovascular diseases, or even sudden death. Early SA detection can help physicians to take interventions for SA patients to prevent malignant events. This paper proposes a lightweight SA detection method of multi-scaled fusion network named SE-MSCNN based on single-lead ECG signals. The proposed SE-MSCNN mainly includes multi-scaled convolutional neural network (CNN) module and channel-wise attention module. In order to facilitate the SA detection performance, various scaled ECG information with different-length adjacent segments are extracted by three sub-neural networks. To overcome the problem of local concentration of feature fusion with concatenation, a channel-wise attention module with a squeeze-to-excitation block is employed to fuse the different scaled features adaptively. Furthermore, the ablation study and computational complexity analysis of the SE-MSCNN are conducted. Extensive experiment results show that the proposed SE-MSCNN has the performance superiority to the state-of-the-art methods for SA detection on the Apnea-ECG benchmark dataset. The SE-MSCNN with the merits of quick response and lightweight parameters can be potentially embedded into a wearable device to provide an SA detection service for individuals in home sleep test (HST). img

Dataset

Apnea-ECG Database

Usage

  1. Get the pkl file
  • Download the dataset Apnea-ECG Database
  • Run Preprocessing.py to get a file named apnea-ecg.pkl
  1. Per-segment classification
  1. Per-recording classification

Requirements

Python==3.6 Keras==2.3.1 TensorFlow==1.14.0

Cite

If our work is helpful to you, please cite:

@article{chen2022toward,
  title={Toward sleep apnea detection with lightweight multi-scaled fusion network},
  author={Chen, Xianhui and Chen, Ying and Ma, Wenjun and Fan, Xiaomao and Li, Ye},
  journal={Knowledge-Based Systems},
  pages={108783},
  year={2022},
  issn = {0950-7051},
  doi = {https://doi.org/10.1016/j.knosys.2022.108783},
  url = {https://www.sciencedirect.com/science/article/pii/S0950705122003689},
  publisher={Elsevier}
}

Email

If you have any questions, please email to: xhchen@nyu.edu

About

Toward Sleep Apnea Detection with Lightweight Multi-scaled Fusion Network

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages