Paper Reproducing (1) -- Deep Learning for EEG Classification

人工智能领域的很多技术早已被应用在一些和医疗相关议题,较为常见的是利用深度学习技术对医学影像进行分析 (如协助判断患者是否患有癌症,预测癌细胞是否出现转移);也有很多是利用深度学习于讯号处理 (如分析病患的脑电波讯号 EEG, 辅助医生诊断睡眠呼吸暂停综合症,Sleep Apnea Syndrome)。

我想实际了解一下如何使用深度学习技术对人的脑电波进行分析,因此选了一篇已经发表的论文,针对其设计的模型进行还原。该论文使用了较为热门的卷积神经网络 (Convolutional Neural Networks, CNN) 作为模型的主要结构,并把问题的定义归为监督式学习(Supervised Learning)。

阅读要求:需要有基础的机器学习相关知识

PS: 所有程式码均不公开到 Github


Related Paper:

O. Tsinalis, P. M. Matthews, Y. Guo, and S. Zafeiriou. (Oct. 2016). “Automatic sleep stage scoring with single-channel EEG using convolutional neural networks.” [Online]. Available:https://arxiv.org/abs/ 1610.01683


Problem Definition

论文使用人体EEG脑波信号,对睡眠状态进行侦测。片段的EEG信号会有相对应的睡眠状态标签,输入资料为EGG信号,输出资料为睡眠状态类别;如某个EGG信号序列为 [1, 0, 2, 3, 4, 5 ...] ,睡眠状态类别是 [ N1],因此 [1, 0, 2, 3, 4, 5 ...] -- [N1] 即为一笔训练资料,因此问题定义为分类问题 (Classification),因此问题用监督式学习 Supervised Learning 可解。


Dataset

为了在复原论文所提出模型的同时,测试其模型的一般性,因此我使用的资料集与论文中所使用的资料集不同,我所使用的是UCDDB资料集 (University College Dublin Sleep Apnea Database)[1]。 UCDDB资料集收集25名病患(21男,4女,平均年龄50岁,平均体重95kg,平均身高173cm)的PSG信号资料。每一份记录包含两个EEG channel的资料(C3-A3和C4-A1 ;指的是头部不同的位点)。资料的取样率为128Hz,即每秒钟有128个资料点;此外,每30秒的讯号资料将对应一个睡眠状态标签 (共有 5 种睡眠状态)。举例而言,一笔150秒的资料包含 5 笔 30 秒的讯号资料,对应5个睡眠状态标签,总共有150 x 128 = 19200个资料点。


Model Structure

论文中,模型的结构如下图所示(我只还原其设计,因为资料的不同,输入向量的大小会不一样)。模型基于CNN网路,输入资料会先经过 20 个 1D-filter 进行特征提取取,然后做max -pooling 的动作;然后 20 个 vector 会堆叠在一起形成一个 2D 的向量空间;这个 2D 的向量会经过 400 个 filter 做进一步的特征提取,400 个向量经过 max -pooling 后进入全连接层,随后输出 5 维的向量,对应 5 个睡眠状态 (如 [0,0,0,0,1] 对应状态1, [0,0,0,1,0] 对应状态2, [0,0,1,0,0] 对应状态3, [0,1,0,0,0] 对应状态4, [1,0,0,0,0] 对应状态5 )。


【Automatic sleep stage scoring with single-channel EEG using convolutional neural networks, https://arxiv.org/abs/ 1610.01683】


参照论文设计相同,程式中用TensorFlow + Keras 输出的模型结构

Experiment Setting

环境:使用VScode编写程式,程式语言为python 3.6,深度学习框架为Tensorflow 2.0 + Keras,运行环境配有多张高性能GPU。

资料前处理:使用raw-data作为输入,不做任何计算特征的动作,仅仅是读原始资料结构,转换为输入资料的结构。

Training and Testing:资料集共包含25名病患的资料,在训练一个模型的过程中,我们的训练资料为前24名病患的资料,测试资料为1名病患的资料。因此我们训练25个模型,最后取平均得到整体的模型准确度。

主要参数设定:optimizer: Adma,epochs:500,validation=0.2, class_weight=“ balanced”, loss=“categorical_crossentropy”。


Results Analysis

以下是实验结果,以 Confusion Matrix 的形式呈现,有25个模型,因此有 25 张图。由于篇幅关系仅选取其中 9 个 Confusion Matrix 的结果。


Result 1
Result 2
Result 3
Result 4
Result 5
Result 6
Result 7
Result 8
Result 9


Conclusion

从初步的结果上看,模型的平均准确度只有0.541,因此效能并不算好 (很不好)。

参数并未做调整,复原论文时没有深究参数设定对模型效能的影响,也无法由此次结果探讨模型之一般性。

复原的结果即便使用新的资料集,依旧与论文呈现的结果有所差异,这部分需要探究论文所使用资料集与本次作业使用的资料集的差异性。


Reference

[1] https://physionet.org/content/ucddb/1.0.0/

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 220,976评论 6 513
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 94,249评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 167,449评论 0 360
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,433评论 1 296
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,460评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 52,132评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,721评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,641评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,180评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,267评论 3 339
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,408评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 36,076评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,767评论 3 332
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,255评论 0 23
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,386评论 1 271
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,764评论 3 375
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,413评论 2 358

推荐阅读更多精彩内容