IJCAI 2021 | 医学AI新坑：心电全景图及其生成网络

2021 年 7 月 3 日 机器之心

机器之心专栏

作者：j.t.chen

来自浙大睿医人工智能研究中心的研究者提出了一个全新的概念：心电全景图，并提出了一个能够通过少数几个导联的 ECG 信号输入来生成心电全景图的神经网络模型 Nef-Net。

当前，心电图（ECG）一般仅展示 8-12 导联记录的心电信号（ECG），每条导联的 ECG 信号都是一个金属贴片记录得到的心脏跳动信息，并显示在一个视图上；而这个视图的「观察视角」可以通过金属贴片的位置和心电终端的连接线表达，如下图 (a) 所示。下图 (b) 展示了 6 个导联贴片的位置，心外科医生可以通过 6 个金属贴片接收到这 6 个「视角」的 ECG 信号。但实际上，仅靠这几个有限的导联不能覆盖医生所需的心电信息，例如在进行射频消融手术时，需要精细的心电信号以定位病灶。

来自浙大睿医人工智能研究中心的研究者提出了一个全新的概念：心电全景图（Electrocardio Panorama），并提出了一个能够通过少数几个导联的 ECG 信号输入来生成心电全景图的神经网络模型（Neural electrocardio field network, Nef-Net）。该研究的论文已被 IJCAI 2021 接收。

论文地址：https://arxiv.org/pdf/2105.06293.pdf
项目地址：https://github.com/WhatAShot/Electrocardio-Panorama

该研究提出的 Nef-Net 经过训练后，只要使用者输入一个或几个已知的 ECG 信号及其对应「视角」，Nef-Net 就能编码出该 ECG 信号对应的心电场的底层特征。然后只需再给定一个想要的心电图视角，Nef-Net 就能将心电场的底层特征转换生成该「视角」下的 ECG 信号（视图），其功能如下图所示。

因此，有了 Nef-Net，使用者仅需获得少数有限「视角」下的 ECG 视图，就能观察到任意「视角」下的 ECG 信号。

模型

Nef-Net 本质上是一个 auto-encoder 模型。它能够从「视角」信息相关的 ECG 视图信号中提取出「视角」无关的底层心电场特征；并把该底层特征用于生成任意「视角」的 ECG 信号。

Nef-Net 的模型结构如下图所示。该研究提出了一个名为 Angular Encoding 的「视角」编码算子，以更好地建模「视角」信息。

该研究还提出了一种称为 Standin Learning 的自监督方法，用于引导模型从 ECG 视图特征中剥离「视角」信息而只留下心电场信息。此外，由于模型能够很好地捕捉到心电场特征，Nef-Net 稍作改动之后，还能用于心电图的从头生成任务。

实验

心电全景图生成可视化

心电全景图理论上有无穷个视图，由于空间限制，研究者展示了 24 个均匀分布在球面上的视图来展示实验效果。心电专家认为这些心电全景图具有较高的可信度。

心电从头生成可视化

如下图所示，该研究的方法的效果显著好于专门的心电从头生成 SOTA 模型 SimGAN：

该研究表示，后续还有许多值得进一步探索的问题：

Nef-Net 模型的输入是归一化之后的 ECG 样本，忽略了 ECG 信号振幅的影响。实际上，ECG 信号的振幅在实际临床中是十分重要的。「如何利用模型对无归一化的 ECG 样本进行处理，同时保持良好的真实性」是未来的一个探索方向。
Nef-Net 本身不是专门用于心电从头生成的模型，但却能生成比专门的心电生成模型（如 SimGAN）更真实的心电信号。「如何将 Nef-Net 的经验用于生成 ECG 的 GAN 和 VAE 模型，以生成更真实的 ECG 样本」也是一个值得探索的问题。
当前 Nef-Net 没有考虑到疾病对心电全景图生成的影响，与疾病相关的 ECG 信号从头生成问题和心电全景图生成问题也值得进一步探究。

感兴趣的读者可以查看论文原文，了解更多研究细节。

WAIC AI开发者论坛：后深度学习的AI时代

7月10日，AI开发者论坛将在上海世博中心举办，设有4场 Keynote、9场主题演讲、线上「开发者十问」、2021年云帆奖颁奖典礼等环节，点击下方图片查看日程与报名方式。

WAIC 隐私计算交流会

7月8日，WAIC隐私计算交流会将在上海世博展览馆举办，内场设置有 Keynote、思辨、论文分享三个活动模块，外场还设有开放式论文 Poster 专区。届时将邀请多位领域内顶级专家、论文作者与现场参会观众围绕安全多方计算、数据安全、联邦学习、差分隐私等前沿议题展开交流。点击下方图片查看日程与报名方式。