Optics and Precision Engineering, Volume. 32, Issue 4, 565(2024)

Brain tumor image segmentation based on Semantic Flow Guided Sampling and Attention Mechanism

Jianli SONG1... Xiaoqi LÜ1,2,* and Yu GU1 |Show fewer author(s)
Author Affiliations
  • 1School of Information Engineering, Inner Mongolia University of Science and Technology, Baotou0400, China
  • 2School of Information Engineering, Inner Mongolia University of Technology, Hohhot010051, China
  • show less

    The automatic segmentation method for brain tumors based on a U-shaped network structure often suffers from information loss due to multiple convolution and sampling operations, resulting in suboptimal segmentation results. To address this issue, this study proposed a feature alignment unit that utilizes semantic information flow to guide the up-sampling feature recovery and design designed a lightweight Dual Attention Feature Alignment Network (DAFANet) based on this unit.Firstly, to validate its effectiveness and generalization, the feature alignment unit was introduced separately into three classic networks, namely 3D UNet, DMFNet, and HDCNet. Secondly, a lightweight dual-attention feature alignment network named DAFANet was proposed based on DMFNet. The feature alignment unit enhanced feature restoration in the up-sampling process, and a 3D Expectation-Maximization attention mechanism was applied to both the feature alignment path and cascade path to capture the full contextual dependency. The generalized Dice loss function was also used to improve segmentation accuracy in the case of data imbalance and accelerate model convergence.Finally, the proposed algorithm is validated on the BraTS2018 and BraTS2019 public datasets, achieving segmentation accuracies of 80.44%, 90.07%, 84.57% and 78.11%, 90.10%, 82.21% in the ET, WT, and TC regions, respectively.Compared to current popular segmentation networks, the proposed algorithm demonstrates better segmentation performance in enhancing tumor regions and is more adept at handling details and edge information.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Jianli SONG, Xiaoqi LÜ, Yu GU. Brain tumor image segmentation based on Semantic Flow Guided Sampling and Attention Mechanism[J]. Optics and Precision Engineering, 2024, 32(4): 565

    Download Citation

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

    Category:

    Received: Apr. 5, 2023

    Accepted: --

    Published Online: Apr. 2, 2024

    The Author Email: LÜ Xiaoqi (lxiaoqi@imut.edu.cn)

    DOI:10.37188/OPE.20243204.0565

    Topics