基于迁移学习和空洞卷积的癫痫状态识别方法  

Epileptic States Recognition using Transfer Learning and Dilated CNN

作　　者：沈雷 耿馨佚[1,2,3,4] 王守岩 Shen Lei;Geng Xinyi;Wang Shouyan(Institute of Science and Technology fo r Brain-inspired Intelligence,Fudan University,Shanghai 200433,China;Key Laboratory of Computational Neuroscience and Brain-Inspired,Intelligence,Fudan University,Shanghai 200433,China;Shanghai Engineering Research Center of AI&Robotics,Fudan University,Shanghai 200433,China;Engineering Research Center of AI&Robotics,Ministry of Education,Fudan University,Shanghai 200433,China)

机构地区：[1]复旦大学类脑智能科学与技术研究院,上海200433 [2]计算神经科学与类脑智能教育部重点实验室,上海200433 [3]复旦大学上海智能机器人工程技术研究中心,上海200433 [4]复旦大学智能机器人教育部工程研究中心,上海200433

出　　处：《中国生物医学工程学报》2020年第6期700-710,共11页Chinese Journal of Biomedical Engineering

基　　金：高等学校学科创新引智计划(B18015);上海市市级重大科技专项(2018SHZDZX01);国家重点研发计划重点专项项目(2018YFC1312900)。

摘　　要：癫痫患者脑电信号的自动检测和发作诊断对临床治疗癫痫具有重要意义。针对训练数据有限及训练与测试数据分布不一致的难点,采用领域间联合知识迁移学习方法,实现小训练数据量下的癫痫状态识别。首先对脑电信号进行4层小波包分解,提取小波包分解系数作为特征,通过边缘分布和联合分布迭代调整,完成源域和目标域特征之间的知识迁移,训练空洞卷积神经网络作为分类器,完成目标域癫痫状态识别。分别在波士顿儿童医院CHB-MIT脑电数据集(22例被试,共计790 h)和波恩大学癫痫脑电数据集(5组,每组100个片段,每段23.6s)上进行算法验证,实验结果表明,所提出的方法对复杂癫痫状态的平均识别准确度、敏感性、特异性在CHB-MIT数据集上达到96.8%、96.1%、96.4%;在波恩数据集上,平均识别准确率为96.9%,有效提高了癫痫状态识别综合性能,实现了癫痫发作稳定可靠检测。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.

关 键 词：脑电信号 小波包变换 迁移学习 空洞卷积 癫痫识别 

分 类 号：R318[医药卫生—生物医学工程]