-
题名基于双注意力3D-UNet的肺结节分割网络模型
被引量:22
- 1
-
-
作者
王磐
强彦
杨晓棠
侯腾璇
-
机构
太原理工大学信息与计算机学院
山西省肿瘤医院放射科
-
出处
《计算机工程》
CAS
CSCD
北大核心
2021年第2期307-313,共7页
-
基金
国家自然科学基金(61872261)
山西省自然科学基金(201801D121139)
虚拟现实技术与系统国家重点实验室开放基金(VRLAB2018A08)。
-
文摘
为提高计算机辅助诊断系统对大尺寸肺结节分割的完整度以及小尺寸肺结节的分割精度,构建双注意力3D-UNet肺结节分割网络模型。将传统3D-UNet网络中的上采样操作替换为DUpsampling结构,通过最小化特征图的像素点与被压缩标签图像之间的损失,得到更具表达能力的特征图,进而提高网络收敛速度。在此基础上,融入空间注意力模块和通道注意力模块,使单通道与多通道中相似的特征彼此相关,增加特征图的全局相关性以提高分割结果的精度。实验结果表明,与3D-UNet等方法相比,该模型有效提高了肺结节分割的准确率,在公共数据集LIDC-IDRI上的MIoU分数达到89.4%。
-
关键词
肺结节分割
空洞卷积
注意力模块
3d-unet网络
DUpsampling结构
-
Keywords
lung nodule segmentation
atrous convolution
attention module
3d-unet network
DUpsampling structure
-
分类号
TP391
[自动化与计算机技术—计算机应用技术]
-