基于改进残差U-Net的乳腺肿块图像分割方法  

Breast Mass Image Segmentation Algorithm Based on Improved Residual U-Net

作　　者：周雨薇 刘志勤[1] 王庆凤[1] 黄俊[1] 王耀彬[1] 徐卫云[2] ZHOU Yuwei;LIU Zhiqin;WANG Qingfeng;HUANG Jun;WANG Yaobin;XU Weiyun(School of Computer Science and Technology,Southwest University of Science and Technology,Mianyang 621010,Sichuan,China;Mianyang Central Hospital,Mianyang 621000,Sichuan,China)

机构地区：[1]西南科技大学计算机科学与技术学院,四川绵阳621010 [2]绵阳市中心医院,四川绵阳621000

出　　处：《西南科技大学学报》2021年第2期68-74,共7页Journal of Southwest University of Science and Technology

基　　金：国家自然科学基金面上项目(61672438)。

摘　　要：针对乳腺钼靶图像中肿块体积小且常被致密组织掩盖导致肿块分割精度较低的问题,提出一种基于复合加权损失函数的U型对称残差语义分割模型SRes-Unet:首先将含有残差结构的卷积模块嵌入U型网络架构中,提升模型整体的特征提取能力;其次,为了解决乳腺图像中因背景较大造成像素类别严重不平衡问题,利用复合型wBCE_DiceLoss作为残差U型网络(SRes-Unet)的损失函数,同时辅以数据增广以减小过拟合风险。实验结果表明,所提分割模型对于乳腺肿块图像能够实现良好的分割效果,DSC值与MIoU值分别达到了0.82和0.86,对比传统U-Net,在DSC和MIoU指标上分别提升了2个百分点和4个百分点。In order to solve the problem of low precision of mass segmentation in breast molybdenum target images,this paper proposed a U-shaped symmetric residual semantic segmentation model named SRes-Unet,based on a compound weighted loss function.First,the residual module was embedded in U-Net to improve the feature extraction ability of the network model.Then,the compound wBCE_DiceLoss was used as the loss function of residual U-type network(SRes-Unet)to solve the problem of serious imbalance of segmented pixel categories caused by large background in breast images,and the data augmentation was used to reduce the risk of overfitting.The results show that the proposed segmentation model can achieve a good segmentation effect for breast masses,with DSC and MIoU reaching 0.82 and 0.86,respectively.Compared with U-Net,the DSC and MIoU indicators are improved by 2%and 4%,respectively.

关 键 词：乳腺钼靶图像 医学图像分割 卷积神经网络 残差结构 数据增强 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程] TP391.41[自动化与计算机技术—控制科学与工程]