Multi-level Glioma Segmentation using 3D U-Net Combined Attention Mechanism with Atrous Convolution

被引:0
作者
Cheng, Jianhong [1 ]
Liu, Jin [1 ]
Liu, Liangliang [1 ]
Pan, Yi [2 ]
Wang, Jianxin [1 ]
机构
[1] Cent South Univ, Sch Comp Sci & Engn, Changsha, Peoples R China
[2] Georgia State Univ, Dept Comp Sci, Atlanta, GA 30303 USA
来源
2019 IEEE INTERNATIONAL CONFERENCE ON BIOINFORMATICS AND BIOMEDICINE (BIBM) | 2019年
基金
中国国家自然科学基金;
关键词
glioma segmentation; 3D U-Net; attention mechanism; atrous convolution;
D O I
10.1109/bibm47256.2019.8983092
中图分类号
Q5 [生物化学];
学科分类号
071010 ; 081704 ;
摘要
Accurate segmentation of glioma from 3D medical images is vital to numerous clinical endpoints. While manual segmentation is subjective and time-consuming, fully automated extraction is quite imperative and challenging due to the intrinsic heterogeneity of tumor structures. In this study, we propose a multi-level glioma segmentation framework, 3D Residual-Attention-Atrous U-Net (RAAU-Net), using 3D U-Net combined attention mechanism with atrous convolution. The 3D RAAU-Net can extract contextual information by combining low- and high-resolution feature maps. The attention mechanism is embedded in each skip connection layer of 3D RAAU-Net to enhance feature representations. Meanwhile, the atrous convolution is adopted in the whole network architecture to incorporate large and rich semantic information. Furthermore, we design a new training scheme to reduce false positives and enhance generalization. Eventually, our proposed segmentation method is evaluated on the validation dataset from the Multimodal Brain Tumor Image Segmentation Challenge (BraTS) 2018 and achieve a competitive result with average Dice score of 88% for the whole tumor, 79% for the tumor core and 73% for the enhancing tumor, respectively. Quantitative results and visual analysis have proven that these improvements in 3D RAAU-Net are effective and achieve a better segmentation accuracy compared with the baseline.
引用
收藏
页码:1031 / 1036
页数:6
相关论文
共 28 条
  • [1] [Anonymous], 2017, MED IMAGE ANAL, DOI DOI 10.1016/J.MEDIA.2016.10.004
  • [2] [Anonymous], 2017, INFORM PROCESSING ME, DOI DOI 10.1007/978-3-319-59050-9_28
  • [3] [Anonymous], 2017, SCI DATA, DOI DOI 10.1038/SDATA.2017.117
  • [4] [Anonymous], 2016, INT CONF 3D VISION, DOI DOI 10.1109/3DV.2016.79
  • [5] [Anonymous], 2019, BIG DATA MINING ANAL
  • [6] [Anonymous], 2018, IEEE RADIO WIRELESS
  • [7] [Anonymous], 2018, INT C CONTROL DECISI
  • [8] [Anonymous], 2018, IEEE T PATTERN ANAL, DOI DOI 10.1109/TPAMI.2017.2699184
  • [9] [Anonymous], 2019, NEUROCOMPUTING, DOI DOI 10.1016/J.NEUCOM.2018.04.080
  • [10] [Anonymous], 2014, TSINGHUA SCI TECHNOL