Epileptic States Recognition using Transfer Learning and Dilated CNN
Shen Lei1,2,3,4, Geng Xinyi1,2,3,4, Wang Shouyan1,2,3,4#*
1(Institute of Science and Technology for Brain-inspired Intelligence, Fudan University, Shanghai 200433, China) 2(Key Laboratory of Computational Neuroscience and Brain-Inspired Intelligence, Fudan University, Shanghai 200433, China) 3(Shanghai Engineering Research Center of AI & Robotics, Fudan University, Shanghai 200433, China) 4(Engineering Research Center of AI & Robotics, Ministry of Education, Fudan University, Shanghai 200433, China)
Abstract：The automatic detection and seizure diagnosis of EEG signals in patients with epilepsy is of great significance for clinical treatment of epilepsy. Aiming at solving the problem in the conventional method that the labeled training data volume is insufficient and the classification accuracy of seizure is low due to the inconsistent distribution of training and test data, a joint knowledge transfer method between domains was proposed in this work. Firstly, the EEG signal was decomposed by four-layer wavelet packet, and the wavelet packet decomposition coefficients of 16 frequency bands were extracted as features. The marginal and joint distribution iterative adaptation were used to complete the knowledge transfer between the source and target domain. The dilated convolutional neural network was trained to complete the target domain recognition. In this study, the algorithms were estimated on two public EEG datasets including CHB-MIT dataset (22 patients, 790 hours' recording) and Bonn dataset (5 groups, one hundred 23.6 s episodes in each group). The experimental results showed that the average recognition accuracy, sensitivity and specificity of the proposed method for different epilepsy states was 96.8%,96.1%,96.4% on CHB-MIT dataset respectively. The average recognition accuracy was 96.9% on the Boon dataset, which effectively improved the comprehensive performance of seizure detection and achieve reliable detection results.
沈雷, 耿馨佚, 王守岩. 基于迁移学习和空洞卷积的癫痫状态识别方法[J]. 中国生物医学工程学报, 2020, 39(6): 700-710.
Shen Lei, Geng Xinyi, Wang Shouyan. Epileptic States Recognition using Transfer Learning and Dilated CNN. Chinese Journal of Biomedical Engineering, 2020, 39(6): 700-710.
[1] Yaffe RB, Borger P, Megevand P, et al. Physiology of functional and effective networks in epilepsy[J]. Clinical Neurophysiology, 2015, 126(2):227-236. [2] Shoeb A, Edwards H, Connolly J, et al. Patient-specific seizure onset detection[J]. IEEE Transactions on Medicine and Biology, 2004, 5(4):483-498. [3] Corinna C, Vladimir V. Support-vector networks[J]. Machine Learning, 1995, 20(3):273-297. [4] Polat K, Salih G. Classification of epileptiform EEG using a hybrid system based on decision tree classifier and fast Fourier transform[J]. Applied Mathematics and Computation, 2007, 187(2):1017-1026. [5] Acharya UR, Oh SL, Hagiwara Y, et al. Deep convolutional neural network for the automated detection and diagnosis of seizure using EEG signals[J]. Computers on Biology and Medicine, 2017, 1(1):270-278. [6] Liu C, Wu X, Jia Y, et al. Transfer latent SVM for joint recognition and localization of actions in videos[J]. IEEE Transactions on Systems, Man, and Cybernetics, 2016, 46(11):2596-2608. [7] Deng Z, Choi KS, Jiang YZ. Generalized hidden-mapping ridge regression, knowledge-leveraged inductive transfer learning for neural networks, fuzzy systems and kernel methods[J]. IEEE Transactions on Cybernetics, 2014, 44(12):2585-2599. [8] Yang C, Deng Z, Choi KS, et al. Transductive domain adaptive learning for epileptic electroencephalogram recognition[J]. Artificial Intelligence in Medicine, 2014, 62(3):165-177. [9] Yang C, Deng Z, Choi KS, et al. Takagi-Sugeno-Kang transfer learning fuzzy logic system for the adaptive recognition of epileptic electroencephalogram signals[J]. IEEE Transactions on Fuzzy Systems, 2016, 24(5):1079-1094. [10] Pan SJ, Yang Q. A Survey on transfer learning[J]. IEEE Transactions on Knowledge and Data Engineering, 2010, 22(10):1345-1359. [11] Deng ZH, Peng X, et al. Transductive joint-knowledge-transfer TSK FS for recognition of epileptic EEG signals[J]. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 2018, 26(8):1481-1494. [12] Goldberger AL, Amaral L, Glass L, et al. PhysioBank, PhysioToolkit, and PhysioNet: Components of a new research resource for complex physiologic signals[J]. Circulation, 2000, 101(23):215-220. [13] Andrzejak RG, Lehnertz K, Mormann F, et al. Indications of nonlinear deterministic and finite-dimensional structures in time series of brain electrical activity: Dependence on recording region and brain state[J]. Physical Review E, 2001, 64(6):61-97. [14] Fong GCY, Manyee NG, Jose MS, et al. Childhood absence epilepsy with tonic-clonic seizures and electroencephalogram 3-4Hz spike and multispike-slow wave complexes: Linkage to chromosome 8q24[J]. America Journal of Human Genetics, 1998, 63(4):1117-1129. [15] Khan YU, Gotman J. Wavelet based automatic seizure detection in intracerebral electroencephalogram[J]. Clinical Neurophysiology, 2003, 114(5):898-908. [16] Long M, Wang J, Ding G, et al. Transfer feature learning with joint distribution adaptation [C] // International Conference on Computer Vision. Sydney: IEEE, 2013: 274-278. [17] Yu F, Koltun V. Multi-Scale Context Aggregation by Dilated Convolutions [C] // International Conference on Learning Representations. Puerto Rico: ICLR, 2015: 101-114. [18] Rafiuddin N, Khan YU, Farooq O. Feature extraction and classification of EEG for automatic seizure detection [C] // International Conference on Multimedia. Barcelona: IEEE, 2011: 456-460. [19] Khan YU, Rafiuddin N, Farooq O. Automated seizure detection in scalp EEG using multiple wavelet scales [C] //International Conference on Signal Processing. Waknaghat Solan: IEEE, 2012: 972-977. [20] Kiranyaz S, Ince T, Zabihi M, et al. Automated patient-specific classification of long-term Electroencephalography[J]. Journal of Biomedical Informatics, 2014, 49(1):16-31. [21] Zabihi M, Kiranyaz S, Rad AB, et al. Analysis of high-dimensional phase space via poincaré section for patient-specific seizure detection[J]. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 2016, 24(3):386-398. [22] Samiee K, Kiranyaz S, Gabbouj M, et al. Long-term epileptic EEG classification via 2D mapping and textural features[J]. Expert Systems with Applications, 2015, 42(20):7175-7185. [23] Joachims T. Transductive Inference for Text Classification using support vector machines [C] // International Conference on Machine Learning. Bled: ICML, 1999: 200-209. [24] Huang GB, Zhu QY, Siew CK. Extreme learning machine: Theory and applications[J]. Neurocomputing, 2006, 70(1-3):489-501. [25] Deng Z, Jiang Y, Ishibuchi H, et al. Enhanced knowledge-leverage-based TSK fuzzy system modeling for inductive transfer learning[J]. ACM Transactions on Intelligent Systems and Technology, 2016, 8(1):1-21. [26] Ansari AH, Cherian PJ, Caicedo A, et al. Neonatal seizure detection using deep convolutional neural networks[J]. International Journal of Neural Systems, 2019, 4(1):1850011. [27] 单绍杰,李汉军,王璐璐,等. 基于模型的单导联脑电癫痫发作预测[J]. 计算机应用研究, 2018, 35(11):57-60. [28] 杨昌健,邓赵红,蒋亦樟,等. 引入迁移学习的癫痫信号自适应识别[J]. 计算机工程, 2015, 41(6):158-164.
