国内刊号:31-1289/TP
国际刊号:1000-3428
发布日期:
作者:王磐, 强彦, 杨晓棠, 侯腾璇
单位:1. 太原理工大学 信息与计算机学院, 山西 晋中 030600;2. 山西省肿瘤医院 放射科, 太原 030000
关键词:肺结节分割,空洞卷积,注意力模块,3D-UNet网络,DUpsampling结构
基金:国家自然科学基金(61872261);山西省自然科学基金(201801D121139);虚拟现实技术与系统国家重点实验室开放基金(VRLAB2018A08)。
为提高计算机辅助诊断系统对大尺寸肺结节分割的完整度以及小尺寸肺结节的分割精度,构建双注意力3D-UNet肺结节分割网络模型。将传统3D-UNet网络中的上采样操作替换为DUpsampling结构,通过最小化特征图的像素点与被压缩标签图像之间的损失,得到更具表达能力的特征图,进而提高网络收敛速度。在此基础上,融入空间注意力模块和通道注意力模块,使单通道与多通道中相似的特征彼此相关,增加特征图的全局相关性以提高分割结果的精度。实验结果表明,与3D-UNet等方法相比,该模型有效提高了肺结节分割的准确率,在公共数据集LIDC-IDRI上的MIoU分数达到89.4%。
来源:2021年第2期
《计算机工程》期刊编辑部