首页|结构图注意力网络的新冠肺炎轻重症诊断

结构图注意力网络的新冠肺炎轻重症诊断

扫码查看
原文链接
  • NETL
  • NSTL
  • 万方数据
  • 维普
目的 为辅助医生快速分辨新型冠状病毒肺炎(corona virus disease 2019,COVID-19)轻、重症患者,以便对症下药减轻医疗负担,提出一种基于结构图注意力网络的轻重症诊断算法.方法 基于胸部CT图像提取的特定特征以及肺段间的位置关系构建结构图,以肺部内不同肺段为节点,以提取特征为节点属性.采用图神经网络汇聚相邻节点特征,再利用池化层获取分别代表左肺叶和右肺叶特征的图表示.使用结构注意力机制计算左、右肺叶的感染情况对结果诊断的重要性,并依据重要性融合左、右肺叶图表示以得到最终图表示,最后执行分类任务.由于数据中存在明显的类别不平衡现象,采用Focal-Loss损失函数优化模型以减轻对分类结果的影响.结果 实验将所提算法分别与传统机器学习方法和流行的图神经网络算法做性能对比.在重症诊断的准确率上,本文算法相较于传统机器学习方法和图神经网络算法分别取得14.2％42.0％和3.6％4.8％的提升.在AUC(area under curve)指标上,本文算法相较于上述两种算法分别取得8.9％18.7％和3.1％3.6％的提升.除此之外,通过消融实验发现具有结构注意力机制的算法相较于未使用的算法在SPE(specificity)、SEN(sensitivity)和AUC 3个指标上分别取得了2.4％、1.4％和1.1％的提升;应用Focal-Loss损失函数的算法相较于未使用的算法提升了2.1％、1.1％和0.9％.结论 所提出的诊断模型综合了图神经网络以及结构注意力机制的优点,引入Focal-Loss损失函数,提升了困难样本的分类准确率,使诊断结果更加准确.
Diagnosis of COVID-19 by using structural attention graph neural network

刘彦北、李赫南、张长青、肖志涛、张芳、隗英、高耀宗、石峰、单飞、沈定刚

展开 >

天津工业大学生命科学学院,天津 300387

天津光电检测技术与系统重点实验室,天津 300387

天津工业大学电子与信息工程学院,天津 300387

天津大学智能与计算学部,天津 300073

上海联影智能医疗科技有限公司,上海 200232

复旦大学附属公共卫生临床中心,上海 201508

上海科技大学生物医学工程学院,上海 201210

展开 >

新冠肺炎(COVID-19)诊断 图神经网络(GNN) 结构注意力机制 拓扑结构图 图分类

国家自然科学基金天津市科学技术京津冀基础研究合作项目

61901297H2021202008

2022

中国图象图形学报
中国科学院遥感应用研究所,中国图象图形学学会 ,北京应用物理与计算数学研究所

中国图象图形学报

CSTPCDCSCD北大核心
影响因子：1.111
ISSN：1006-8961
年,卷(期)：2022.27(3)
  • 2
  • 1