基于改进TransUNet模型的脑肿瘤图像分割方法研究
Research on Brain Tumor Image Segmentation Method Based on Improved TransUNet Model
朱玉婷 1袁晓1
作者信息
- 1. 四川大学 电子信息学院,四川 成都 610064
- 折叠
摘要
针对肿瘤细胞图像与正常组织图像之间具有强相似性、边界模糊以及染色变化大等特点,提出了基于TransUNet网络的优化改进分割模型.此分割模型在以 TransUNet 为主干网络的基础上于编码器部分引入注意力机制,抑制不相关的部分以突显深层特征的语义信息.同时,改变上采样过程中的融合方式,引入BiFusion模块进行选择性地融合,从而使特征数据能够保留更多高分辨率细节信息.该分割模型在Kaggle脑部低级别胶质瘤数据集上验证.实验结果表明,改进后算法的均交并比,召回率和平均精度均值分别为:97.31%,99.91%和 98.72%,与目前医学图像分割的主流方法相比具有更优的性能.
Abstract
In view of the strong similarity between tumor cell image and normal tissue image,fuzzy boundary and large staining change,an improved optimization model based on TransUNet network is proposed.Based on TransUNet as the backbone network,this segmentation model introduces the attention mechanism into the encoder part to suppress the irrele-vant part to highlight the semantic information of the deep features.At the same time,the fusion method in the up-sampling process is changed,and the BiFusion module is introduced for selective fusion,so that the feature data can retain more high-resolution details.The model is verified on the Kaggle brain low-grade glioma data set.The experimental results show that the MIoU,R and mAP of the improved algorithm are 97.31%,99.91%and 98.72%respectively,which has better perform-ance compared with the current mainstream methods of medical image segmentation.
关键词
脑肿瘤/医学图像分割/注意力机制/特征融合Key words
brain tumor/medical image segmentation/attention mechanism/feature fusion引用本文复制引用
出版年
2024
国家科技期刊平台
NSTL
万方数据