首页 | 官方网站   微博 | 高级检索  
     

基于双注意力3D-UNet的肺结节分割网络模型
引用本文:王磐,强彦,杨晓棠,侯腾璇.基于双注意力3D-UNet的肺结节分割网络模型[J].计算机工程,2021,47(2):307-313.
作者姓名:王磐  强彦  杨晓棠  侯腾璇
作者单位:1. 太原理工大学 信息与计算机学院, 山西 晋中 030600;2. 山西省肿瘤医院 放射科, 太原 030000
基金项目:国家自然科学基金;山西省自然科学基金;虚拟现实技术与系统国家重点实验室开放基金
摘    要:为提高计算机辅助诊断系统对大尺寸肺结节分割的完整度以及小尺寸肺结节的分割精度,构建双注意力3D-UNet肺结节分割网络模型。将传统3D-UNet网络中的上采样操作替换为DUpsampling结构,通过最小化特征图的像素点与被压缩标签图像之间的损失,得到更具表达能力的特征图,进而提高网络收敛速度。在此基础上,融入空间注意力模块和通道注意力模块,使单通道与多通道中相似的特征彼此相关,增加特征图的全局相关性以提高分割结果的精度。实验结果表明,与3D-UNet等方法相比,该模型有效提高了肺结节分割的准确率,在公共数据集LIDC-IDRI上的MIoU分数达到89.4%。

关 键 词:肺结节分割  空洞卷积  注意力模块  3D-UNet网络  DUpsampling结构  
收稿时间:2019-12-25
修稿时间:2020-02-16

Network Model for Lung Nodule Segmentation Based onDouble Attention 3D-UNet
WANG Pan,QIANG Yan,YANG Xiaotang,HOU Tengxuan.Network Model for Lung Nodule Segmentation Based onDouble Attention 3D-UNet[J].Computer Engineering,2021,47(2):307-313.
Authors:WANG Pan  QIANG Yan  YANG Xiaotang  HOU Tengxuan
Affiliation:1. School of Information and Computer, Taiyuan University of Technology, Jinzhong, Shanxi 030600, China;2. Department of Radiology, Shanxi Province Cancer Hospital, Tainyuan 030000, China
Abstract:In order to improve the segmentation integrity of large pulmonary nodules and the segmentation accuracy of small pulmonary nodules in computer-aided diagnosis systems,this paper constructs a Double Attention 3 D-UNet(DA3 D-UNet)for lung nodule segmentation.The traditional upsampling operation in the 3 D-UNet network is replaced with the DUpsampling structure. By minimizing the loss between the pixels of the feature map and the compressed label image,a more expressive feature map is obtained,thereby improving the network convergence speed.On this basis,the spatial attention module and the channel attention module are integrated to make the similar features in the single channel and the multi-channel correlated with each other,increase the global correlation of the feature map,and improve the accuracy of the segmentation result.Experimental results show that compared with 3 D-UNet and other methods,this model effectively improves the accuracy of lung nodule segmentation,and its MIoU score on the public data set LIDCIDRI reaches 89.4%.
Keywords:lung nodule segmentation  atrous convolution  attention module  3D-UNet network  DUpsampling structure
本文献已被 维普 万方数据 等数据库收录!
点击此处可从《计算机工程》浏览原始摘要信息
点击此处可从《计算机工程》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号