Sleep staging is a crucial task in sleep monitoring and diagnosis, but clinical sleep staging is both time-consuming and subjective. In this study, we proposed a novel deep learning algorithm named feature fusion temporal convolutional network (FFTCN) for automatic sleep staging using single-channel EEG data. This algorithm employed a one-dimensional convolutional neural network (1D-CNN) to extract temporal features from raw EEG, and a two-dimensional CNN (2D-CNN) to extract time-frequency features from spectrograms generated through continuous wavelet transform (CWT) at the epoch level. These features were subsequently fused and further fed into a temporal convolutional network (TCN) to classify sleep stages at the sequence level. Moreover, a two-step training strategy was used to enhance the model's performance on an imbalanced dataset. Our proposed method exhibits superior performance in the 5-class classification task for healthy subjects, as evaluated on the SHHS-1, Sleep-EDF-153, and ISRUC-S1 datasets. This work provided a straightforward and promising method for improving the accuracy of automatic sleep staging using only single-channel EEG, and the proposed method exhibited great potential for future applications in professional sleep monitoring, which could effectively alleviate the workload of sleep technicians.
基金:
National Natural Science Foundation of China [62271385, 32071372]; National Key R&D Program of China [2022YFC2503803]; Sichuan Science and Technology Program [2022YFS0030]; National Heart, Lung, and Blood Institute [U01HL53916, U01HL53931, U01HL53934, U01HL53937, U01HL64360, U01HL53938, U01HL53940, U01HL53941, U01HL63463, R24 HL114473, 75N92019R002]
第一作者机构:[1]Xi An Jiao Tong Univ, Inst Biomed Engn, Sch Life Sci & Technol, Key Lab Biomed Informat Engn,Minist Educ, Xian 710049, Peoples R China[2]Sichuan Digital Econ Ind Dev Res Inst, Chengdu 610036, Peoples R China[3]Minist Civil Affairs, Key Lab Neuroinformat & Rehabil Engn, Xian 710049, Peoples R China
通讯作者:
通讯机构:[1]Xi An Jiao Tong Univ, Inst Biomed Engn, Sch Life Sci & Technol, Key Lab Biomed Informat Engn,Minist Educ, Xian 710049, Peoples R China[2]Sichuan Digital Econ Ind Dev Res Inst, Chengdu 610036, Peoples R China[3]Minist Civil Affairs, Key Lab Neuroinformat & Rehabil Engn, Xian 710049, Peoples R China[10]Xi An Jiao Tong Univ, Inst Engn & Med Interdisciplinary Studies, Sch Mech Engn, Xian 710049, Peoples R China[11]Xi An Jiao Tong Univ, Sch Mech Engn, State Key Lab Mfg Syst Engn, Xian 710049, Peoples R China
推荐引用方式(GB/T 7714):
Bao Jiameng,Wang Guangming,Wang Tianyu,et al.A Feature Fusion Model Based on Temporal Convolutional Network for Automatic Sleep Staging Using Single-Channel EEG[J].IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS.2024,28(11):6641-6652.doi:10.1109/JBHI.2024.3457969.
APA:
Bao, Jiameng,Wang, Guangming,Wang, Tianyu,Wu, Ning,Hu, Shimin...&Wang, Gang.(2024).A Feature Fusion Model Based on Temporal Convolutional Network for Automatic Sleep Staging Using Single-Channel EEG.IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS,28,(11)
MLA:
Bao, Jiameng,et al."A Feature Fusion Model Based on Temporal Convolutional Network for Automatic Sleep Staging Using Single-Channel EEG".IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS 28..11(2024):6641-6652