多重注意力引导的超声乳腺癌肿瘤图像分割
Multiple Attention-guided Mechanisms for Ultrasound Breast Cancer Tumor Image Segmentation
郭洪洋 1程前 2康晓东 1杨靖怡 1杨舒琪 1李芳 3张蕊1
作者信息
- 1. 天津医科大学医学影像学院 天津 300202
- 2. 重庆大学附属黔江医院 重庆 409000
- 3. 天津医科大学医学影像学院 天津 300202;北京市化工职业病防治院 北京 100093
- 折叠
摘要
传统基于U-Net超声乳腺图像分割任务中存在预测尺度单一和信息丢失等问题.针对存在的问题,提出一种由多重注意力引导机制的U-Net超声乳腺肿瘤图像分割.首先,在U-Net的编码结构中,引入多个SE通道注意力,对输入的乳腺肿瘤图像进行多层级的语义信息提取,引导编码器聚焦乳腺肿瘤特征,减少冗余背景信息带来的干扰;其次,通过设计特征融合处理模块,对编码器传来的特征图进行复杂语义特征的融合处理;最后,在解码器部分,加入金字塔结构捕获全局空间信息,提高模型对肿瘤图像的多尺度特征提取能力,以提高整体网络的表达能力和分割性能.在乳腺肿瘤图像数据集上对该方法进行了仿真实验,结果表明,与其他U-Net改进策略相比,该方法具有更强的准确率和鲁棒性.
Abstract
There are some problems such as single prediction scale and information loss in traditional U-Net ultrasound breast image segmentation tasks.To solve these problems,a multi-attention-guided U-Net ultrasound image segmentation method for breast tumors is proposed.Firstly,multiple SE attention module are introduced into the encoding structure of U-Net to extract multi-level semantic information from the input breast tumor images,which guides the encoder to focus on the features of breast tumor and reduces the interference caused by redundant background information.Secondly,by designing a feature fusion process-ing module,the complex semantic feature fusion processing is carried out on the feature graph from the encoder.Finally,in the de-coder part,the pyramid structure is added to capture global spatial information to improve the multi-scale feature extraction abili-ty of the model for tumor images,so as to improve the expression ability and segmentation performance of the whole network.The proposed method is simulated on breast tumor image data set,and the results show that compared with other U-Net im-proved strategies,the proposed method has better accuracy and robustness.
关键词
多重注意力引导/乳腺/U-Net/超声/图像分割Key words
Multiple attention guidance/Mammary gland/U-Net/Ultrasound/Image segmentation引用本文复制引用
基金项目
京津冀协同创新项目(17YEXTZC00020)
出版年
2024
NETL
NSTL
万方数据