计算机技术与发展2024,Vol.34Issue(4):82-88,7.DOI:10.20165/j.cnki.ISSN1673-629X.2024.0013
基于改进的V-Net模型肺结节分割算法的研究
Research on Lung Nodule Segmentation Algorithm Based on Improved V-Net Model
摘要
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分类
信息技术与安全科学引用本文复制引用
李丽,林晓明,彭丰平,潘家辉..基于改进的V-Net模型肺结节分割算法的研究[J].计算机技术与发展,2024,34(4):82-88,7.基金项目
广东省普通高校特色创新项目(2022KTSCX035) (2022KTSCX035)
国家自然科学基金面上项目(62076103) (62076103)