A Multi-Scale Lightweight Brain Glioma Image Segmentation Network

被引:0
|
作者
Yang J. [1 ]
Chen H. [1 ]
Guan X. [1 ]
Li Q. [1 ]
机构
[1] School of Microelectronics, Tianjin University, Tianjin
基金
中国国家自然科学基金;
关键词
Artificial intelligence; Brain tumor; Dilated convolution; Feature fusion; Hybrid loss function; Image segmentation; Multi-scale;
D O I
10.12141/j.issn.1000-565X.220042
中图分类号
学科分类号
摘要
Manual segmentation of brain tumor areas in magnetic resonance imaging (MRI) images is time-consuming and laborious, and it can be easily influenced by individual subjectivity. To reliably and efficiently segment brain tumors semi-automatically or automatically is particularly important for medically assisted diagnosis. In recent years, convolutional neural network-based methods for automatic segmentation of brain tumor images have made great progress, but the existing methods still cannot effectively fuse features in terms of large-scale contours and small-scale texture details of tumor images, and the rich global background information is ignored during training. In view of these problems, this paper proposed a multi-scale lightweight brain tumor image segmentation network MSL-Net. First, the base convolution in the U-Net network was replaced with an improved hierarchical decoupled convolution to expand the perceptual field while efficiently exploring multi-scale multi-view spatial information. Then, a bidirectional feature pyramid network structure was introduced at the skipping connection to fuse multi-scale features, and a hybrid loss function combining the generalized Dice loss function and the Focal loss function was used to improve segmentation accuracy and accelerate convergence in the case of pixel count imbalance between tumor and non-tumor regions. Experimental results on the BraTS 2019 dataset show that the Dice similarity coefficients of the proposed MSL-Net network in the overall tumor region, core tumor region and enhanced tumor region are 0.900 3, 0.830 6 and 0.777 0, respectively, and the number of parameters and computation (floating-point operations per second) are 3.9×105 and 3.16×1010, respectively. Compared with the current state-of-the-art methods, the method proposed in the paper achieves high segmentation accuracy while achieving light weight. © 2022, Editorial Department, Journal of South China University of Technology. All right reserved.
引用
收藏
页码:132 / 141
页数:9
相关论文
共 27 条
  • [1] TABATABAI G, STUPP R, van den BENT M J, Et al., Molecular diagnostics of gliomas:the clinical perspective[J], Acta Neuropathologica, 120, 5, pp. 585-592, (2010)
  • [2] WADHWA A, BHARDWAJ A, VERMA V S., A review on brain tumor segmentation of MRI images[J], Magnetic Resonance Imaging, 61, pp. 247-259, (2019)
  • [3] CUI S, MAO L, JIANG J, Et al., Automatic semantic segmentation of brain gliomas from MRI images using a deep cascaded neural network[J], Journal of Healthcare Engineering, 2018, (2018)
  • [4] PINTO A, PEREIRA S, RASTEIRO D, Et al., Hierarchical brain tumor segmentation using extremely randomized trees[J], Pattern Recognition, 82, pp. 105-117, (2018)
  • [5] DEEPAK S, AMEER P M., Brain tumor classification using deep CNN features via transfer learning[J], Computers in Biology and Medicine, 111, (2019)
  • [6] TIWARI A, SRIVASTAVA S, PANT M., Brain tumor segmentation and classification from magnetic resonance images:review of selected methods from 2014 to 2019[J], Pattern Recognition Letters, 131, pp. 244-260, (2020)
  • [7] HAVAEI M, DAVY A, WARDE-FARLEY D, Et al., Brain tumor segmentation with deep neural networks[J], Medical Image Analysis, 35, pp. 18-31, (2017)
  • [8] ZHAO X, WU Y, SONG G, Et al., A deep learning model integrating FCNNs and CRFs for brain tumor segmentation[J], Medical Image Analysis, 43, pp. 98-111, (2018)
  • [9] LONG J, SHELHAMER E, DARRELL T., Fully convolutional networks for semantic segmentation[C], Proceedings of 2015 IEEE Conference on Computer Vision and Pattern Recognition, pp. 3431-3440, (2015)
  • [10] RONNEBERGER O, FISCHER P, BROX T., U-Net:convolutional networks for biomedical image segmentation[C], Proceedings of the 18th International Confe-rence on Medical Image Computing and Computer-Assisted Intervention, pp. 234-241, (2015)