An Efficient Implementation of Deep Convolutional Neural Networks for MRI Segmentation

J Digit Imaging. 2018 Oct;31(5):738-747. doi: 10.1007/s10278-018-0062-2.

Abstract

Image segmentation is one of the most common steps in digital image processing, classifying a digital image into different segments. The main goal of this paper is to segment brain tumors in magnetic resonance images (MRI) using deep learning. Tumors having different shapes, sizes, brightness and textures can appear anywhere in the brain. These complexities are the reasons to choose a high-capacity Deep Convolutional Neural Network (DCNN) containing more than one layer. The proposed DCNN contains two parts: architecture and learning algorithms. The architecture and the learning algorithms are used to design a network model and to optimize parameters for the network training phase, respectively. The architecture contains five convolutional layers, all using 3 × 3 kernels, and one fully connected layer. Due to the advantage of using small kernels with fold, it allows making the effect of larger kernels with smaller number of parameters and fewer computations. Using the Dice Similarity Coefficient metric, we report accuracy results on the BRATS 2016, brain tumor segmentation challenge dataset, for the complete, core, and enhancing regions as 0.90, 0.85, and 0.84 respectively. The learning algorithm includes the task-level parallelism. All the pixels of an MR image are classified using a patch-based approach for segmentation. We attain a good performance and the experimental results show that the proposed DCNN increases the segmentation accuracy compared to previous techniques.

Keywords: Deep convolutional neural networks; Deep learning; Image segmentation; MRI; MRI segmentation; Medical image.

MeSH terms

  • Algorithms
  • Brain / diagnostic imaging
  • Brain Neoplasms / diagnostic imaging*
  • Deep Learning
  • Humans
  • Image Interpretation, Computer-Assisted / methods*
  • Machine Learning
  • Magnetic Resonance Imaging / methods*
  • Neural Networks, Computer*