SE-MSCNN: A Lightweight Multi-scaled Fusion Network for Sleep Apnea Detection Using Single-Lead ECG Signals
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).
- Get the pkl file
- Download the dataset Apnea-ECG Database
- Run Preprocessing.py to get a file named apnea-ecg.pkl
- Per-segment classification
- Run SE-MSCNN.py
- Per-recording classification
- Run evaluate.py
- The performance is shown in Table 2.csv
Python==3.6 Keras==2.3.1 TensorFlow==1.14.0
If our work is helpful to you, please cite:
@article{CHEN2022108783,
title = {Toward sleep apnea detection with lightweight multi-scaled fusion network},
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},
author={Chen, Xianhui and Chen, Ying and Ma, Wenjun and Fan, Xiaomao and Li, Ye}}
If you have any questions, please email to: [email protected]