Asymmetric U-shaped network with hybrid attention mechanism for kidney ultrasound images segmentation

被引:22
作者
Chen, Gong -Ping [1 ]
Zhao, Yu [1 ]
Dai, Yu [1 ]
Zhang, Jian-Xun [1 ]
Yin, Xiao-Tao [2 ]
Cui, Liang [3 ]
Qian, Jiang [4 ]
机构
[1] NanKai Univ, Coll Artificial Intelligence, Tianjin, Peoples R China
[2] Peoples Liberat Army Gen Hosp, Dept Urol, Med Ctr 4, Beijing, Peoples R China
[3] Civil Aviat Gen Hosp, Dept Urol, Beijing, Peoples R China
[4] Anhui Business Coll, Dept Informat Management Ctr, Wuhu, Peoples R China
基金
中国国家自然科学基金;
关键词
Kidney ultrasound; Images segmentation; Deep learning; Attention mechanism; Deep supervision;
D O I
10.1016/j.eswa.2022.118847
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Kidney ultrasound (KUS) images segmentation is one of the key steps in computer-aided diagnosis. The perturbation of heterogeneous structure, similar intensity distribution, and kidney morphology pose challenges for the segmentation of KUS images. In this paper, we proposed an asymmetric U-shaped network based on the U -net core architecture to segment KUS images accurately and reliably. Specifically, the architecture mainly consists of a dense residual connection encoder, a multi-step up-sampling decoder with the hybrid attention module, and a side-out deep supervision module. The design of the dense residual connection encoder can capture sufficient kidney feature information to improve the representation ability of the network. The devel-opment of the hybrid attention module can further guide the network to pay more attention to the representation of the kidney. In addition, the introduction of the side-out deep supervision model can help the network obtain segmentation results that are closer to the ground-truth masks. Moreover, to reduce network parameters, we proposed a multi-step up-sampling optimization strategy to simplify the design of the network. We compare with several state-of-the-art medical image segmentation methods on the same KUS dataset using seven quantitative metrics. The results of our method on Jaccard, Dice, Accuracy, Recall, Precision, ASSD and AUC are 89.95%, 94.59%, 98.65%, 94.47%, 95.07%, 0.3006 and 0.9703, respectively. Experimental results demonstrate that the proposed method achieves the most competitive segmentation performance on KUS images.
引用
收藏
页数:11
相关论文
共 45 条
[1]  
Abraham N, 2019, I S BIOMED IMAGING, P683, DOI 10.1109/ISBI.2019.8759329
[2]   Managing computational complexity using surrogate models: a critical review [J].
Alizadeh, Reza ;
Allen, Janet K. ;
Mistree, Farrokh .
RESEARCH IN ENGINEERING DESIGN, 2020, 31 (03) :275-298
[3]   Two-stage ultrasound image segmentation using U-Net and test time augmentation [J].
Amiri, Mina ;
Brooks, Rupert ;
Behboodi, Bahareh ;
Rivaz, Hassan .
INTERNATIONAL JOURNAL OF COMPUTER ASSISTED RADIOLOGY AND SURGERY, 2020, 15 (06) :981-988
[4]  
Ardon R, 2015, I S BIOMED IMAGING, P268, DOI 10.1109/ISBI.2015.7163865
[5]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[6]  
Behboodi B, 2019, IEEE ENG MED BIO, P6628, DOI [10.1109/embc.2019.8857218, 10.1109/EMBC.2019.8857218]
[7]   Renal Segmentation From 3D Ultrasound via Fuzzy Appearance Models and Patient-Specific Alpha Shapes [J].
Cerrolaza, Juan J. ;
Safdar, Nabile ;
Biggs, Elijah ;
Jago, James ;
Peters, Craig A. ;
Linguraru, Marius George .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (11) :2393-2402
[8]  
Cerrolaza JJ, 2014, I S BIOMED IMAGING, P633, DOI 10.1109/ISBI.2014.6867950
[9]  
Chen G., 2022, Comput. Methods Programs Biomed
[10]  
Chen G., 2022, AAU-net: An Adaptive Attention U-net for Breast Lesions Segmentation in Ultrasound Images