李林 编译整理
量子位 报道 | 公众号 QbitAI
吴恩达带领的斯坦福机器学习组(Stanford ML Group)最近开发了一种深度学习新算法,能诊断14类心律不齐,准确率可以媲美人类医生。
他们的论文周四公布在了Arxiv上,题为Cardiologist-Level Arrhythmia Detection with Convolutional Neural Networks,用卷积神经网络进行心脏病专家级的心律不齐检测,作者包括Pranav Rajpurkar、Awni Hannun、Masoumeh Haghpanahi、Codie Bourn和吴恩达。
心律不齐的诊断难题
如果患者疑似有心律不齐症状,通常会去医院做个心电图。但是,在医院做的几分钟或者十几分钟心电图,往往发现不了问题,这时候医生就可能会给患者一个可穿戴的心电图监测器,要戴两周。
两周,就要产生几百小时的心电图数据,医生要一秒一秒地检查,从中找出患者是否出现了有问题的心律不齐。
更难的是,还得把有问题的心律不齐和安全无害的心律不齐区分开,它们在心电图上看起来非常像。
论文的主要作者之一、斯坦福大学研究生Pranav Rajpurkar说:“心律信号之间的差异可能非常细微,比如说,有两种心律不齐都叫做二度房室传导阻滞的,它们在心电图上的表现也非常相似,但一种需要立刻治疗,一种不需要。”
吴恩达带领的斯坦福机器学习小组把这看作一个数据科学问题。
深度学习方法
斯坦福机器学习组的同学们,要用深度神经网络,从心电图中识别14类心律不齐。
△ iRhythm的可穿戴心电图监测器Zio
他们和可穿戴心电图监测设备厂商iRhythm合作,收集并标注了来自29163名患者的64121份心电图数据,以200 Hz的频率采样,构建了大规模的训练集和校验集,用来训练神经网络。
随后,他们又从328名患者处收集了336份心电图数据,构成了测试集。
数据集中的心电图样本每段30秒,都经过专业医生标注。测试集的数据标注则经过3名心脏病专家组成的委员会“会诊”得出。
为识别心电图,他们构建了34层神经网络,包括33个卷积层、一个全连接层和softmax,以原始心电图时间序列为输入,每秒输出一次预测标签。
下图展示了这个深度神经网络的结构:
结果
最后,他们找了6名心脏病专家,分别从测试集的心电图中识别心律不齐,又以3名专家会诊的结果作为标准,将人类识别的结果、算法识别的结果与之进行比较。
他们发现,算法的准确率与受过专业训练的人类心脏病专家相当,在很多情况下甚至比人类更好。
上图蓝色表示算法的识别成绩,黄色是人类专家的识别成绩;Sequence F1表示算法/人类预测与标准结果之间多个标签平均重合度,Set F1表示唯一分类标签的重合度。
识别准确率差不多的情况下,这个算法与人类相比有着巨大优势:它看几百小时的心电图也不会累,可以持续、即时地诊断心律不齐。
长期来看,斯坦福ML组希望能借助专家级的心律不齐识别算法,让更多无法及时就医的人能获得诊断,也希望能让可穿戴设备在危险的心律不齐发生时,能及时进行提醒。
吴恩达说,这正是深度学习提高患者诊断质量、帮医生节约时间的机会之一。
相关链接
项目简介:
https://stanfordmlgroup.github.io/projects/ecg/
论文:
https://arxiv.org/pdf/1707.01836.pdf
【完】
一则通知
量子位读者5群开放申请,对人工智能感兴趣的朋友,可以添加量子位小助手的微信qbitbot2,申请入群,一起研讨人工智能。
另外,量子位大咖云集的自动驾驶技术群,仅接纳研究自动驾驶相关领域的在校学生或一线工程师。申请方式:添加qbitbot2为好友,备注“自动驾驶”申请加入~
招聘
量子位正在招募编辑记者、运营、产品等岗位,工作地点在北京中关村。相关细节,请在公众号对话界面,回复:“招聘”。
△ 扫码强行关注『量子位』
追踪人工智能领域最劲内容