-
题名融合CNN和ViT的乳腺超声图像肿瘤分割方法
- 1
-
-
作者
彭雨彤
梁凤梅
-
机构
太原理工大学电子信息与光学工程学院
-
出处
《智能系统学报》
CSCD
北大核心
2024年第3期556-564,共9页
-
基金
山西省重点研发计划项目(202102030201012).
-
文摘
针对乳腺超声图像肿瘤区域形状大小差异大导致分割困难,卷积神经网络(convolutional neural networks,CNN)建模长距离依赖性和空间相关性方面存在局限性,视觉Transformer(vision Transformer,ViT)要求数据量巨大等问题,提出一种融合CNN和ViT的分割方法。使用改进的Swin Transformer模块和基于可形变卷积的CNN编码器模块分别提取全局特征和局部细节特征,设计使用交叉注意力机制融合这两种尺度的特征表示,训练过程采取二元交叉熵损失混合边界损失函数,有效提高分割精度。在两个公共数据集上的实验结果表明,与现有经典算法相比所提方法的分割结果有显著提升,dice系数提升3.8412%,验证所提方法的有效性和可行性。
-
关键词
卷积神经网络
乳腺超声图像分割
Swin
Transformer
交叉注意力机制
混合损失函数
可形变卷积
多头跳跃注意力
深度学习
-
Keywords
convolutional neural network
breast ultrasound image segmentation
Swin Transformer
crossover attention mechanism
hybrid-loss function
deformable convolution
multihead skip attention
deep learning
-
分类号
TP391
[自动化与计算机技术—计算机应用技术]
-