计算机技术与发展2024,Vol.34Issue(4) :82-88.DOI:10.20165/j.cnki.ISSN1673-629X.2024.0013

基于改进的V-Net模型肺结节分割算法的研究

Research on Lung Nodule Segmentation Algorithm Based on Improved V-Net Model

李丽 林晓明 彭丰平 潘家辉
计算机技术与发展2024,Vol.34Issue(4) :82-88.DOI:10.20165/j.cnki.ISSN1673-629X.2024.0013

基于改进的V-Net模型肺结节分割算法的研究

Research on Lung Nodule Segmentation Algorithm Based on Improved V-Net Model

李丽 1林晓明 2彭丰平 1潘家辉1
扫码查看

作者信息

  • 1. 华南师范大学软件学院,广东佛山 528225
  • 2. 广东佛山联创工程研究生院,广东佛山 528311
  • 折叠

摘要

由于CT图像是三维图像,在原始的V-Net模型分割中,易出现结节漏检和边界分割不清晰,以及损失函数Dice训练时不稳定等问题.根据这些问题,提出3D多尺度SE V-Net,简称MSEV-Net网络,同时通过联合损失函数来提高训练的稳定性.该网络模型在V-Net网络的基础上,使用多尺度卷积模块来替换原有的5×5×5卷积,同时在残差连接后加入SE通道注意力模块,通过不同尺度的特征融合和学习不同通道之间的关系,解决肺结节小不易分割的问题.同时在V-Net网络残差连接基础上加一条短跳跃连接,使得整个网络更好利用全局特征.联合损失函数选择Dice和交叉熵损失函数进行融合,可以很好地解决训练不稳定问题.提出的MSEV-Net网络模型和联合损失函数在平均分割准确率PA达到0.998,DSC达到0.837.实验结果表明,该方法在提高肺结节分割精度方面具有一定的效果.

Abstract

Since CT images are three-dimensional images,in the original V-Net model segmentation,it is prone to the problems of nodal omission and unclear boundary segmentation,as well as the instability during the training of the loss function Dice.According to these problems,we propose 3D multi-scale SE V-Net referred to as MSEV-Net network,while improving the stability of training by joint loss function.This network model is based on the V-Net network,using the multi-scale convolution module to replace the original 5×5×5 convolution,while adding the SE channel attention module after the residual connection to solve the problem of small lung nodules that are not easy to segment by fusing features of different scales and learning the relationship between different channels.At the same time,a short jump connection is added on top of the residual connection of the V-Net network,which makes the whole network better utilize the global features.The joint loss function selects Dice and cross-entropy loss function for fusion,which can well solve the problem of training instability.The MSEV-Net network model and joint loss function proposed reach 0.998 in the average segmentation accuracy PA and 0.837 in the DSC.The experimental results show that the proposed method is effective in improving the segmentation accuracy of lung nodules.

关键词

肺结节分割/V-Net网络/联合损失函数/多尺度卷积/SE模块

Key words

pulmonary nodule segmentation/V-Net network/joint loss function/multi-scale convolution/SE modules

引用本文复制引用

基金项目

广东省普通高等学校特色创新项目(2022KTSCX035)

国家自然科学基金面上项目(62076103)

出版年

2024
计算机技术与发展
陕西省计算机学会

计算机技术与发展

CSTPCD
影响因子:0.621
ISSN:1673-629X
参考文献量20
段落导航相关论文