DASGC-Unet: An Attention Network for Accurate Segmentation of Liver CT Images

被引:3
作者
Zhang, Xiaoqian [1 ]
Chen, Yufeng [1 ]
Pu, Lei [1 ]
He, Youdong [1 ]
Zhou, Ying [2 ]
Sun, Huaijiang [3 ]
机构
[1] Southwest Univ Sci & Technol, Sch Informat Engn, Mianyang 621010, Sichuan, Peoples R China
[2] Mianyang Cent Hosp, Radiol Dept, Mianyang 621010, Sichuan, Peoples R China
[3] Nanjing Univ Sci & Technol, Sch Comp Sci & Engn, Nanjing 210094, Peoples R China
基金
中国国家自然科学基金;
关键词
Image segmentation; Attention mechanism; Unet; Liver CT images;
D O I
10.1007/s11063-023-11421-y
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The precise segmentation of lesions can assist doctors to complete efficient disease diagnosis. Unet is widely used in the field of medical image segmentation due to its excellent feature fusion ability. However, the deep network based on Unet has poor ability to extract lesion features and insufficient segmentation accuracy. This is because the amount of medical image data is generally small, the lesion area is small, and Unet ignores the importance of different information. To overcome these shortcomings, we propose a Unet-based attention network for accurate segmentation of liver CT images. Specifically, we first creatively design an attention mechanism module (DASGC) that pays attention to both multi-scale spatial information and inter-channel information at the same time, which can give more weight to important feature information and perform the feature information screening task well. Secondly, based on the advantages of DASGC's efficient development of limited information, we masterly design an improved Unet network (DASGC-Unet) to solve the problem that the Unet network cannot effectively use less image information to complete accurate segmentation. Finally, on the LiTS2017 public dataset, our method achieves the best results on mIoU, IoU, and Dice coefficient compared to other advanced attention mechanism networks.
引用
收藏
页码:12289 / 12308
页数:20
相关论文
共 46 条
[1]  
Bilic P., 2019, Med Image Anal
[2]  
Brock Andrew., 2018, arXiv
[3]   GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond [J].
Cao, Yue ;
Xu, Jiarui ;
Lin, Stephen ;
Wei, Fangyun ;
Hu, Han .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2019, :1971-1980
[4]   End-to-End Object Detection with Transformers [J].
Carion, Nicolas ;
Massa, Francisco ;
Synnaeve, Gabriel ;
Usunier, Nicolas ;
Kirillov, Alexander ;
Zagoruyko, Sergey .
COMPUTER VISION - ECCV 2020, PT I, 2020, 12346 :213-229
[5]   Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation [J].
Chen, Liang-Chieh ;
Zhu, Yukun ;
Papandreou, George ;
Schroff, Florian ;
Adam, Hartwig .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :833-851
[6]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848
[7]  
Chen LB, 2017, IEEE INT SYMP NANO, P1, DOI 10.1109/NANOARCH.2017.8053709
[8]   SCA-CNN: Spatial and Channel-wise Attention in Convolutional Networks for Image Captioning [J].
Chen, Long ;
Zhang, Hanwang ;
Xiao, Jun ;
Nie, Liqiang ;
Shao, Jian ;
Liu, Wei ;
Chua, Tat-Seng .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6298-6306
[9]  
Chollet Francois, 2017, PROC CVPR IEEE, P1800, DOI [DOI 10.1109/CVPR.2017.195, 10.48550/arXiv.1610.02357, DOI 10.48550/ARXIV.1610.02357, 10.1109/CVPR.2017.195, 10.1109/CVPR.2017.195.5]
[10]   Attentional Feature Fusion [J].
Dai, Yimian ;
Gieseke, Fabian ;
Oehmcke, Stefan ;
Wu, Yiquan ;
Barnard, Kobus .
2021 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION WACV 2021, 2021, :3559-3568