An improved segmentation algorithm of CT image based on U-Net network and attention mechanism

被引:13
|
作者
Yang, Jin [1 ,2 ]
Qiu, Kai [2 ]
机构
[1] Cent South Univ, Sch Informat Sci & Engn, Changsha 410083, Hunan, Peoples R China
[2] Changsha Social Work Coll, Informat Ctr, Changsha 410004, Hunan, Peoples R China
关键词
CT image segmentation; U-Net network structure; Batch standardization layer; Attention mechanism; Cross entropy loss functions; Dice coefficient;
D O I
10.1007/s11042-021-10841-z
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Medical image segmentation is one of the important steps in clinical diagnosis, and accurate segmentation of lesions is of great significance to clinical treatment. Therefore, a CT image segmentation algorithm based on depth learning is proposed to solve the problems of poor robustness, weak anti noise ability and low segmentation accuracy of existing image segmentation algorithms.. Firstly, we improve the u-net network structure, increase the batch standardization layer to improve the robustness of the network model, and introduce the attention mechanism to focus on specific things according to the needs, improve the recognition ability of the model. Then, the improved U-Net network structure is applied to CT image segmentation, and the cross entropy loss function is used to reduce the possibility of insufficient segmentation and segmentation leakage, and improve the accuracy of image segmentation. Finally, on the basis of data preprocessing, the segmentation network is trained to get the image segmentation model based on deep learning, and the prediction is made on the test set to get the segmentation results. The experimental results show that compared with other algorithms, the proposed method achieves 0.9594 in the common evaluation standard Dice coefficient, and has strong robustness. It can accurately segment the lung organs in CT images, which is helpful for doctors to obtain pathological information and assist in the diagnosis of lung diseases.
引用
收藏
页码:35983 / 36006
页数:24
相关论文
共 50 条
  • [1] An improved segmentation algorithm of CT image based on U-Net network and attention mechanism
    Jin Yang
    Kai Qiu
    Multimedia Tools and Applications, 2022, 81 : 35983 - 36006
  • [2] IECAU-Net: A Wood Defects Image Segmentation Network Based on Improved Attention U-Net and Attention Mechanism
    Dong, Yingda
    He, Chunguang
    Xiang, Xiaoyang
    Cui, Yuhan
    Kang, Yongkang
    Ding, Aiming
    Duo, Huaqiong
    Wang, Ximing
    BIORESOURCES, 2025, 20 (02): : 3545 - 3556
  • [3] 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
  • [4] Tooth CT Image Segmentation Method Based on the U-Net Network and Attention Module
    Tao, Sha
    Wang, Zhenfeng
    Computational and Mathematical Methods in Medicine, 2022, 2022
  • [5] Tooth CT Image Segmentation Method Based on the U-Net Network and Attention Module
    Tao, Sha
    Wang, Zhenfeng
    COMPUTATIONAL AND MATHEMATICAL METHODS IN MEDICINE, 2022, 2022
  • [6] SBC-UNet: A Network Based on Improved Hourglass Attention Mechanism and U-Net for Medical Image Segmentation
    Wane, Yiyi
    Su, Jia
    Gan, Guoxi
    Zhang, Shenmeng
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2024, PT XV, 2025, 15045 : 150 - 164
  • [7] An improved supervised and attention mechanism-based U-Net algorithm for retinal vessel segmentation
    Ma, Zhendi
    Li, Xiaobo
    COMPUTERS IN BIOLOGY AND MEDICINE, 2024, 168
  • [8] CT Image Segmentation of Rectal Tumor Based on U-Net Network
    Wang, Yu-Xue
    Xu, Chao
    Proceedings - 2021 2nd International Conference on Big Data and Informatization Education, ICBDIE 2021, 2021, : 69 - 72
  • [9] Cochlear CT image segmentation based on u-net neural network
    Li, Cheng
    Li, Xiaojun
    Zhou, Rong
    JOURNAL OF RADIATION RESEARCH AND APPLIED SCIENCES, 2023, 16 (02)
  • [10] Semantic segmentation network of uav image based on improved U-net
    Liu, Ziyi
    Huang, Jin
    2019 INTERNATIONAL CONFERENCE ON ADVANCES IN CIVIL ENGINEERING, ENERGY RESOURCES AND ENVIRONMENT ENGINEERING, 2019, 330