基于可调Q因子小波变换和迁移学习的癫痫脑电信号检测

doi:10.11896/jsjkx.200200104

摘要/Abstract

摘要： 针对癫痫脑电信号的检测问题,提出一种基于可调Q因子小波变换和迁移学习的癫痫脑电信号检测方法。首先,对EEG信号进行可调Q因子小波变换,并选择能量差异较大的子带进行部分重构,重排重构信号,将其表示为二维彩色图像数据;其次,通过对现有的癫痫发作自动检测算法和深度可分离卷积网络Xception模型的分析,使用ImageNet数据集分类的预训练模型参数进行网络参数初始化,得到深度可分离卷积网络Xception的预训练模型;最后,利用迁移学习方法将Xception模型的预训练结果迁移至癫痫发作自动检测任务。所提方法在BONN癫痫数据集上的准确度达到99.37%,敏感度达到100%,特异度达到98.48%,证明了该模型在癫痫发作自动检测任务上具有良好的泛化能力。与传统检测方法和其他深度学习方法相比,所提自动检测方法达到了较高的准确率,避免了人工设计和提取特征的过程,具有较好的应用价值。

关键词: 癫痫, 可调Q因子小波变换, 迁移学习, 深度可分离卷积网络, 自动检测

Abstract: Aiming at the detection of epileptic EEG signals,a method of detecting epileptic EEG signals based on Tunable Q-factor wavelet transform and transfer learning is proposed.Firstly,the EEG signals are transformed by Tunable Q-factor wavelet transform,and the subbands with large energy differences are selected for partial reconstruction.The reconstructed signals are rearranged and expressed as two-dimensional color image data.Secondly,through the analysis of the existing automatic seizure detection algorithm and the Xception model of deep separable convolutional networks,the parameters of the pre-training model classified by the ImageNet dataset are used to initialize the network parameters,and the pre-training model of the depth separable convolution network Xception is obtained.Finally,the transfer learning method is used to transfer the pre-training results of the Xception model to the automatic seizure detection task.The performance of this method is verified on the BONN epilepsy dataset,and the accuracy,sensitivity and specificity reaches 99.37%,100% and 98.48%respectively,proving that the model has good generalization ability in automatic seizure detection task.Compared with traditional detection methods and other deep lear-ning methods based,the automatic detection method proposed in this paper achieves higher accuracy,avoids the process of artificial design and feature extraction,and has better application value.

Key words: Automatic detection, Depth separable convolutional network, Epilepsy, Transfer learning, Tunable Q-factor wavelet transform

中图分类号:

TP391.4

罗婷瑞, 贾建, 张瑞. 基于可调Q因子小波变换和迁移学习的癫痫脑电信号检测[J]. 计算机科学, 2020, 47(7): 199-205. https://doi.org/10.11896/jsjkx.200200104

LUO Ting-rui, JIA Jian, ZHANG Rui. Epileptic EEG Signals Detection Based on Tunable Q-factor Wavelet Transform and Transfer Learning[J]. Computer Science, 2020, 47(7): 199-205. https://doi.org/10.11896/jsjkx.200200104

参考文献

[1]HASSAN A R,SIULY S,ZHANG Y.Epileptic seizure detection in EEG signals using tunable-Q factor wavelet transform and bootstrap aggregating[J].Computer Methods and Programs in Biomedicine,2016(137):247-259.
[2]SHAFIQUE A,SAYEED M,TSAKALIS K.Nonlinear Dynamical Systems with Chaos and Big Data:A Case Study of Epileptic Seizure Prediction and Control[M].Cham:Springer InternationalPublishing,2018.
[3]SHARMA R,PACHORI R B.Classification of epileptic seizures in EEG signals based on phase space representation of intrinsic mode functions[J].Expert Systems with Applications,2015,42(3):1106-1117.
[4]PATIDAR S,PANIGRAHI T.Detection of epileptic seizure using Kraskov entropy applied on tunable-Q wavelet transform of EEG signals[J].Biomedical Signal Processing and Control,2017,34:74-80.
[5]SHARMA M,PACHORI R B,RAJENDRA A U.A new ap-proach to characterize epileptic seizures using analytic time-frequency flexible wavelet transform and fractal dimension[J].Pattern Recognition Letters,2017,94:172-179.
[6]PEKER M,SEN B,DELEN D.A Novel Method for Automated Diagnosis of Epilepsy Using Complex-Valued Classifiers[J].IEEE Journal of Biomedical and Health Informatics,2016,20(1):108-118.
[7]SAXENA S,HEMANTH C,SANGEETHA R G.Classification of normal,seizure and seizure-free EEG signals using EMD and EWT[C]//International Conference on Nextgen Electronic Technologies:Silicon toSoftware (ICNETS2).IEEE,2017:360-366.
[8]JIA J,BALAJI,SONG J L,et al.Automated identification of epi-leptic seizures in EEG signals based on phase space representation and statistical features in the CEEMDdomain[J].Biome-dical Signal Processing and Control,2017,38:148-157.
[9]RIAZ F,HASSAN A,REHMAN S,et al.EMD-Based Temporal and Spectral Featuresfor the Classification of EEG SignalsUsing Supervised Learning[J].IEEE transactions on neural systems and rehabilitation engineering:a publication of the IEEE Engineering in Medicine and Biology Society,2016,24(1):28-35.
[10]MUTLU A Y.Detection of epileptic dysfunctions in EEG signals using Hilbert vibration decomposition[J].Biomedical Signal Processing and Control,2018,40:33-40.
[11]LU Y,MA Y,CHEN C,et al.Classification of single-channel EEG signals for epileptic seizures detection based on hybrid features[J].Technology and Health Care,2018,26(S1):337-346.
[12]HE K,ZHANG X,REN S,et al.Deep residual learning for ima-ge recognition[C]//Proc.of the IEEE Conference on Computer Vision and Pattern Recognition.2016:770-778.
[13]SELESNICK,IVAN W.Wavelet Transform with Tunable Q-Factor[J].IEEE Transactions on Signal Processing,2011,59(8):3560-3575.
[14]BAYRAM I,SELESNICK I W.Frequencydomain design ofovercomplete rational-dilation wavelet transforms[J].IEEE Transactions on Signal Processing,2009,57(8):2957-2972.
[15]CHEN W Z,WANG X X,ZHANG T.Research of Discrimination Between Left and Right Hand Motor ImageryEEG Patterns Based on Tunable Q-Factor Wavelet Transform[J].Journal of Electronics & Information Technology,2019,41(3):530-536.
[16]KRIZHEVSKY A,SUTSKEVER I,HINTON G.ImageNetClassification with Deep Convolutional Neural Networks[J].Advancesin Neural Information Processing Systems,2012,25(2):1097-1105.
[17]ZHANG Z Z,GAO J Y,LV G,et al.Pathological Image Classification of Gastric Cancer Based on Depth Learning[J].ComputerScience,2018,45(S2):263-268.
[18]YU J Y,DING P C,WANG C.Overview:Application of Convolutional Neural Network in Object Detection [J].Computer Science,2018,45(S2):17-26.
[19]WANG H L,QI X L,WU G S.Research Progress of Object Detection Technology Based on Convolutional Neural Network in Deep Learning[J].Computer Science,2018,45(9):11-19.
[20]ZHOU F Y,JIN L P,DONG J.Review of convolutional neuralnetwork[J].Chinese Journal of Computers,2017,40(6):1229-1251.
[21]SZEGEDY C,VANHOUCKE V,IOFFE S,et al.Rethinking the inception architecture for computer vision[C]//Proc.of the IEEE Conference on Computer Vision and Pattern Recognition.2016:2818-2826.
[22]CHOLLET F.Xception:Deep learning with depth wise separable convolutions[C]//Proc.of the IEEE Conference onCompu-ter Visionand Pattern Recognition.2017:1251-1258.
[23]HU H G,KONG X Y,ZHOU Q W,et al.Melanoma Classification Method by Integrating Deep Convolutional Residual Network [J].Computer Science,2019,46(5):247-253.
[24]TAJBAKHSH N,SHIN J Y,GURUDU S R,et al.Convolutio-nal Neural Networks for Medical Image Analysis:Fine Tuning or Full Training? [J].IEEE Transactions on Medical Imaging,2016,35(5):1299-1312.
[25]JIA D,WEI D,SOCHER R,et al.ImageNet:A large-scale hie-rarchical image database[C]//Proc of IEEE Computer Vision & Pattern Recognition.2009:248-255.
[26]ANDRZEJAK R G,LEHNERTZ K,MORMANN F,et al.Indications of nonlinear deterministic and finite-dimensional structuresin time series of brain electrical activity:Dependence on recording region and brain state[J].Physical Review E,2001,64(6):061907.
[27]POWERS D M W.Visualization of tradeoffin evaluation:fromprecision-recall & PN to LIFT,ROC & BIRD[J].arXiv:1505.00401,2015.
[28]HASSAN A R,SUBASI A.Automatic identification of epileptic
seizures from EEG signals using linear programming boosting[J].Computer Methods and Programs in Biomedicine,2016,136:65-77.
[29]KUMAR Y,DEWAL M L,ANAND R S.Epileptic seizures detection in EEG using DWT-based ApEn and artificial neural network[J].Signal,Image and Video Processing,2014,8(7):1323-1334.
[30]SAMIEE K,KOVACS P,GABBOUJ M.Epileptic Seizure Classification of EEG TimeSeries Using Rational Discrete Short-Time Fourier Transform[J].IEEE Transactions onBiomedical Engineering,2015,62(2):541-552.
[31]AHMEDT-ARISTIZABAL D,FOOKES C,NGUYEN K,et al.Deep classification of epileptic signals[C]//2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).IEEE,2018:332-335.
[32]ZHANG J,WU H,SU W,et al.A New Approach for Classification of Epilepsy EEG Signals Based on Temporal Convolutional Neural Networks[C]//11th International Symposium on Computational Intelligence and Design (ISCID).IEEE,2018:80-84.
[33]ABBASI M U,RASHAD A,BASALAMAH A,et al.Detection of Epilepsy Seizures in Neo-Natal EEG Using LSTM Architecture[J].IEEE Access,2019,7:179074-17908.
[34]HE W P,YANG Y,et al.Tunable Q-factor wavelet transform denoising with neighboring coefficients and its application to rotating machinery fault diagnosis[J].Science China (Technological Sciences),2013(8):126-135.

相关文章 15

[1]	方义秋, 张震坤, 葛君伟. 基于自注意力机制和迁移学习的跨领域推荐算法 Cross-domain Recommendation Algorithm Based on Self-attention Mechanism and Transfer Learning 计算机科学, 2022, 49(8): 70-77. https://doi.org/10.11896/jsjkx.210600011
[2]	王君锋, 刘凡, 杨赛, 吕坦悦, 陈峙宇, 许峰. 基于多源迁移学习的大坝裂缝检测 Dam Crack Detection Based on Multi-source Transfer Learning 计算机科学, 2022, 49(6A): 319-324. https://doi.org/10.11896/jsjkx.210500124
[3]	彭云聪, 秦小林, 张力戈, 顾勇翔. 面向图像分类的小样本学习算法综述 Survey on Few-shot Learning Algorithms for Image Classification 计算机科学, 2022, 49(5): 1-9. https://doi.org/10.11896/jsjkx.210500128
[4]	赵人行, 徐频捷, 刘瑶. 基于深度卷积残差网络的心电单导联房颤检测方法 ECG-based Atrial Fibrillation Detection Based on Deep Convolutional Residual Neural Network 计算机科学, 2022, 49(5): 186-193. https://doi.org/10.11896/jsjkx.220200002
[5]	谭珍琼, 姜文君, 任演纳, 张吉, 任德盛, 李晓鸿. 基于二分图的个性化学习任务分配 Personalized Learning Task Assignment Based on Bipartite Graph 计算机科学, 2022, 49(4): 269-281. https://doi.org/10.11896/jsjkx.210500125
[6]	左杰格, 柳晓鸣, 蔡兵. 基于图像分块与特征融合的户外图像天气识别 Outdoor Image Weather Recognition Based on Image Blocks and Feature Fusion 计算机科学, 2022, 49(3): 197-203. https://doi.org/10.11896/jsjkx.201200263
[7]	张舒萌, 余增, 李天瑞. 跨领域文本的可迁移情绪分析方法 Transferable Emotion Analysis Method for Cross-domain Text 计算机科学, 2022, 49(3): 218-224. https://doi.org/10.11896/jsjkx.210400034
[8]	李星燃, 张立言, 姚树婧. 结合特征融合和注意力机制的微表情识别方法 Micro-expression Recognition Method Combining Feature Fusion and Attention Mechanism 计算机科学, 2022, 49(2): 4-11. https://doi.org/10.11896/jsjkx.210900028
[9]	侯宏旭, 孙硕, 乌尼尔. 蒙汉神经机器翻译研究综述 Survey of Mongolian-Chinese Neural Machine Translation 计算机科学, 2022, 49(1): 31-40. https://doi.org/10.11896/jsjkx.210900006
[10]	熊朝阳, 王婷. 基于卷积神经网络的建筑构件图像识别 Image Recognition for Building Components Based on Convolutional Neural Network 计算机科学, 2021, 48(6A): 51-56. https://doi.org/10.11896/jsjkx.200500122
[11]	杜金莲, 孙鹏飞, 金雪云. 一种用于威胁检测的反目标攻击树模型 Anti-target Attack Tree Model for Threat Detection 计算机科学, 2021, 48(6A): 468-476. https://doi.org/10.11896/jsjkx.200900205
[12]	吴兰, 王涵, 李斌全. 基于自监督任务最优选择的无监督域自适应方法 Unsupervised Domain Adaptive Method Based on Optimal Selection of Self-supervised Tasks 计算机科学, 2021, 48(6A): 357-363. https://doi.org/10.11896/jsjkx.201000030
[13]	李达, 雷迎科, 张海川. 基于LTE网络的室外指纹定位 Outdoor Fingerprint Positioning Based on LTE Networks 计算机科学, 2021, 48(6A): 404-409. https://doi.org/10.11896/jsjkx.200700170
[14]	刘昱彤, 李鹏, 孙云云, 胡素君. 基于深度动态联合自适应网络的图像识别方法 Image Recognition with Deep Dynamic Joint Adaptation Networks 计算机科学, 2021, 48(6): 131-137. https://doi.org/10.11896/jsjkx.210100008
[15]	刘林芽, 吴送英, 左志远, 曹子文. 基于YOLOv3算法的山区铁路边坡落石检测方法研究 Research on Rockfall Detection Method of Mountain Railway Slope Based on YOLOv3 Algorithm 计算机科学, 2021, 48(11A): 290-294. https://doi.org/10.11896/jsjkx.201200113

Metrics

Viewed

Full text

Abstract

Cited

Shared

Discussed