Acta Optica Sinica, Volume. 41, Issue 9, 0910002(2021)

End-to-End Segmentation of Brain White Matter Hyperintensities Combining Attention and Inception Modules

Xin Zhao1、*, Xin Wang1, and Hongkai Wang2
Author Affiliations
  • 1School of Information Engineering, Dalian University, Dalian, Liaoning 116622, China
  • 2School of Biomedical Engineering, Dalian University of Technology, Dalian, Liaoning 116024, China
  • show less

    Aim

    ing at solving the problems of low segmentation accuracy in the automatic segmentation of brain white matter hyperintensity region on magnetic resonance imaging brain images and easy to miss small lesions, a U-Net segmentation model combining attention and inception is proposed. In the coding stage of U-NET, the Inception module is added to increase the width of the network, so that it has the ability to extract multi-scale features, and the attention module is added to enhance the attention of the network to the segmentation target. The addition and fusion of the two can effectively improve the feature extraction and expression capabilities of the network. Simultaneously, adding residual connections on each convolutional layer in the decoding stage can improve the optimization speed of the network. In addition, because of the problem that sample imbalance easily leads to too many false negatives in the segmentation results, the Tversky loss function with balance adjustment ability is employed to optimize network training. The experimental results show that the proposed method can segment brain white matter hyperintensity region, especially the small lesion area, and each segmentation index is better than those of multiple comparison methods.

    Tools

    Get Citation

    Copy Citation Text

    Xin Zhao, Xin Wang, Hongkai Wang. End-to-End Segmentation of Brain White Matter Hyperintensities Combining Attention and Inception Modules[J]. Acta Optica Sinica, 2021, 41(9): 0910002

    Download Citation

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

    Category: Image Processing

    Received: Sep. 30, 2020

    Accepted: Dec. 30, 2020

    Published Online: May. 10, 2021

    The Author Email: Zhao Xin (zhaoxin@dlu.edu.cn)

    DOI:10.3788/AOS202141.0910002

    Topics