计算机工程与设计2024,Vol.45Issue(10) :3033-3041.DOI:10.16208/j.issn1000-7024.2024.10.020

轻量化及边界加强的医学图像分割模型

Lightweight and boundary-strengthening medical image segmentation model

葛彩成 武丽 张征浩 俞俊 朱蒙
计算机工程与设计2024,Vol.45Issue(10) :3033-3041.DOI:10.16208/j.issn1000-7024.2024.10.020

轻量化及边界加强的医学图像分割模型

Lightweight and boundary-strengthening medical image segmentation model

葛彩成 1武丽 2张征浩 1俞俊 1朱蒙3
扫码查看

作者信息

  • 1. 南京信息工程大学电子与信息工程学院,江苏南京 210044
  • 2. 南京信息工程大学电子与信息工程学院,江苏南京 210044;无锡学院电子信息工程学院,江苏无锡 214105
  • 3. 徐州医科大学医学影像学院,江苏徐州 221004
  • 折叠

摘要

为提升医学图像分割模型轻量化水平及分割精准度,在TransUNet基础上通过引入具有稀疏化自注意力计算方式的Transformer、边界分割加强机制和强化细节特征提取的互补注意力机制,采用深度可分离卷积和CARAFE模块取代TransUNet原有的常规卷积和上采样,设计一种具有相对轻量化的边界精准分割模型LB-TransUNet.在Synapse多器官分割数据集上的实验结果表明,LB-TransUNet的Dice系数达到79.30,Hausdorff距离达到21.03%,相较于TransUNet、Swin-UNet等模型,LB-TransUNet可以更精准分割出各器官.

Abstract

To improve the lightweight level and segmentation accuracy of the medical image segmentation model,the Transformer with sparse self-attention calculation method,the boundary segmentation enhancement mechanism and the complementary atten-tion mechanism were introduced to enhance detail feature extraction on the basis of TransUNet,and the original conventional convolution and the upsampling of TransUNet were replaced through deep separable convolution and CARAFE modules.A boundary-accurate segmentation model LB-TransUNet with relative lightweight was designed.Experimental results on the Synapse multi-organ segmentation dataset show that the Dice coefficient and the Hausdorff distance of LB-TransUNet can reach 79.30 and 21.03%respectively,realizing more accurate segmentation effects compared with that of TransUNet,Swin-UNet and other models.

关键词

医学图像分割/稀疏化自注意力/互补注意力/TransUNet模型/Transformer模型/轻量化/边界精准分割

Key words

medical image segmentation/sparse self-attention/complementary attention/TransUNet model/Transformer model/lightweight/precise boundary segmentation

引用本文复制引用

基金项目

国家自然科学基金青年基金项目(62106111)

无锡学院2021年第二批产学合作协同育人基金项目(202102563020)

出版年

2024
计算机工程与设计
中国航天科工集团二院706所

计算机工程与设计

CSTPCD北大核心
影响因子:0.617
ISSN:1000-7024
段落导航相关论文