Laser & Optoelectronics Progress, Volume. 57, Issue 14, 141030(2020)

Magnetic Resonance Brain Tumor Image Segmentation Based on Attention U-Net

Lingmei Ai1、*, Tiandong Li1、**, Fuyuan Liao2, and Kangzhen Shi1
Author Affiliations
  • 1School of Computer Science, Shaanxi Normal University, Xi'an, Shaanxi 716000, China
  • 2School of Electronic Information Engineering, Xi'an Technological University, Xi'an, Shaanxi 716000, China
  • show less

    Herein, U-Net structure was improved to segment magnetic resonance (MR) images of brain tumors to address the loss of information in image segmentation in the full convolutional neural network and low segmentation accuracy caused by fixed weights. Based on the attention module in the U-Net contraction path, the weights were distributed to different size convolutional layers, which is beneficial to information usage for image space and context. Replacing the original convolution layer with the residual compact module can extract more features and promote network convergence. The brain tumor MR image database provided by BraTS (The Brain Tumor Image Segmentation Challenge) is used to validate the proposed new model and evaluate the segmentation effect using the Dice score. The accuracy of 0.9056, 0.7982, and 0.7861 was obtained in the total tumor region, core tumor region, and tumor enhancement, respectively, demonstrating that the proposed U-Net structure can enhance the accuracy and efficiency of MR image segmentation.

    Tools

    Get Citation

    Copy Citation Text

    Lingmei Ai, Tiandong Li, Fuyuan Liao, Kangzhen Shi. Magnetic Resonance Brain Tumor Image Segmentation Based on Attention U-Net[J]. Laser & Optoelectronics Progress, 2020, 57(14): 141030

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Nov. 6, 2019

    Accepted: Dec. 31, 2019

    Published Online: Jul. 28, 2020

    The Author Email: Ai Lingmei (almsac@163.com), Li Tiandong (litiandong@snnu.edu.cn)

    DOI:10.3788/LOP57.141030

    Topics