Advertisement
Browse Subject Areas
?

Click through the PLOS taxonomy to find articles in your field.

For more information about PLOS Subject Areas, click here.

  • Loading metrics

A deep convolutional neural network-based automatic delineation strategy for multiple brain metastases stereotactic radiosurgery

  • Yan Liu,

    Roles Conceptualization, Data curation, Formal analysis, Investigation, Methodology, Software, Validation, Writing – original draft, Writing – review & editing

    Affiliations School of Electrical Engineering and Information, Sichuan University, Chengdu, Sichuan, China, Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Strahinja Stojadinovic,

    Roles Data curation, Writing – review & editing

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Brian Hrycushko,

    Roles Writing – review & editing

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Zabi Wardak,

    Roles Conceptualization, Data curation, Writing – review & editing

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Steven Lau,

    Roles Formal analysis, Methodology, Writing – review & editing

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Weiguo Lu,

    Roles Software, Writing – review & editing

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Yulong Yan,

    Roles Software, Writing – review & editing

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Steve B. Jiang,

    Roles Resources

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Xin Zhen,

    Roles Formal analysis, Methodology, Software, Writing – review & editing

    Affiliations Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America, Department of Biomedical Engineering, Southern Medical University, Guangzhou, Guangdong, China

  • Robert Timmerman,

    Roles Conceptualization, Data curation, Writing – review & editing

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Lucien Nedzi,

    Roles Conceptualization, Data curation, Writing – review & editing

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

  • Xuejun Gu

    Roles Conceptualization, Data curation, Formal analysis, Methodology, Project administration, Resources, Software, Supervision, Validation, Writing – original draft, Writing – review & editing

    xuejun.gu@utsouthwestern.edu

    Affiliation Department of Radiation Oncology, The University of Texas Southwestern Medical Center, Dallas, TX, United States of America

A deep convolutional neural network-based automatic delineation strategy for multiple brain metastases stereotactic radiosurgery

  • Yan Liu, 
  • Strahinja Stojadinovic, 
  • Brian Hrycushko, 
  • Zabi Wardak, 
  • Steven Lau, 
  • Weiguo Lu, 
  • Yulong Yan, 
  • Steve B. Jiang, 
  • Xin Zhen, 
  • Robert Timmerman
PLOS
x

Abstract

Accurate and automatic brain metastases target delineation is a key step for efficient and effective stereotactic radiosurgery (SRS) treatment planning. In this work, we developed a deep learning convolutional neural network (CNN) algorithm for segmenting brain metastases on contrast-enhanced T1-weighted magnetic resonance imaging (MRI) datasets. We integrated the CNN-based algorithm into an automatic brain metastases segmentation workflow and validated on both Multimodal Brain Tumor Image Segmentation challenge (BRATS) data and clinical patients' data. Validation on BRATS data yielded average DICE coefficients (DCs) of 0.75±0.07 in the tumor core and 0.81±0.04 in the enhancing tumor, which outperformed most techniques in the 2015 BRATS challenge. Segmentation results of patient cases showed an average of DCs 0.67±0.03 and achieved an area under the receiver operating characteristic curve of 0.98±0.01. The developed automatic segmentation strategy surpasses current benchmark levels and offers a promising tool for SRS treatment planning for multiple brain metastases.

Introduction

The incidence of brain metastases has increased with the advanced modern cancer therapy technology and prolonged cancer survival. [1]. Stereotactic radiosurgery (SRS), a standard of care for brain metastases [2], requires accurate delineation of tumor/target volumes for treatment planning, but manually contouring multiple brain metastases can be a time- and labor-intensive process. Developing an accurate and efficient automated delineation tool would benefit clinical practice by improving the efficiency of SRS treatment planning.

Researchers have been investigating automatic brain tumor segmentation methods for decades [3,4] and have developed various tools [57]. Currently available automatic brain tumor delineation methods can be divided into two categories: information theory-based methods and learning-based methods [3]. The information theory-based methods use image data itself and utilize traditional image processing tools to detect abnormalities. Exemplary algorithms include watershed segmentation algorithms, active contour algorithms, and region-growing segmentation algorithms [811]. Learning-based methods consider segmentation tasks as classification problems and require a certain number of expert-segmented images to train classification models. On those expert-segmented images, manually-designed image features, such as mean, standard deviation, gray level co-occurrence matrix (GLCM), and local binary pattern features (LBP), are extracted and fed into machine learning models, such as support vector machine (SVM) or artificial neural network (ANN), to classify target abnormalities [1217].

Automatic brain metastases segmentation requires special considerations in its clinical implementation. First, SRS is often used to treat small tumors, e.g. the tumor diameter is less than 1cm [2], which are easy to miss in information theory-based segment methods. Second, clusters of brain metastases complicate automatic segmentation, and simultaneous delineation of multiple lesions is difficult. Third, contrast-enhanced T1 magnetic resonance imaging (MRI) is generally the only imaging modality acquired for treatment planning, which eliminates the application of many advanced multi-modality image segmentation tools. Furthermore, SRS is usually a one-day outpatient procedure, which requires fast segmentation for a rapid clinical workflow.

Recently, we developed an automatic brain metastatic tumor segmentation strategy for an SRS clinical workflow [7]. The developed strategy integrates a set of traditional image processing tools and takes advantage of image intensity information to discriminate tumor regions from surrounding tissues. This method achieved high accuracy in automatic contouring in both simulated data and clinical patient image sets. However, the developed method has difficulties in segmenting small brain metastases with volumes less than 1.500cc, especially when the tumor is surrounded by other high intensity structures, such as the superior sagittal sinus or a confluence of sinuses. Its intensity histogram-based abnormalities-detecting strategy hinders its application in segmenting small-size tumors, since a limited number of voxels in a small-size has an undistinguishable intensity histogram from surrounding structures. Conventional artifact feature-based learning methods could fail with small-size tumor segmentation as well, because the limited number of voxels is unlikely to provide statistically significant features for segmentation. Deep learning convolutional neural networks (CNN), which require neither manual image feature extraction nor tumor intensity histogram, may have advantages in classifying small-size abnormalities. Advantage in classifying small targets have been proved in reference [18] [19]. The CNN algorithm utilizes a stack of sequentially connected convolutional filters to study the nonlinear relationship between abnormal voxels and their neighbors, and automatically derives a voxel characterization model. This self-learning procedure is promising for small-size tumor segmentation.

In this paper, we report our newly developed CNN-based brain metastases auto-segmentation strategy. This work aims to delineate the small lesions (<1.5cc) accurately and efficiently. The network architecture is branchy and made up of three sub-paths to incorporate multi-scale information to perform delineation of small lesions on MRI.

Methods and materials

Ethics statement

This retrospective patient study was approved by Human Research Protection Program Office (HRPPO)/Institutional Review Board (IRB) of The University of Texas Southwestern Medical Center. All methods in this study were conducted in accordance with the relevant guidelines and regulations. Considering that this is not a therapeutical treatment study, our institutional review board waived the need for obtaining written informed consent from the participants.

Automatic delineation workflow

The auto-segmentation workflow that we developed is illustrated in Fig 1. The entire workflow can be divided into three sections: image preprocessing, segmentation, and false positive contour removal. The first section, image preprocessing, removes the skull from the original T1c image by employing a robust learning-based MRI brain extraction system (ROBEX) [20]. ROBEX combines a discriminative and a generative model to achieve the final result. When a new image is presented to ROBEX, the tool uses a Random Forest classifier to detect the brain boundary. Then the generative model is explored to find a highest likelihood contour. The brain contour is refined by free deformation and used for skull stripping. The third section removes false positive contours by utilizing the spherical geometry characteristics of brain metastases. Essentially, we use a sphericity metric [21] to quantify the delineated structure shape. The structure with a sphericity value smaller than a predefined threshold is removed from the final contour sets. The intermediate second step is a learning-based segmentation method consisting of a CNN architecture, which we will describe in detail in the following sections.

thumbnail
Fig 1. Workflow of the proposed deep learning-based brain metastases auto-segmentation.

https://doi.org/10.1371/journal.pone.0185844.g001

A Modified DeepMedic CNN architecture

A CNN framework treats segmentation as a voxel-wise classification problem. The lesion can be delineated from the background if the probability of each image voxel belonging to the target is known. Our work was inspired by DeepMedic, a CNN architecture proposed by Kamnitsas et al. [22]. The network architecture consists of a sequence of four sections: input, convolution, fully connected, and classification sections, as illustrated in Fig 2. The input section processes the original images to obtain the designed image patches. The designed image patches are then fed into a convolution section, where multi-layer convolutional filters operate and output feature maps. The convolution section is followed by a fully connected layer that groups all feature maps. The final classification section calculates a prediction score to classify each image voxel and yield a segmentation map. Our proposed CNN method enhances the original DeepMedic structure by including an additional sub-path (sub-path 2) to capture multi-scale image features for accurate image segmentation. In addition, the entire structure is implemented on a graphic processing unit (GPU) platform to improve computation efficiency. For simplicity, we call this modified DeepMedic CNN architecture “En-DeepMedic” in this paper.

thumbnail
Fig 2. En-DeepMedic architecture.

αl@3DConv (k1,k2,k3): αl is the number of feature maps captured at layer l, (k1,k2,k3) is convolution filters size.

https://doi.org/10.1371/journal.pone.0185844.g002

Input section.

The input section generates image patches for the rest of the network. En-DeepMedic is a voxel-wise classification system, where each voxel is classified based on the linear and non-linear relationship between the focal voxel’s intensity and its neighbors. Because of the large image size, as the case of the 3D brain MRI images studies in the paper, calculating the linear or nonlinear relationship between all voxels in the entire image is computation-intensive. We divide the image into small patches to calculate the voxel relationship within a limited region instead of the entire image. Using small image patches saves computational time and memory space. Such strategy is important for GPU implementation; especially on-chip memory is limited on the current market-available GPU cards.

The En-DeepMedic architecture extracts both local and global patches as inputs for the convolution section. Fig 3 illustrates a two-dimensional patch extraction strategy, though the actual image patches used in the algorithm are three-dimensional. Each extraction selected the central voxel at random and simultaneously extracted concentric local and global image patches. Neighboring voxels around the central voxels form the local patch, which provides local information. The global patch covers a larger region, which contains relative global information. To mitigate the computational burden caused by the larger global patch, we down-sampled all global patches. In this study, we specified the local image patch size as (25, 25, 25) and down-sampled the global patches from (57, 57, 57) to (19, 19, 19).

Modified convolution section.

The convolution section has multiple layers that sequentially capture the image features by convolution operations. These captured image features include low-level features, such as edge and corner, and high-level features, such as non-linear intensity relationship between neighboring voxels. The feature maps are the output of each convolution layer. The entire convolution section comprises three different sub-paths. Sub-paths 1 and 2 use the same local patch with different filter sizes to capture different neighborhood patterns, while sub-path 3 works on the down-sampled global patch to reveal global features. For each sub-path, let L be the depth of the convolution filter stack, which is the number of convolution layers used in a sub-path. αl (l ∈ [1,L]) be the number of feature maps in the lth layer, and (i ∈ [1,αl]) be the ith feature map of the lth layer. In this cascaded network, is calculated by nonlinearly activating the convolution of the l − 1th layer feature map and the lth layer filter with certain bias, as shown in Eq 1: (1) Here, g(∙) is the PReLU function [23]. The PReLU is a neuron activation function defined as , where a is a network parameter. is the filter connecting the jth feature map in the l − 1th layer and the ith feature map in the lth layer, and is the bias from the artificial neuron model [24]. A typical 3D convolutional operation is illustrated in Fig 4(A). The generation of the ith feature map in the lth layer is illustrated in Fig 4(B). The filter number of layer l is αl−1αl. The total filter coefficient number of each individual convolutional layer stack is , where is the filter size of the lth layer. As there are 3 sub-paths in this section, the total number of to-be-learned filter coefficients is ∑pNp, p = 1,2,3. With a filter size of (3,3,3) in sub-path 1 (and 3) and (5,5,5) in sub-path 2, a local patch size of (25,25,25), and a down-sampled global patch size of (19, 19, 19), the size of the output feature map is (9,9,9) in sub-path 1 and 2 and (3,3,3) in subpath 3.

thumbnail
Fig 4. Process of 3D convolution layer.

(a) 3D convolution of a feature map with a filter. (b) Generation of the ith feature map in the lth layer.

https://doi.org/10.1371/journal.pone.0185844.g004

Compared to the original DeepMedic method, our En-DeepMedic is distinguished by an additional sub-path (sub-path 2) included in the convolution section. Sub-path 2 uses a larger convolution filter (5, 5, 5) and operates on every other convolution layer, unlike sub-path 1. This design is useful for capturing different multiscale features in addition to those captured by sub-path 1.

Fully connected section.

The fully connected (FC) section fuses all of the feature maps generated by the convolution section to preserve the consistency of spatial information. The FC section operates similarly to the convolution section, where each coefficient node works as a convolution filter with size (1,1,1). It treats the output feature maps of the three sub-paths equally, which requires feature maps to have the same size.

Classification section.

The classification section generates the categorical probability for each voxel. Like most convolutional neural networks, a softmax function maps the feature map to categorical probabilities. A cost function (Eq 2) is defined to maximize the logarithmic likelihood between the input patch, Is, and the corresponding ground truth segmented patch, cs. (2) where xv and are the vth target voxel’s position and ground truth, v ∈ [1,V]. , where is the Lth layer feature map size. s ∈ [1,B], where B is the size of a batch. is the output of the softmax function. The parameter Θ is denoted as the filter coefficients and bias, which are determined during network training through the Stochastic Gradient Descent (SGD) method[25]. Here, the network training refers to using a set of segmented images (called labelled images) to find the optimal network parameters. SGD is a stochastic approximation of the gradient descent optimization method for minimizing an objective function iteratively. In each iteration, it calculates the gradient from a subset of labelled dataset. It helps to speed up the training in a large training set.

Hyper parameters.

Before training, we predefined a set of hyper parameters, which could not be learned automatically through training. The hyper parameters used in our En-DeepMedic architecture are listed in Table 1 along with their associated workflows. Initialization generates initial coefficients for the filter weights and bias. Instead of using random values, we employed the method used by He et al. [23] to generate initial coefficients. The Dropout technique [26] avoids overfitting by randomly discarding part of the obtained coefficients in each iteration. We performed the dropout scheme on the FC section only with a dropout rate of 0.5. Data augmentation balances positive and negative samples, because there is a large gap between the number of tumor and non-tumor voxels. All positive patches were flipped along the sagittal plane to increase their numbers. A k-fold cross validation strategy trains, validates, and tests our En-DeepMedic method. We randomly partitioned the original dataset into k subsets with equal size and performed the training and validation phases k times with one subset for validation and the other k-1 subsets for training. Batch size and epochs depend on computer memory and could be modified based on the machine architecture.

Validation dataset

We used two groups of data to train and validate our model. One dataset was from BRATS [27], a database for evaluating brain tumor segmentation methods. BRATS provides each patient’s T1-weighted MRI with Gadolinium contrast (T1c) and T2-weighted Fluid-Attenuated Inversion Recovery (FLAIR) images. We used this dataset to compare our automatic delineation method with other competitive algorithms. All images in the dataset were resized to 1.0mm × 1.0mm × 1.0mm after skull removal. The other dataset consisted of 240 brain metastases patients with T1c MRI scans collected at the University of Texas Southwestern Medical Center (UTSW) from 2009 to 2014 [28]. The number of brain metastases per case varies from 1 to 93 (5.679 ± 8.917 per case). The mean tumor size is 0.672 ± 1.994cc. All scans were acquired on a SIEMENS 3T MRI system prior to radiosurgery on the treatment day. The brain metastases contours drawn by neurosurgeons following the standard UTSW clinical protocol were used as ground-truth in this study. Each data group was divided into three sets: training, validation, and testing. Both training and validation sets were used in the training phase to generate the proper network coefficients. The testing set was used to evaluate the trained model. BRATS (265 cases) and clinical data (225 cases) were selected and randomly assigned to the training and validation sets to generate the network coefficients.

Evaluation metrics

We evaluated the performance of the En-DeepMedic-based auto-delineation method with respect to both geometrical measurements of individual tumor segmentation accuracy and statistical measurements of receiver operator characteristic (ROC) curve analysis. We quantitatively evaluated the geometrical accuracy of individual tumor segmentation with 1) DICE coefficients (), where A and B are the auto- and manual-segmented volumes, respectively; and 2) the mean value and standard deviation of surface-to-surface distance (SSD): MSSD = meancC(mindDcd2), SDSSD = stdcC(mindDcd2), where c and d are points on the A and B volumes’ surfaces C and D, respectively. For statistical measurement, we conducted a ROC curve analysis, where sensitivity () and specificity () were calculated with voxel-wised TP (true positive), TN (true negative), FN (false negative), and FP (false positive). Here, TP and TN are the number of the auto-delineation method classified tumor voxels and non-tumor voxels that agreed with the ground truth. FN is the number of voxels classified as tumor voxels in the ground truth but missed by the auto-delineation, and FP is the number of voxels classified as tumor voxels by the auto-delineation but not in the ground truth. We performed the ROC curve analysis by plotting the curve of sensitivity against (1-specificity). We calculated the Areas under the ROC curves (AUC) to quantify the classification performance.

Results

Validation on BRATS

The BRATS database includes five different target structures: necrosis (label 1), edema (label 2), non-enhanced tumor (label 3), enhanced tumor (label 4), and background. In addition to the individual labeled structures, BRATS defines the whole tumor region, which consists of four structures (label 1+2+3+4), and the tumor core, which consists of necrosis, enhanced and non-enhanced tumor (label 1+3+4). As described by Menze et al. [27], the edema is segmented on T2 and FLAIR images, and the gross tumor is delineated on T1c together with T1 images, while necrosis and enhanced tumor are segmented on T1c images. Non-enhanced tumor is a residual volume derived by subtracting necrosis and enhanced tumor volumes from the gross tumor volume.

Multi-modality segmentation–Comparison with methods in BRATS.

For fair comparison with the multi-modality-based methods in BRATS, we trained both the DeepMedic architecture and our En-DeepMedic architecture with T1c and FLAIR images. Table 2 lists the mean DC (± standard deviation, SD) for each method. Compared with other algorithms in the BRATS2015 challenge, our En-DeepMedic method achieved better DCs than most of the reported methods (Table 2).

T1c segmentation–Comparison with DeepMedic.

Since our method is primarily intended for mono-modality segmentation, we trained the model using only the T1c image with a 5-fold cross validation. The quantitative results given by geometrical metrics are shown in Fig 5. The mean value of each metric resulting from DeepMedic is also plotted using magenta stars in the figure for the comparison.

thumbnail
Fig 5. Box plots of geometrical metrics.

(a) DCs, (b) MSSD, and (c) SDSSD of 5-fold cross validation with BRATS data using En-DeepMedic. The red line and cyan diamond inside each box denote medium value and mean value, respectively. The magenta star indicates the mean value of the geometrical metrics results from DeepMedic.

https://doi.org/10.1371/journal.pone.0185844.g005

The ROC curves measuring the classification performance are plotted in Fig 6. The red curve indicates the developed En-DeepMedic model, and the blue curve indicates the DeepMedic model. We calculated the AUC of the two curves and obtained 0.99 for the developed En-DeepMedic method and 0.97 for DeepMedic.

thumbnail
Fig 6. ROC curves for using DeepMedic and En-DeepMedic methods for BRATS data segmentation.

https://doi.org/10.1371/journal.pone.0185844.g006

Validation with clinical data

Sample cases.

We present two sample cases to illustrate En-DeepMedic performance on small tumor segmentation. The first case (named Pt. #1) is a 65-year-old male patient who underwent Gamma Knife radiosurgery for brain metastasis of melanoma. There were three lesions, including a 0.221cc volume in the left mid frontal, a 0.276cc volume lesion in the left paracentral lobule, and a 0.293cc volume in the right globus pallidus, respectively. The distribution of the lesions is shown in Fig 7(A), where the red contours represent the ground truth, and the green contours illustrate the result of the En-DeepMedic method. Fig 7(B), 7(C) and 7(D) show three lesions’ delineation results, each with coronal, sagittal, and transverse views. The auto-delineated contours overlap with the manual ones and achieve an average DC of 0.84. The second case (named Pt. #2) is a 47-year-old female patient who underwent Gamma Knife radiosurgery for brain metastasis of melanoma. This patient has one lesion located in the left choroid plexus with size of 0.194cc and another previous treated lesion in the post temporal lobe. The delineation results on 3 orthogonal cross sections views are shown in Fig 8. The quantitative evaluation of these two patients segmentation results are listed in Table 3.

thumbnail
Fig 7. Segmentation results of Pt #1.

(a) 3D rendering of the tumor locations inside the brain; (b)~(d): segmentation of lesion in (b) the right globus pallidus, (c) the left paracentral lobule, and (d) left mid frontal.

https://doi.org/10.1371/journal.pone.0185844.g007

thumbnail
Fig 8. Segmentation results of Pt #2.

(a) Segmentation of lesion in the left choroid plexus; (b): zoomed-in of the transverse view in (a).

https://doi.org/10.1371/journal.pone.0185844.g008

thumbnail
Table 3. Quantitative geometrical accuracy of each lesion of sample cases.

https://doi.org/10.1371/journal.pone.0185844.t003

Overall performance.

We employed a 5-fold cross validation strategy to estimate the overall performance of the segmentation strategy. The resulting mean values for the geometrical metrics are: DCs 0.67±0.03, MSSD 0.9±0.3mm, and SDSSD 0.8±0.1mm. The detailed distributions of each geometrical metric are shown in Fig 9. The ROC curves of the En-DeepMedic classifier from each fold are plotted in Fig 10 by the five colored solid lines, and the red dotted line indicates the average curve. AUC is 0.98±0.01, close to 1.

thumbnail
Fig 9. Box plots of clinical patient data 5-fold cross validation results.

(a) DCs, (b) MSSD, and (c) SDSSD using the En-DeepMedic method. The red line and the cyan diamond inside each box denote medium value and mean value, respectively.

https://doi.org/10.1371/journal.pone.0185844.g009

thumbnail
Fig 10. ROC curves of using En-DeepMedic for patient data segmentation.

https://doi.org/10.1371/journal.pone.0185844.g010

Comparison with our previous intensity-based segmentation method

We compared the En-DeepMedic method with our previous intensity-based method [7]. We selected ten tumors with different sizes, ranging from 6.918cc to 0.129cc. Both methods delineated the large lesions (≥1.500cc) successfully. Fig 11 illustrates the delineation results of a large tumor (2.532cc) and a small tumor (0.537cc). Table 4 lists the corresponding geometrical metrics values from both methods. The small lesions were more challenging, and the intensity-based method failed to delineate 4 of the 6 metastases smaller than 1.500cc. In contrast, the En-DeepMedic performed well with the small lesions.

thumbnail
Fig 11. Segmentation results.

(a) the large tumor (2.532cc), auto-contours from intensity-based method (blue) and En-DeepMedic method (green) overlaid on physician drawn contour (red); (b) zoomed-in view of ROI in orange rectangle in (a); (c) the small tumor (0.537cc), auto-contours from En-DeepMedic method (green) overlaid on physician drawn contour (red); (d) zoomed-in view of ROI in orange rectangle in (c).

https://doi.org/10.1371/journal.pone.0185844.g011

thumbnail
Table 4. Comparison geometrical metric values of En-DeepMedic and intensity-based method on small- and large-size brain metastases tumors.

https://doi.org/10.1371/journal.pone.0185844.t004

Discussion

In this study, we presented En-DeepMedic, a CNN-based delineation method designed for automatic brain metastases segmentation for radiosurgery. To compare with other algorithms, the developed En-DeepMedic method was trained on multi-modality BRATS data and achieved good DCs. The competitive results obtained from BRATS2015 are shown in Table 2. The developed method ranked first in the enhanced tumor region (label 4 in BRATs) and second in the tumor core (label 1+3+4). This result indicates that En-DeepMedic performs well on multimodality image segmentation. Also, our En-DeepMedic is mainly designed for delineating small lesions. It performs well on BRAST2015 dataset which accurately doesn’t have much small size brain tumors. We believe if we adjust some hyper-parameters and image patch size, the En-DeepMedic could achieve even better performance. Our En-DeepMedic also performed better than its parent DeepMedic method on mono-modality segmentation (Figs 5 and 6). We achieved this improvement by adding an additional sub-path (sub-path 2) to the architecture, as shown in Fig 2. Sub-path 2 employed larger convolution filters (5,5,5) compared to the filter size (3,3,3) in sub-path 1, and thus captures different image features than sub-path 1. The hybrid features learned from sub-paths of different filter sizes and multiple scales can provide comprehensive local and global information for accurate classification of voxels. Our En-DeepMedic method also performed well in evaluating clinical data with satisfactory geometric accuracy and ROC curve analysis (Figs 9 and 10). The achieved AUC was 0.98±0.01, which indicates good performance.

The sample case shown in Fig 8 didn’t achieve high DC with En-DeepMedic method, even though the location of lesion was correctly identified. The main reason for this low DC is that two contours are merged together. The targeted lesion in the left choroid plexus is close to the abnormality in the post temporal lobe that was treated 5 months ago. The shortest separation between two lesions boundaries is ~ 1mm, which is the size about or smaller than the voxel resolution as highlighted by the blue circle in Fig 8(B). Our En-DeepMedic delineation method was failed to separate them. To overcome this problem, contour post-processing is needed.

Compared to our previous intensity-based method, the En-DeepMedic method achieved superior results in small tumor segmentation, while the previous method often fails to detect tumors smaller than 1.500cc. We chose 10 cases that varied from 6.918cc to 0.129cc and used both methods to conduct segmentation. Both methods succeeded in auto-segmenting lesions larger than 1.500cc, but the intensity-based method failed on four of the six small cases, while the En-DeepMedic method worked robustly (Table 4). The two methods are, however, designed for different applications [7]. The intensity-based method can capture abnormalities without any prior information; the only input is the image set’s intensity. Generally, this method requires larger tumor volumes that have adequate voxels representing discriminative information. This constraint limits its application for small lesion segmentation. The En-DeepMedic method is better suited to small lesion segmentation, as it employs a deep learning strategy and classifies each voxel individually based on knowledge learned from a training dataset. This method considers multi-scale information and the nonlinear relationship between neighboring voxels. Theoretically, this voxel-wise segmentation strategy has the capability to segment a tumor, even if it is formed by only one voxel. However, the En-DeepMedic method results in a relatively zigzag contour when delineating large targets compared to our previous intensity-based method (Fig 11(B)), because the voxel-wise classification strategy cannot guarantee a smooth contour, while our previous method includes a contour evolving step that smoothes contours automatically. To overcome this limitation, additional postprocessing methods, such as using a localized active contour model [36], could be employed to refine the final contours, but the postprocessing step would increase computation time.

The input image patch size affects the performance of En-DeepMedic algorithm. A large image patch size requires a big memory allocated on the GPU. Our study is conducted on a laptop with an NVIDIA Quadro M2000M graphic card and an Intel Xeon E3-1505 processor. The largest local image patches the computer can handle is (37,37,37). We evaluate the En-DeepMedic performance with local patch sizes varied from (19,19,19) to (37,37,37) on our clinical dataset. One thing we have to point out is that some hyper parameters (training, validating, and testing batch size) are revised to fit GPU computation resource limit. For the local patch size (19,19,19) and (25,25,25), batch sizes are the same as those given in Table 1. For the local patch size (31,31,31), the batch size for training, validation and testing are 5,9, 10, respectively. For the local patch size (37,37,37), the batch size for training, validation and testing are revised as 1, 9 and 10, respectively. The results of quantitative geometry metrics are listed in Table 5 and ROC curve are plotted in Fig 12. As results shown, the best The best geometric metric and ROC are both achieved at a local patch size of (25,25,25).

thumbnail
Table 5. The quantitative metrics of the En-DeepMedic with different local patch size.

https://doi.org/10.1371/journal.pone.0185844.t005

Based on the results above, we choose the local patch size (25,25,25), the corresponding global patch size is (57,57,57), and the down-sampled global patch size is (19,19,19).

Another advantage of the En-DeepMedic method is computational efficiency. Though it takes time to train a CNN model, segmentation itself is fast. In our study, the entire workflow was implemented using a laptop with an NVIDIA Quadro M2000M graphic card and an Intel Xeon E3-1505 processor. The training of each fold of clinical data takes about two days, but auto-segmenting tumors on a typical image set with the size of 256 × 256 × 176 takes approximately two minutes. Moreover, the segmentation time is independent of the number of tumors, since it classifies voxels in an image all at once.

The En-DeepMedic method is also image modality-independent. It can be applied to both multi-modality and mono-modality images, as long as the training data used are the same modality as the images to be segmented. Since our application focuses on SRS of brain metastases, we thoroughly evaluated its performance on mono-modality T1c image data. However, the training and testing results on multi-modality BRATS data already show its promise for expanding to multi-modality image segmentation. Another important feature of our En-DeepMedic method is that it is parameter-less. No parameters need tuning after the model is established, so the segmentation is more robust and requires less human intervention.

Conclusion

In this work, we developed a deep convolutional neural network machine learning method for automatic segmentation of small brain metastases. We validated the method comprehensively on both BRATS and clinical data sets, and it demonstrated superior performance when compared with reported algorithms in the BRATS challenge. The developed auto-segmentation strategy achieved high accuracy and efficiency and shows promise as a tool for accurate and efficient SRS treatment planning.

Acknowledgments

We thank Dr. Jonathan Feinberg for editing the manuscript.

References

  1. 1. Tsao MN, Rades D, Wirth A, Lo SS, Danielson BL, et al. (2012) Radiotherapeutic and surgical management for newly diagnosed brain metastasis(es): An American Society for Radiation Oncology evidence-based guideline. Pract Radiat Oncol 2: 210–225. pmid:25925626
  2. 2. National Guideline C (2011) Brain metastases.
  3. 3. Gordillo N, Montseny E, Sobrevilla P (2013) State of the art survey on MRI brain tumor segmentation. Magnetic Resonance Imaging 31: 1426–1438. pmid:23790354
  4. 4. Stefan B, Roland W, Lutz PN, Mauricio R (2013) A survey of MRI-based medical image analysis for brain tumor studies. Physics in Medicine and Biology 58: R97. pmid:23743802
  5. 5. Schmidt P, Gaser C, Arsic M, Buck D, Förschler A, et al. (2012) An automated tool for detection of FLAIR-hyperintense white-matter lesions in Multiple Sclerosis. NeuroImage 59: 3774–3783. pmid:22119648
  6. 6. LaMontagne P, Milchencko M, Vélez M, Abraham C, Marcus D, et al. (2014) NI-50SEGMENTATION OF METASTATIC LESIONS IN LARGE-SCALE REGISTRIES: COMPARISON OF EXPERT MANUAL SEGMENTATION VS. SEMI-AUTOMATED METHODS. Neuro-Oncology 16: v149–v149.
  7. 7. Yan L, Strahinja S, Brian H, Zabi W, Weiguo L, et al. (2016) Automatic metastatic brain tumor segmentation for stereotactic radiosurgery applications. Physics in Medicine and Biology 61: 8440. pmid:27845915
  8. 8. Ilunga-Mbuyamba E, Cruz-Duarte JM, Avina-Cervantes JG, Correa-Cely CR, Lindner D, et al. (2016) Active contours driven by Cuckoo Search strategy for brain tumour images segmentation. Expert Systems with Applications 56: 59–68.
  9. 9. Subudhi BN, Thangaraj V, Sankaralingam E, Ghosh A (2016) Tumor or abnormality identification from magnetic resonance images using statistical region fusion based segmentation. Magnetic Resonance Imaging 34: 1292–1304. pmid:27477599
  10. 10. Dera D, Bouaynaya N, Fathallah-Shaykh HM (2016) Automated Robust Image Segmentation: Level Set Method Using Nonnegative Matrix Factorization with Application to Brain MRI. Bulletin of Mathematical Biology 78: 1450–1476. pmid:27417984
  11. 11. Bleau A, Leon LJ (2000) Watershed-based segmentation and region merging. Computer Vision and Image Understanding 77: 317–370.
  12. 12. Sachdeva J, Kumar V, Gupta I, Khandelwal N, Ahuja CK (2016) A package-SFERCB-"Segmentation, feature extraction, reduction and classification analysis by both SVM and ANN for brain tumors". Applied Soft Computing 47: 151–167.
  13. 13. Ozkan M, Dawant BM, Maciunas RJ (1993) NEURAL-NETWORK-BASED SEGMENTATION OF MULTIMODAL MEDICAL IMAGES—A COMPARATIVE AND PROSPECTIVE-STUDY. Ieee Transactions on Medical Imaging 12: 534–544. pmid:18218446
  14. 14. Bergner N, Romeike BFM, Reichart R, Kalff R, Krafft C, et al. (2013) Tumor margin identification and prediction of the primary tumor from brain metastases using FTIR imaging and support vector machines. Analyst 138: 3983–3990. pmid:23563220
  15. 15. Glotsos D, Tohka J, Ravazoula P, Cavouras D, Nikiforidis G (2005) Automated diagnosis of brain tumours astrocytomas using Probabilistic Neural Network clustering and Support Vector Machines. International Journal of Neural Systems 15: 1–11. pmid:15912578
  16. 16. Zhang N, Ruan S, Lebonvallet S, Liao QM, Zhu YM (2011) Kernel feature selection to fuse multi-spectral MRI images for brain tumor segmentation. Computer Vision and Image Understanding 115: 256–269.
  17. 17. Bauer S, Nolte LP, Reyes M (2011) Fully Automatic Segmentation of Brain Tumor Images Using Support Vector Machine Classification in Combination with Hierarchical Conditional Random Field Regularization. In: Fichtinger G, Martel A, Peters T, editors. Medical Image Computing and Computer-Assisted Intervention, Miccai 2011, Pt Iii. Berlin: Springer-Verlag Berlin. pp. 354–361.
  18. 18. CHEVALIER M, Thome N, Cord M, Fournier J, Henaff G, et al. (2016) LOW RESOLUTION CONVOLUTIONAL NEURAL NETWORK FOR AUTOMATIC TARGET RECOGNITION. 7th International Symposium on Optronics in Defence and Security. Paris, France.
  19. 19. Christ PF, Ettlinger F, Grün F, Elshaera MEA, Lipkova J, et al. (2017) Automatic Liver and Tumor Segmentation of CT and MRI Volumes using Cascaded Fully Convolutional Neural Networks. ArXiv e-prints.
  20. 20. Iglesias JE, Liu CY, Thompson PM, Tu Z (2011) Robust brain extraction across datasets and comparison with publicly available methods. IEEE Trans Med Imaging 30: 1617–1634. pmid:21880566
  21. 21. Wadell H (1935) Volume, Shape, and Roundness of Quartz Particles. The Journal of Geology 43: 250–280.
  22. 22. Kamnitsas K, Ledig C, Newcombe VFJ, Simpson JP, Kane AD, et al. (2016) Efficient Multi-Scale 3D CNN with Fully Connected CRF for Accurate Brain Lesion Segmentation. ArXiv e-prints.
  23. 23. He KM, Zhang XY, Ren SQ, Sun J, Ieee (2015) Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification. 2015 Ieee International Conference on Computer Vision. New York: Ieee. pp. 1026–1034.
  24. 24. Gurney K (1997) An Introduction to Neural Networks: Taylor \\& Francis, Inc. 288 p.
  25. 25. LeCun Y, Bottou L, Orr GB, Muller KR (1998) Efficient backprop. Neural Networks: Tricks of the Trade 1524: 9–50.
  26. 26. Salakhutdinov GHNSAKISR (2012) Improving neural networks by preventing co-adaptation of feature detectors. Computer Science 3: 212–223.
  27. 27. Menze BH, Jakab A, Bauer S, Kalpathy-Cramer J, Farahani K, et al. (2015) The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS). IEEE Trans Med Imaging 34: 1993–2024. pmid:25494501
  28. 28. Wardak Z, Augustyn A, Zhu H, Mickey BE, Whitworth LA, et al. (2016) Pre-treatment factors associated with detecting additional brain metastases at stereotactic radiosurgery. J Neurooncol 128: 251–257. pmid:26966096
  29. 29. Mikael Agn OP, Ian Law, Per Munck af Rosenschold, Van Leemput Koen (2015) Brain Tumor Segmentation by a Generative Model with a Prior on Tumor Shape. In: Menze BH MR, Farahani K, Kalpathy-Cramer J, Kwon D, editor. MICCAI-BRATS 2015. Munich. pp. 1–4.
  30. 30. Spyridon Bakas KZ, Sotiras Aristeidis, Rathore Saima, Akbari Hamed,Gaonkar Bilwaj, Rozycki Martin, Pati Sarthak, Davatzikos Christos (2015) Segmentation of Gliomas in Multimodal Magnetic Resonance Imaging Volumes Based on a Hybrid Generative-Discriminative Framework. In: Menze BH MR, Farahani K, Kalpathy-Cramer J, Kwon D, editor. MICCAI-BRATS 2015. Munich. pp. 5–12.
  31. 31. Assaf Hoogi AL, Bharadwaj Vivek, Rubin Daniel L. (2015) Multimodal Brain Tumor Segmentation (BRATS) using Sparse Coding and 2-layer Neural Network. In: Menze BH MR, Farahani K, Kalpathy-Cramer J, Kwon D, editor. MICCAI-BRATS 2015. Munich. pp. 34–37.
  32. 32. Eric Malmi SP, Peyrat Jean-Marc, Abinahed Julien, Chawla Sanjay (2015) CaBS: A Cascaded Brain Tumor Segmentation Approach. In: Menze BH MR, Farahani K, Kalpathy-Cramer J, Kwon D, editor. MICCAI-BRATS 2015. Munich. pp. 42–47.
  33. 33. Raphael Meier VK, Habegger Simon, Wiest Roland, Mauricio Reyes M. (2015) Parameter Learning for CRF-based Tissue Segmentation of Brain Tumors. In: Menze BH MR, Farahani K, Kalpathy-Cramer J, Kwon D, editor. MICCAI-BRATS 2015. Munich. pp. 48–51.
  34. 34. Sergio Pereira AP, Alves Vıctor, Silva Carlos A. (2015) Deep Convolutional Neural Networks for the Segmentation of Gliomas in Multi-Sequence MRI. In: Menze BH MR, Farahani K, Kalpathy-Cramer J, Kwon D, editor. MICCAI-BRATS 2015. Munich. pp. 52–55.
  35. 35. Kiran Vaidhya RS, Thirunavukkarasu Subramaniam, Alex Varghese, Krishnamurthi Ganapathy (2015) Multi-Modal Brain Tumor Segmentation Using Stacked Denoising Autoencoders. In: Menze BH MR, Farahani K, Kalpathy-Cramer J, Kwon D, editor. MICCAI-BRATS 2015. Munich. pp. 60–64.
  36. 36. Yan Liu SS, Hrycushko Brian, Wardak Zabi, Lu Weiguo, Yan Yulong, Jiang Steve B, Timmerman Robert, Abdulrahman Ramzi, Nedzi Lucien, Gu Xuejun (2016) Automatic metastatic brain tumor segmentation for stereotactic radiosurgery applications. Physics in Medicine and Biology 61: 8440. pmid:27845915