改进U-Net的超声乳腺肿瘤分割网络

Enhanced Network for Ultrasound Breast Tumor Segmentation Based on U-Net

乳腺超声图像具有肿瘤大小形态多变、阴影较多、边界模糊等特点,经典U-Net的乳腺肿瘤分割结果与标注图像出入较大。对此,提出改进网络MultiMixU-Net。该网络在U-Net结构中引入MultiMix block以及Respath。MultiMix block通过空洞卷积通路提高网络区分目标以及背景的能力,并通过级联该通路中各卷积层输出,融合普通卷积通路的输出来提取多尺度特征信息。Respath的改进部署使网络中收缩路径与扩张路径之间对应特征信息的传递更加有效。该改进网络在公开的超声乳腺肿瘤分割数据集上进行了测试,实验表明,MultiMixU-Net分割结果优于其他网络且参数量较少。相较于U-Net,所提网络分割结果在所有评价指标上均有提升,其中IoU、DSC分别提升0.1541、0.1273。

The conventional U-Net performs poorly in the segmentation of breast tumors in ultrasound images,due to the variability of breast tumors’shape,the image shadows and the obscure boundary.To overcome these difficulties,this paper proposes an enhanced network,namely MultiMixU-Net.The network introduces the MultiMix block and the Respath into the conventional U-Net.The MultiMix block uses the atrous convolution path to strengthen the ability to discern the target from the background.To extract multi-scale feature information,the MultiMix block concatenates outputs from convolu-tion layers in the atrous convolution path and fuses outputs from the normal convolution paths.The network also modifies the layout of the Respath.This modification can make the transformation of feature information from the contract path to the expanse path more efficient.The experiments on the public ultrasound breast tumor segmentation dataset show that the MultiMixU-Net outperforms other networks and has fewer parameters.Compared with the conventional U-Net,the Multi-MixU-Net in all performance metrics has better results.IoU and DSC are improved by 0.1541 and 0.1273,respectively.