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

被引:17
作者
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
    Alizadeh, Reza
    Allen, Janet K.
    Mistree, Farrokh
    [J]. RESEARCH IN ENGINEERING DESIGN, 2020, 31 (03) : 275 - 298
  • [3] Two-stage ultrasound image segmentation using U-Net and test time augmentation
    Amiri, Mina
    Brooks, Rupert
    Behboodi, Bahareh
    Rivaz, Hassan
    [J]. 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
    Badrinarayanan, Vijay
    Kendall, Alex
    Cipolla, Roberto
    [J]. 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
  • [7] Renal Segmentation From 3D Ultrasound via Fuzzy Appearance Models and Patient-Specific Alpha Shapes
    Cerrolaza, Juan J.
    Safdar, Nabile
    Biggs, Elijah
    Jago, James
    Peters, Craig A.
    Linguraru, Marius George
    [J]. 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