• 基于改进TransUNet的医学图像分割

    Medical image segmentation based on improved TransUNet

    • 医学图像分割在疾病诊断、治疗计划和效果评估中具有重要意义。然而,由于器官边缘的复杂性和形状差异,分割精度仍然面临挑战。为提高分割精度,提出一种改进TransUNet的分割模型。该模型为编码解码结构,编码器由CNN和Transformer组成,CNN负责提取局部特征,Transformer则学习全局特征;解码器通过级联上采样结构逐步恢复图像特征。编码阶段,模型引入高效通道注意力,提升对图像通道特征的关注;解码阶段,模型使用动态卷积适应上采样过程中不同的特征输入,同时采用密集上采样卷积更好地保留图像细节信息。在Synapse多器官分割和自动心脏诊断挑战数据集上进行了验证,实验结果表明性能优于现有模型。

       

      Abstract: Medical image segmentation is important in disease diagnosis, treatment planning and outcome assessment. However, segmentation accuracy is still challenging due to the complexity and shape differences of organ edges.In order to improve the segmentation accuracy, this paper proposes an improved TransUNet segmentation model. The model is a coding and decoding structure, the encoder consists of CNN and Transformer, CNN is responsible for extracting local features, and Transformer learns global features; the decoder gradually recovers the image features by cascading the up-sampling structure, which includes the feature fusion module, the up-sampling module, and the jump connection module. In the encoding stage, the model introduces efficient channel attention to improve the attention to image channel features; in the decoding stage, dynamic convolution is used to adapt to different feature inputs in the up-sampling process, and at the same time, dense up-sampling convolution is used to better preserve image detail information. In this paper, we validate the model on the Synapse multi-organ segmentation and automatic cardiac diagnosis challenge datasets, and the experimental results show that the performance outperforms the existing models.

       

    /

    返回文章
    返回