光电子·激光2021,Vol.32Issue(1) :47-55.DOI:10.16136/j.joel.2021.01.0334

面向肺结节多语义特征分类的不确定性多任务损失方法

Uncertain multi-task loss method for multi-semantic feature classi-fication of lung nodules

张帅威 冯旭鹏 刘利军 黄青松
光电子·激光2021,Vol.32Issue(1) :47-55.DOI:10.16136/j.joel.2021.01.0334

面向肺结节多语义特征分类的不确定性多任务损失方法

Uncertain multi-task loss method for multi-semantic feature classi-fication of lung nodules

张帅威 1冯旭鹏 2刘利军 3黄青松1
扫码查看

作者信息

  • 1. 昆明理工大学信息工程与自动化学院 昆明650500;昆明理工大学云南省计算机技术应用重点实验室 昆明650500
  • 2. 昆明理工大学信息化建设管理中心 昆明650500
  • 3. 昆明理工大学信息工程与自动化学院 昆明650500;云南大学信息学院 昆明650091
  • 折叠

摘要

肺结节的早期诊断对后续的治疗非常重要.尽管深度学习方法在肺结节良恶性分类等任务中取得了良好的结果,但是这些方法没有提供有意义的诊断功能,导致获得的结果缺乏客观性.越来越多的研究者引入了肺结节的其他语义特征来解决这个问题,但是多个语义特征的引入会造成模型的负迁移.为了解决肺结节多个语义特征之间同步共享知识的程度不同造成的负迁移问题.本文提出一种基于不确定性多任务损失的深度学习模型,对肺结节的9个语义特征(精细度,内部结构,钙化,球形度,边缘,分叶征,毛刺征,纹理,恶性程度)进行分类,通过每个任务的同质不确定性来权衡多个损失函数的权重.我们在基准数据集LIDC-IDR1上验证了该方法,本文提出的模型在恶性程度上的分类准确率为93.6%,ROC曲线下面积为95.5%,查全率为84.6%,特异性为94.5%.我们的模型通过肺结节多个语义特征共享知识的程度不同进而改变多个语义特征相对权重提高了恶性程度的分类性能.

关键词

多任务/肺结节/卷积神经网络/深度学习/损失函数

引用本文复制引用

基金项目

国家自然科学基金(81860318)

国家自然科学基金(81560296)

出版年

2021
光电子·激光
天津理工大学 中国光学学会

光电子·激光

CSCD北大核心
影响因子:1.437
ISSN:1005-0086
参考文献量26
段落导航相关论文