基于改进TransUnet网络的肺结节分割算法

PULMONARY NODULE SEGMENTATION ALGORITHM BASED ON IMPROVED TRANSUNET NETWORK

  • 摘要: 目前,在医学图像分割领域普遍存在网络模型臃肿、自动分割精度低等问题,难以应用于临床诊断。针对上述问题,提出一种轻量型肺结节分割模型,即Mobile-TransUnet。Mobile-TransUnet模型主要分为三个部分,分别为编码器、Transformer块与解码器部分。在编码器部分将线性瓶颈倒残差算法与注意力机制相结合代替传统卷积块(Convolutional Neural Networks,CNNs);在Transformer部分对提取得到的特征图进行位置序列编码;在解码器部分采用双线性插值法。实验结果表明,Mobile-TransUnet与采用相同层数标准卷积的TransUnet相比分割精度与运行效率大幅提升,明显优于现有多数的肺结节分割算法。

     

    Abstract: At present, there are common problems in the field of medical image segmentation, such as bloated network models and low automatic segmentation accuracy, which are difficult to apply in clinical diagnosis. To address the above problems, a lightweight pulmonary nodule segmentation model, Mobile-TransUnet is proposed. The Mobile-TransUnet model was mainly divided into three parts, which were composed of an encoder, a Transformer block and a decoder. In the encoder part, the linear bottleneck inverted residual algorithm and the attention mechanism were combined to replace the traditional convolutional blocks (Convolutional Neural Networks, CNNs). In the Transformer part, the extracted feature maps were encoded in position sequences. In the decoder part, bilinear interpolation was used. Experimental results show that Mobile-TransUnet has greatly improved segmentation accuracy and operating efficiency compared with TransUnet using standard convolution with the same number of layers, and is significantly better than most existing pulmonary nodule segmentation algorithms.

     

/

返回文章
返回