基于RDU-Net网络的肺部CT分割算法研究

LUNG CT SEGMENTATION ALGORITHM BASED ON RDUNET NETWORK

  • 摘要: 提出RDU-Net网络进行肺结节的分割工作。该网络以U-Net网络为基础,针对网络训练过程中经常出现的梯度消失现象,引入残差单元对基础网络进行改进,该操作很好地解决了网络模型训练过程中出现的梯度消失问题;为了提高网络的泛化能力,在网络中增加了Dropout层,以避免网络在训练过程中过拟合现象,进一步提高分割精度。该网络在LIDC-IDRI数据集上进行实验,其AUC和Dice分别达到了0.89和0.76,相较于基础网络其分割精度和分割效果都有一定的提高。

     

    Abstract: This paper proposes the RDU-Net network to segment lung nodules. The network is based on the U-Net network. Aimed at the phenomenon of gradient disappearance that often occurred during network training, the residual unit was introduced to improve the basic network. This operation solved the problem of gradient disappearance during the network model training process. In order to improve the generalization ability of the network, a Dropout layer was added to the network to avoid overfitting during the training process and further improve the segmentation accuracy. The network was tested on the LIDC-IDRI dataset, and its AUC and Dice reached 0.89 and 0.76, respectively. Compared with the basic network, its segmentation accuracy and segmentation effect were improved to a certain extent.

     

/

返回文章
返回