###
计算机系统应用英文版:2022,31(4):117-122
本文二维码信息
码上扫一扫!
基于注意力机制多任务的肺结节癌变风险判断方法
(青岛科技大学 信息科学技术学院, 青岛 266061)
Risk Assessment of Lung Nodule Canceration Based on Attention Mechanism and Multitask
(Information Science and Technology, Qingdao University of Science and Technology, Qingdao 266061, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 648次   下载 1598
Received:July 07, 2021    Revised:August 11, 2021
中文摘要: 对于CT影像中检测出的肺部结节, 需要自动判断其是否有癌变风险. 不同于大多数现有的研究方法只区分结节良恶性, 本文提出了一个基于注意力机制的多任务学习模型, 将与结节良恶性相关的语义特征属性一并判断输出, 通过判断9个结节特征(对比度、分叶征、毛刺征、球形度、边缘、纹理、钙化程度、大小以及恶性程度)的同时实现内在特征的共享, 以达到提高各子任务性能的目的. 选择视觉转换器(ViT)模型作为多任务共享特征提取层, 整体模型采用动态加权平均方法来对各子任务的Loss函数进行优化. 在LUNA16数据集上的实验表明, 该学习框架可以提升肺结节癌变风险判断的性能, 且同时对其他语义特征的判断也能提升结果的可解释性.
Abstract:For pulmonary nodules detected in computed tomography (CT) images, it is necessary to automatically determine whether they are at the risk of canceration. This study proposes a multitask learning model based on the attention mechanism. Different from most existing research methods which only distinguish between the benignity and malignancy of nodules, the proposed model also assesses and outputs the semantic features related to the benignity and malignancy of nodules. The assessment of nine nodule features (subtlety, lobulation, spiculation, sphericity, margin, texture, calcification, diameter, and malignancy) and the sharing of internal characteristics are conducted at the same time to improve the performance of each subtask. The vision transformer (ViT) model is selected as the multitask shared feature extraction layer, and the whole model uses the dynamic weighted average method to optimize the Loss function of each subtask. Experiments on the LUNA16 dataset show that the proposed learning framework can improve the risk assessment of pulmonary nodule canceration and that the assessment of other semantic features can also enhance the interpretability of the results.
文章编号:     中图分类号:    文献标志码:
基金项目:国家自然科学基金 (61806107, 61973180, 62002190)
引用文本:
王广涵,程远志,史操,许灿辉.基于注意力机制多任务的肺结节癌变风险判断方法.计算机系统应用,2022,31(4):117-122
WANG Guang-Han,CHENG Yuan-Zhi,SHI Cao,XU Can-Hui.Risk Assessment of Lung Nodule Canceration Based on Attention Mechanism and Multitask.COMPUTER SYSTEMS APPLICATIONS,2022,31(4):117-122