U-Net 网络肺结节分割模型

谭 小波, 何 明亮
沈阳理工大学信息科学与工程学院

摘要

针对 U-Net 在对肺结节进行训练时，存在梯度消失和特征利用率低的问题，提出了一种基于改进 U-Net 的肺结
节分割模型，即 RAU-Net(U-Net network with residual element and attention mechanism) 模型。该模型基于 U-Net 基础框架，
在编码器和解码器中分别加入残差单元，以加深网络深度，提高网络学习能力；同时在网络长连接加入注意力机制，尽可
能获取含有较多上下文语义和空间位置的特征信息。实验结果表明，RAU-Net 模型在 MeanIOU、Dice、F1-Score 三个指
标均高于 Res U-Net 模型，分别提高了 1.61%，3.65%，2.01%，证明了 RAU-Net 模型具有良好的分割性能。

关键词

U-Net；RAU-Net；残差单元；注意力机制；肺结节分割

参考

[1] 程安琪 , 肖丹 . 确诊肺癌后戒烟降低肺癌进展和死亡

率风险 [J]. 中华医学杂志 , 2022(102-024).

[2] 程勇 , 陈卫国 . 医学影像 CAD 的研究和应用 [J]. 医疗

设备信息 ,2006(01):24-27+33.

[3] OTSU N.A threshold selection method from graylevel histograms[J] .IEEE Transactions on Systems,Man,and

Cybernetics,1979,9(1):62-66.

[4] CHEN YuKe,WU Xiao Ming,CAI Ken,CT Image

Segmentation based on Clustering and Graph-Cuts Elsevier,2011,

5179-5184.

[5] JIANG HuiYan,SI YuePeng.Medical image segmentation

based on improved Ostu algorithm and regional growth algorithm[J].

Journal of Northeastern University,2006,27(4):398-401.

[6] 钟思华 , 郭兴明 , 郑伊能 . 改进 U-Net 网络的肺结节

分割方法 [J]. 计算机工程与应用 ,2020,56(17):203-209.

[7] 王磐 , 强彦 , 杨晓棠等 . 基于双注意力 3D-UNet 的肺

结节分割网络模型 [J]. 计算机工程 ,2021,47(02):307-313.

[8] 闫欢兰 , 陆慧娟 , 叶敏超 , 严珂 , 金群 , 徐一格 . 结合

Sobel 算子和 Mask R-CNN 的肺结节分割 [J]. 小型微型计算

机系统 ,2020,41(01):161-165.