Automatic liver tumor segmentation used the cascade multi-scale attention architecture method based on 3D U-Net

被引:6
|
作者
Wu, Yun [1 ,2 ]
Shen, Huaiyan [2 ]
Tan, Yaya [2 ]
Shi, Yucheng [2 ]
机构
[1] Guizhou Univ, State Key Lab Publ Big Data, Guiyang 550025, Peoples R China
[2] Guizhou Univ, Coll Comp Sci & Technol, Guiyang 550025, Peoples R China
基金
美国国家科学基金会;
关键词
Liver tumor segmentation; 3D U-Net; Cascade structure; Multi-scale features; Attention mechanism;
D O I
10.1007/s11548-022-02653-9
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Purpose Due to the complex structure of liver tumors and the low contrast with normal tissues make it still a challenging task to accurately segment liver tumors from CT images. To address these problems, we propose an end-to-end segmentation method for liver tumors. Methods The method uses a cascade structure to improve the network's extraction of information. First, the Side-output Feature Fusion Attention block is used to fuse features at different levels and combine with attention mechanism to focus on important information. Then, the Atrous Spatial Pyramid Pooling Attention block is used to extract multi-scale semantic features. Finally, the Multi-scale Prediction Fusion block is used to fully fused the features captured at each layer of the network. Result To verify the performance of the proposed model and the effectiveness of each module, we evaluate it on LiTS and 3DIRCADb datasets and obtained Dice per Case of 0.665 and 0.719, respectively, and Dice Global of 0.812 and 0.784, respectively. Conclusion The proposed method is compared with the basic model 3D U-Net, as well as some mainstream methods based on U-Net variants, and our method achieves better performance on the liver tumor segmentation task and is superior to most segmentation algorithms.
引用
收藏
页码:1915 / 1922
页数:8
相关论文
共 50 条
  • [41] Improved U-Net based insulator image segmentation method based on attention mechanism
    Han Gujing
    Zhang Min
    Wu Wenzhao
    He Min
    Liu Kaipei
    Qin Liang
    Liu Xia
    ENERGY REPORTS, 2021, 7 : 210 - 217
  • [42] Volumetric Hippocampus Segmentation Using 3D U-Net Based On Transfer Learning
    Widodo, Ramadhan Sanyoto Sugiharso
    Purnama, I. Ketut Eddy
    Rachmadi, Reza Fuad
    2024 IEEE INTERNATIONAL CONFERENCE ON COMPUTATIONAL INTELLIGENCE AND VIRTUAL ENVIRONMENTS FOR MEASUREMENT SYSTEMS AND APPLICATIONS, CIVEMSA 2024, 2024,
  • [43] MS-FANet: Multi-scale feature attention network for liver tumor segmentation
    Chen, Ying
    Zheng, Cheng
    Zhang, Wei
    Lin, Hongping
    Chen, Wang
    Zhang, Guimei
    Xu, Guohui
    Wu, Fang
    COMPUTERS IN BIOLOGY AND MEDICINE, 2023, 163
  • [44] CTUNet: automatic pancreas segmentation using a channel-wise transformer and 3D U-Net
    Lifang Chen
    Li Wan
    The Visual Computer, 2023, 39 : 5229 - 5243
  • [45] CTUNet: automatic pancreas segmentation using a channel-wise transformer and 3D U-Net
    Chen, Lifang
    Wan, Li
    VISUAL COMPUTER, 2023, 39 (11) : 5229 - 5243
  • [46] Attention U-Net Oriented Towards 3D Depth Estimation
    Ocsa Sanchez, Leonel Jaime
    Gutierrez Caceres, Juan Carlos
    INTELLIGENT COMPUTING, VOL 3, 2024, 2024, 1018 : 466 - 483
  • [47] A 3D attention U-Net network and its application in geological model parameterization
    Li X.
    Li X.
    Yan L.
    Zhou T.
    Li S.
    Wang J.
    Li X.
    Shiyou Kantan Yu Kaifa/Petroleum Exploration and Development, 2023, 50 (01): : 167 - 173
  • [48] A 3D attention U-Net network and its application in geological model parameterization
    Li, Xiaobo
    Li, Xin
    Yan, Lin
    Zhou, Tenghua
    Li, Shunming
    Wang, Jiqiang
    Li, Xinhao
    PETROLEUM EXPLORATION AND DEVELOPMENT, 2023, 50 (01) : 183 - 190
  • [49] Retinal Vessel Segmentation Based on Attention Mechanism and Multi-Path U-Net
    Hou X.
    Li Z.
    Niu J.
    Liu H.
    Jisuanji Fuzhu Sheji Yu Tuxingxue Xuebao/Journal of Computer-Aided Design and Computer Graphics, 2023, 35 (01): : 55 - 65
  • [50] An automatic detection method of cerebral aneurysms in time-of-flight magnetic resonance angiography images based on attention 3D U-Net
    Chen, Geng
    Meng, Chen
    Ruoyu, Di
    Dongdong, Wang
    Liqin, Yang
    Wei, Xia
    Yuxin, Li
    Daoying, Geng
    COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2022, 225