SS symmetry Article
A Novel Tool for Supervised Segmentation Using 3D Slicer Daniel Chalupa *
and Jan Mikulka
Department of Theoretical and Experimental Electrical Engineering, Brno University of Technology, Technická 3082/12, 616 00 Brno, Czech Republic;
[email protected] * Correspondence:
[email protected] Received: 15 October 2018; Accepted: 7 November 2018; Published: 12 November 2018
Abstract: The rather impressive extension library of medical image-processing platform 3D Slicer lacks a wide range of machine-learning toolboxes. The authors have developed such a toolbox that incorporates commonly used machine-learning libraries. The extension uses a simple graphical user interface that allows the user to preprocess data, train a classifier, and use that classifier in common medical image-classification tasks, such as tumor staging or various anatomical segmentations without a deeper knowledge of the inner workings of the classifiers. A series of experiments were carried out to showcase the capabilities of the extension and quantify the symmetry between the physical characteristics of pathological tissues and the parameters of a classifying model. These experiments also include an analysis of the impact of training vector size and feature selection on the sensitivity and specificity of all included classifiers. The results indicate that training vector size can be minimized for all classifiers. Using the data from the Brain Tumor Segmentation Challenge, Random Forest appears to have the widest range of parameters that produce sufficiently accurate segmentations, while optimal Support Vector Machines’ training parameters are concentrated in a narrow feature space. Keywords: 3D slicer; classification; extension; random forest; segmentation; sensitivity analysis; support vector machine; tumor
1. Introduction 3D Slicer [1] is a free open-source platform for medical image visualization and processing. Its main functionality comes from the Extension Library, which consists of various modules that allow specific analyses of the input data, such as filtering, artefact suppression, or surface reconstruction. There is a lack of machine-learning extensions except for the open-source DeepInfer [2]. This deep-learning deployment kit uses 3D convolutional neural networks to detect and localize the target tissue. The development team demonstrated the use of this kit on the prostate segmentation problem for image-guided therapy. Researchers and practitioners are able to select a publicly available task-oriented network through the module without the need to design or train it. To enable the use of other machine-learning techniques, we developed the Supervised Segmentation Toolbox as an extensible machine-learning platform for 3D Slicer. Currently, Support Vector Machine (SVM) and Random Forest (RF) classifiers are included. These classifiers are well-researched and often used in image-processing tasks, as demonstrated in References [3–8]. SVMs [9] train by maximizing the distance between marginal samples (also referred to as support vectors) and a discriminative hyperplane by maximizing f in the equation: f ( α1 · · · α n ) = Symmetry 2018, 10, 627; doi:10.3390/sym10110627
1
→ →
∑ αi − 2 ∑ ∑ αi α j yi y j xi · x j , i
(1)
j
www.mdpi.com/journal/symmetry
Symmetry 2016, 2018, xx, 10, x627 Symmetry
22ofof99
where yy is defined defined as as +1 +1 for for aa class class A sample and –1 for a class B sample, α is a Lagrangian multiplier, multiplier, where → and x is the feature vector of the individual sample. On real data, this is often too strict because and ~x is the feature vector of the individual sample. On real data, this is often too strict because of noisy noisy samples samples that that might might cross cross their their class class boundary. boundary. This This is is solved solved by by using using aa combination combination of of of techniques known known as as the the kernel kernel trick trick and and soft soft margining. margining. The The kernel kernel trick trick uses uses aa kernel kernel function function techniques to remap remap the the original original feature feature space space to to aa higher-dimensional higher-dimensional one one by by replacing replacing the the dot dot product product in in φφ to Equation (1) (1)with withφφ( x( xi )i )· Thisallows allowslinear linear separation data as required by SVM. the SVM. Equation · φ( xxjj).. This separation of of thethe data as required by the An An example of the kernel function is the radial basis function usedininthis thisstudy. study.Incorporating Incorporating aa soft soft example of the kernel function is the radial basis function used margin allows allows some some samples samples to to cross cross their their class class boundary. boundary. Examples Examples of of such such soft-margin soft-margin SVMs SVMs are are the the margin C-SVM [10] [10] and and N-SVM N-SVM [11]. [11]. The The C C parameter of the C-SVM modifies the influence influence of of each each support support C-SVM vector on on the the final final discriminatory discriminatory hyperplane. hyperplane. The The larger larger the the C, C, the closer the soft-margin soft-margin C-SVM C-SVM is is to to vector hard-margin SVM. SVM. The The N parameter of the N-SVM defines a minimum number of support vectors vectors aa hard-margin and, consequently, consequently, an an upper upper bound bound of of the the guaranteed guaranteed maximum maximum percentage percentage of of misclassifications. misclassifications. and, Further modifications modifications to to the the SVM SVM can can also also be be done, done, such such as as using using fuzzy-data fuzzy-data points points for for the the training training Further dataset, as as demonstrated demonstrated in in References References [5,6]. [5,6]. dataset, RF uses uses aa voting voting system system on on the the results results of of the the individual individual decision decision trees. trees. Each Each decision decision tree tree is is RF created on on the the basis basis of of aa bootstrapped bootstrapped sample sample of of the the training training data data [12]. [12]. created 2. Materials Materials and and Methods Methods 2. The data data used used in in this this study study describe describe Low Low Grade Grade Glioma Glioma (LGG) (LGG) obtained obtained by by MRI. MRI. The The dataset dataset The consists of of four four images images of of one one patient patient (LG_0001, (LG_0001, Figure Figure 1): 1): T1T1- and and T2-weighted, T2-weighted, contrast-enhanced contrast-enhanced consists T1C, and and Fluid Fluid Attenuated Attenuated Inversion Inversion Recovery Recovery (FL). (FL). These These data data are are part part of of the the training training set set featured featured in in T1C, the Brain Tumor Segmentation (BraTS) 2012 Challenge. the Brain Tumor Segmentation (BraTS) 2012 Challenge.
Figure Figure 1. 1. Slice Slice 88 88 of of the the LG_0001 LG_0001 data data from from the the 2012 2012 Brain Brain Tumor Tumor Segmentation Segmentation (BraTS) (BraTS) Challenge. Challenge. The dataset consists of a (a) T1-weighted image with labels as overlay; (b) postcontrast T1-weighted The dataset consists of a (a) T1-weighted image with labels as overlay; (b) postcontrast T1-weighted image; image; (c) (c) T2-weighted T2-weighted image; image; and and (d) (d) Fluid Fluid Attenuated Attenuated Inversion Inversion Recovery Recovery (FL) (FL) image. image.
The The free free and and open-source open-source Supervised Supervised Segmentation Segmentation Toolbox Toolbox extension extension [24] [13] of of the the 3D 3D Slicer Slicer was was used used throughout throughout this this study. study. The The extension extension allows allows the the user user to to train train aa range range of of classifiers classifiers using using labeled labeled data. data. The The user user is is also also able able to to perform perform aa grid grid search search to to select select the the optimal optimal parameters parameters for for the the classifier. classifier. To achieve this, the extension uses either an already available function or a cross-validation algorithm To achieve this, the extension uses either an already available function or a cross-validation algorithm developed developed by by the the author author of of the the extension, extension, depending depending on on the the classifier classifier library library used. used. Currently, Currently, N-SVM N-SVM and and C-SVM C-SVM from from the the dlib dlib library library [13] [14] and and C-SVM C-SVM and and Random Random Forest Forest from from Shark-ml Shark-ml library library [14] [15] are are
Symmetry 2016, xx, x
3 of 9
Symmetry 2018, 10, 627
3 of 9
Symmetry 2016, xx, x 3 of 9 incorporated. The extension takes care of the parallelizable parts of the training and classification subtasks, thus significantly reducing computation times. A preprocessing algorithm selection is also a incorporated. The takes care of parallelizable parts the classification part of the extension. This allows artefact or feature extraction. Theand extension workflow incorporated. The extension extension takesfor care of the thecorrection parallelizable parts of of the training training and classification subtasks, thus significantly reducing computation times. A preprocessing algorithm selection is alsoa issubtasks, depictedthus in Figure 2. significantly reducing computation times. A preprocessing algorithm selection is also apart partofofthe theextension. extension.This Thisallows allowsfor forartefact artefactcorrection correctionor orfeature feature extraction. extraction. The The extension extension workflow workflow is depicted in Figure 2. is depicted in Figure 2.
Training Samples Training Samples
Preprocessing
Preprocessing
Training
Segmentation
Training
Segmentation
Figure 2. Supervised segmentation extension workflow. Figure 2. 2. Supervised Supervised segmentation segmentation extension extension workflow. workflow. 3. Results and Discussion Figure
3. Results Results andofDiscussion Discussion A series tests were performed in order to provide a sense of the speed and accuracy of the 3. and provided classifiers. Sensitivity and specificity metrics were usedoftothe evaluate the results. Aofclassifier A series series of of tests tests were were performed performed in in order order to to provide provide aa sense sense of speed and and accuracy accuracy of the A the speed the that had aclassifiers. larger sumSensitivity of specificity and sensitivity (orwere theirused respective means, provided and specificity specificity metrics to evaluate evaluate the when results.cross-validation A classifier classifier provided classifiers. Sensitivity and metrics were used to the results. A was considered a better classifier. During firstrespective test run, each type of classifier was trained thatused) had aawas larger sum of of specificity specificity and sensitivity sensitivity (orthe their means, when cross-validation that had larger sum and (or their respective means, when cross-validation and using the single patient image set. the Optimal training parameters of the classifiers wasevaluated used) was was considered considered better classifier. During the first test test run, each each type of of classifier classifier was trained trainedwere was used) aa better classifier. During first run, type was obtained using a grid-search approach. The results are presented in Figures 3–5. The γ parameter and evaluated evaluated using using the the single single patient patient image image set. set. Optimal Optimal training training parameters parameters of of the the classifiers classifiers were and were isobtained common in both SVM classifiers and influences the variance of the radial basis kernel. A large γ obtained using a grid-search approach. The results are presented in Figures 3–5. The γ parameter using a grid-search approach. The results are presented in Figures 3–5. The γ parameter means that more data points will look thusthe preventing overfitting. Using the aforementioned is common common in both both SVM classifiers andsimilar, influences the variance of of the radial radial basis basis kernel. kernel. A large large γ γ is in SVM classifiers and influences variance the A dataset, the results indicate a relative insensitivity of the classification accuracy on this parameter. means that more data points will look similar, thus preventing overfitting. Using the aforementioned means that more data points will look similar, thus preventing overfitting. Using the aforementioned dataset, the results results indicate relative insensitivity of the the classification accuracyThe on this this parameter. parameter. For the given dataset, C values of the C-SVM larger than 1 seem optimal. best results of the dataset, the indicate aa relative insensitivity of classification accuracy on For the the classifier given dataset, dataset, C values values of the the C-SVM10% larger than 11combined seem optimal. optimal. The best results resultsradial of the thebasis N-SVM are obtained with N around or lower with aThe high-variance For given C of C-SVM larger than seem best of N-SVM classifier are obtained with N around 10% or lower combined with a high-variance radial basis function. Optimalare RFobtained trainingwith parameters a small node size with of under 5, number radial of trees higher N-SVM classifier N aroundwere: 10% or lower combined a high-variance basis function. Optimal RF training training parameters were: aa small small node size size of of under under 5, 5, number number of of trees trees higher higher function. Optimal RF parameters were: node than 800, no out-of-bag samples, and no random attributes. than 800, 800, no no out-of-bag out-of-bag samples, samples, and and no no random random attributes. attributes. than 1
1
1
1
0.9
0.02 0.02
0.9
0.04 0.04
0.8
0.9
0.02 0.02
0.9
0.04 0.04
0.8
0.8
0.8
0.7 0.7
0.06 0.06
0.7 0.7
0.06 0.06
0.6
0.6 0.6
N
0.08 0.08
0.5
N
0.5
N
N
0.6
0.08
0.4
0.1 0.1
0.4
0.5
0.08
0.1
0.5
0.4
0.1
0.4
0.3
0.3
0.12 0.12
0.3
0.3
0.12
0.12 0.2
0.2
0.14 0.14
0.10.1
-3-3
-2 -2
-1 -1
00
11
log 10 .. log 10
(a) (a)
22
33
0.2
0.140.14
0 0
-3
-3
-2
-2 -1
-1 0 0 1 log 10 log . 10 .
1 2
2 3
3
0.2
0.1
0.1
0
0
(b) (b)
Figure 3. N-Support N-Support Vector (a)(a) sensitivity andand (b) (b) specificity using different Figure VectorMachine Machine(SVM)-classifier (SVM)-classifier sensitivity specificity using different parameter pairs. parameter pairs.
Symmetry 2016, xx, x
4 of 9
Symmetry 2016, xx, x
4 of 9
Symmetry 2018, 10, 627
4 of 9
-3.5
-3.5
1
-3-3.5
1
-3-3.5
0.9 1
0.9 1
-2.5 -3
0.8 0.9
-2-2.5
0.7 0.8
-2-2.5
0.7 0.8
-1.5 -2
0.6 0.7
-1.5 -2
0.6 0.7
-1-1.5
0.5 0.6
-1-1.5
0.5 0.6
0.4 0.5
0-0.5
log 10 C
log 10 C
-0.5 -1
log 10 C
0.8 0.9
log 10 C
-2.5 -3
-0.5 -1
0.4 0.5
0.3 0.4
0-0.5
0.3 0.4
0.5 0
0.2 0.3
0.5 0
0.2 0.3
1 0.5
0.1 0.2
1 0.5
0.1 0.2
1.5 1
0
-2
-1.5
-1
-0.5
0
0.5
1
1.5
2
log 10 .
1.5 -2
-1.5
-1
-0.5
1.5 1
0.1
0.5
1
1.5
-1.5
-1
-0.5
0
0.5
1
1.5
2
log 10 .
1.5
0
0
0
-2
2
-2
-1.5
-1
-0.5
log(a) . 10
0.1
0
0
0.5
1
1.5
2
log(b) . 10
Figure 4. C-SVM-classifier (a) sensitivity and (b) specificity using different(b) parameter pairs. (a) Figure 4. 4. C-SVM-classifier (a)(a) sensitivity and (b)(b) specificity using different parameter pairs. Figure C-SVM-classifier sensitivity and specificity using different parameter pairs. 1
1
Sensitivity Specificity
Sensitivity Specificity
0.85
0.8
0.98 0.97
0.97 0.96
0.94 0.96 0.92 0.94 0.9 0.92
0.88 0.9
0.86 0.88
0.95 0.94 800 1000 1200 1400 1600 1800 2000
0.75 0
5
10
15
20
0.8 0 0.82
0.2
0.4
Number of Trees
Node size 0.75
0.94 800
0.94 0.96 0.92 0.94 0.9 0.92
0.88
0.9
0.84 0.86
0.82 0.84
0.8
Sensitivity Specificity
0.86 0.88
0.84 0.86
0.96 0.95
Sensitivity Specificity
1
0.96 0.98
Sensitivity and Specificity
0.9
0.85
0.99 0.98
Sensitivity Specificity
0.96 0.98
Sensitivity and Specificity
Sensitivity and Specificity
0.9
Sensitivity Specificity
Sensitivity and Specificity
Sensitivity and Specificity
0.95
0.98
1
0.99
Sensitivity Specificity
0.95
Sensitivity and Specificity
0.98
1
Sensitivity and Specificity
1
1 Sensitivity Specificity
Sensitivity and Specificity
1
0.6
0.82 0.84 0
0.8
0.5
1
1.5
2
Number of Random Attributes
Oob 0.8
0.82
Figure 5. Random Forest classifier sensitivity parameters. Number Left oftoRandom right: Number of Trees and specificity using different Node size Attributes Oob Different node size, number of trees, OOB and number of random attributes. Figure 5. Random Forest classifier sensitivity and specificity using different parameters. Left to right: Different node size, of trees, OOB aand numbernumber of random attributes. The second test runnumber consisted of using different of slices around the center of the tumor Figure 5. Random Forest classifier sensitivity and specificity using different parameters. Left to right: to reveal the impact of the size of the training set on the specificity and sensitivity of all classifiers Different node of trees, OOB and number ofnumber random of attributes. The second testsize, runnumber consisted of using a different slices around the center of the 0
5
10
15
20
1000 1200 1400 1600 1800 2000
0
0.2
0.4
0.6
0.8
0
0.5
1
1.5
2
(Figures 6a and 7). The results indicate that reducing the number of unique training samples has tumor to reveal the impact of the size of the training set on the specificity and sensitivity of all a negligible effect on subsequent accuracy. RF shows slightly better classification The second testthe run consisted classification of using a different number of slices around the center of the classifiers (Figures 6a–7). The results indicate that reducing the number of unique training samples accuracy improvement when using a larger training vector. Using a reduced training dataset influences tumor to reveal the impact of the size of the training set on the specificity and sensitivity of all has a negligible effect on the subsequent classification accuracy. RF shows slightly better classification training process length andThe might result in a simpler classifier, which is easier to interpret and has classifiers (Figures 6a–7). results indicate that reducing the number of unique training samples accuracy improvement when using a larger training vector. Using a reduced training dataset influences shorter classification computation times. classification The classification timeRF is shows a limiting factor of using these has a negligible effect on the subsequent accuracy. slightly better classification training process length and might result in a simpler classifier, which is easier to interpret and has methods in real-time applications. accuracy improvement when using a larger training vector. Using a reduced training dataset influences shorter classification computation times. The classification time is a limiting factor of using these training process length and might result in a simpler classifier, which is easier to interpret and has methods in real-time applications. shorter classification computation times. The classification time is a limiting factor of using these methods in real-time applications.
Symmetry 2016, xx, x
5 of 9
Symmetry 2016, xx, x
5 of 9
Symmetry 2018, 10, 627
5 of 9
#10 5
1 Unique Training Samples Sensitivity Specificity
1.8 #10 5 2
1 Unique Training Samples Sensitivity Specificity
1.8 #10 5 2
Unique Training Samples Sensitivity Specificity
0.98 1
1.6 1.8
1.4 1.6
0.94 0.96
1.4 1.6
0.94 0.96
1.2 1.4
0.92 0.94
1.2 1.4
0.92 0.94
1 1.2
0.9 0.92
1 1.2
0.9 0.92
0.8 1
0.88 0.9
0.8 1
0.88 0.9
0.6 0.8
0.86 0.88
0.6 0.8
0.86 0.88
0.4 0.6
0.84 0.86
0.4 0.6
0.84 0.86
0.2 0.4
0.82 0.84
0.2 0.4
0.82 0.84
0 0.2 0
10
20
30
40
50
60
70
80
and specificity Sensitivity Sensitivity and specificity
0.96 0.98
Unique Training Training Samples UniqueSamples
1.6 1.8
Unique Training Training Samples UniqueSamples
0.98 1
Unique Training Samples Sensitivity Specificity
#10 5
2
0.8 1000.82
90
0 0.2 0
10
20
30
40
50
Slice Count 10
20
30
40
60
70
80
90
0.8 1000.82
60
70
80
90
0.8 100
Slice Count
0 0
0.96 0.98
and specificity Sensitivity Sensitivity and specificity
2
50
60
(a)
70
80
0.8 100
90
0 0
10
20
30
40
50
(b)
Slice Count
Slice Count
Figure 6. (a) N-SVM(a) and (b) C-SVM sensitivity and specificity using different(b) training vector sizes. Figure 6. 6. (a) (a) N-SVM N-SVM and and (b) (b) C-SVM C-SVM sensitivity sensitivity and and specificity specificity using using different different training training vector vector sizes. sizes. Figure #10 5
1 Unique Training Samples Sensitivity Specificity
1.8 #10 5 2
Unique Training Samples Sensitivity Specificity
Unique Training Training Samples UniqueSamples
1.6 1.8
0.98 1 0.96 0.98
1.4 1.6
0.94 0.96
1.2 1.4
0.92 0.94
1 1.2
0.9 0.92
0.8 1
0.88 0.9
0.6 0.8
0.86 0.88
0.4 0.6
0.84 0.86
0.2 0.4
0.82 0.84
0 0.2 0
10
20
30
40
50
60
70
80
and specificity Sensitivity Sensitivity and specificity
2
0.8 1000.82
90
Slice Count 0 0.8 Figure 7. RF-classifier sensitivity different training vector sizes. 0 10 20 30 and 40 specificity 50 60 using 70 80 90 100 Slice Count Figure 7. RF-classifier sensitivity and specificity using different training vector sizes.
1
0.9
0.9
0.8
0.8
0.7
0.7 Sensitivity and specificity
1
0.6
0.5
0.4
0.6
0.5
0.4
0.3
0.3
0.2
0.2
0.1
Sensitivity Specificity
Sensitivity Specificity
0.1
(a)
1
0.9
0.8
FL
L
FL
2,
2,
,T
,T C
1C ,T
FL
(b)
Figure (b). Figure8. 8. Classifier Classifiersensitivity sensitivityusing usingdifferent differentinput inputimages. images. N-SVM N-SVM(a), (a),C-SVM C-SVM(b).
T1
T1
2,
1C ,F
,T T1
,T
L
2
,F T2
1C ,T ,T
T1
Image Types
T1
L ,F C T1
L
2
,F
,T C
T1
T1
2
1C
,T
,T
T1
T1
C
FL
T2
T1
T1
2
2 ,T C
1,
T1
C
1C ,T ,T
FL
FL
,T
T2
T1
1,
1,
,T
,T FL
2
2 1C ,T
,T
FL
1C
,T T1
,T FL
Image Types
FL
1
2
,T
,T C
FL
T1
2
1C
,T
,T
T1
T1
C T1
FL
0 T2
0 T1
Sensitivity and specificity
The effect of different image types on classifier accuracy was examined in the last test run (Figures and 9). 88 ofimage thesensitivity LG_0001 images were used a source of training Figure 7. RF-classifier and specificity usingasdifferent vector sizes. The8a effect of Slices different types on classifier accuracy was training examined insamples. the lastSufficient test run sensitivity and obtained images by onlywere usingused T1- and images. Furthermore, (Figures 8a–9). Slices 88 ofwere the LG_0001 as a T2-weighted source of training samples. Sufficient Symmetry 2016, xx, specificity x 6 ofall 9 The effect of different image types on classifier T1-weighted accuracy was examined the last achieved test run classifiers benefited from the addition of abypostcontrast image. The RFinclassifier sensitivity and specificity were obtained only using T1- and T2-weighted images. Furthermore, all (Figures 8a–9). Slices 88 of the LG_0001 images were used a source of training samples. Sufficient best overall results with of FL,of and postcontrast T1-as and T2-weighted images. classifiers benefited fromthe theuse addition a postcontrast T1-weighted image. The RF classifier achieved sensitivity and specificity were obtained by only using T1- and T2-weighted images. Furthermore, all best overall results with the use of FL, and postcontrast T1- and T2-weighted images. classifiers benefited from the addition of a postcontrast T1-weighted image. The RF classifier achieved best overall results with the use of FL, and postcontrast T1- and T2-weighted images.
,T 1C
T1 ,T
T1 C
,T
2,
2,
FL
L
FL
FL
,F
2,
1C
T1 ,T
L
2 ,T 1C
T1 ,T
L
,F
,F
2
L
(a)
T1 ,T
C
T2
T1
,F
,T C
T1
,T T1
T1
T1
1C
2
FL
,T
T2
T1 C
,T
T1 C
T1
2
2
C
,T 1C
1,
Image Types
FL ,T
T2
T1
FL ,T
FL ,T
1,
,T 1C
1,
2
2
Image Types
FL ,T
T1 ,T
1C
,T
,T FL
FL
1
2
,T
,T C
FL
T1
2
1C ,T T1
FL
,T T1
T2
T1 C
0 T1
0
(b)
Figure Symmetry 2018, 10, 627 8. Classifier sensitivity using different input images. N-SVM (a), C-SVM(b).
6 of 9
1
0.9
0.8
Sensitivity and specificity
0.7
0.6
0.5
0.4
0.3
0.2 Sensitivity Specificity
0.1
2
2
C
1,
T1 C
,T
,T
T1
,T 1C
,T FL
FL
,T 1,
1, ,T FL
FL
2
T2
2
,T ,T 1C
1C
,T T1
,T FL
Image Types
FL
1
2
,T
,T C
FL
T1
T1
,T
1C
2 ,T T1
FL
C T1
T2
T1
0
Figure 9. RF-Classifier sensitivity using different input images.
The following standardized procedure was designed in order to compare classifier performance. Training samples were extracted from the whole volume of the unmodified T1C and T2 images. Then, sensitivity and specificity were obtained using fivefold cross-validation. The best performing parameters and results are reported in Table 1. Segmentations are shown in Figure 10. Classification results can be further improved by using preprocessed data instead of raw data, and by means of [16]. postprocessing to remove the outlying voxels and inlying holes as demonstrated in Reference [15]. Table 1. Classifier comparison and best-performing parameters. Parameters Parameters
Sensitivity Sensitivity
Specificity
Acc.
Prec. Prec.
DICE DICE
Jaccard Jaccard
C-SVM C-SVM
1.0, C C == 1.0 1.0 γγ == 1.0,
0.72 0.72
0.98
0.99 0.99
0.96 0.96
0.80 0.80
0.66 0.66
N-SVM N-SVM
10.0−−33,, N N = 0.1 γγ == 10.0
0.77 0.77
0.97
0.99 0.99
0.96 0.96
0.82 0.82
0.70 0.70
1.00 1.00
0.95
1.00 1.00
0.96 0.96
0.94 0.94
0.89 7 of 9 0.89
0 % OOB,
Symmetry 2016, xx,0xrandom attributes, RF RF
1200 trees, node size 2
(a)
(b)
(c)
Figure Figure 10. 10. (a) (a) RF, RF, (b) (b) C-SVM, C-SVM, and and (c) (c) N-SVM N-SVM classification classification results results of of the the slice slice 88 88 (white) (white) and and ground ground truth (red). truth (red).
Lastly, Lastly, the the performance performance of of the the RF RF classifier classifier trained trained on on all all tumor tumor cores cores of of the the 20 20 real real high-grade high-grade glioma glioma volumes volumes using using the the 3D 3D Slicer Slicer extension extension were were compared compared to to similar similar studies studies performed performed on on the the BraTS BraTS dataset. dataset. The The values values were were obtained obtained as as aa mean mean of of fivefold fivefold cross-validation. cross-validation. This This comparison comparison is is shown in Table 2. The other DICE values are from Reference [17]. shown in Table 2. The other DICE values are from Reference [16]. The to combine combine the theresults resultsofofdifferent differentclassifiers classifiers further expand usability of The means to to to further expand thethe usability of the the Supervised Segmentation Toolbox extension were added. Currently, logical AND andOR ORand anda Supervised Segmentation Toolbox extension were added. Currently, logical AND and amajority majorityvoting votingsystem systemare areimplemented. implemented. An An addition addition of of aa Multiple Classifier System (MCS) is is currently currently considered. considered. A A review review of of the the advantages advantages of of MCS MCS is provided by Wozniak et al. [18]. [17]. Termenon Termenon and and Graña Graña [19] [18] used used aa two-stage two-stage MCS MCS where where the the second second classifier classifier was was trained trained on on low-confidence low-confidence data data obtained obtained by by training training and and analysis analysis of of the the first first classifier. classifier. In In the the future, future, the the authors authors expect expect implementing implementing additional classifiers as well. Adding a Relevance Vector Machine (RVM), for example, might bring an improvement over SVM [19]. Table 2. RF classifier comparison with similar studies. The classifier was trained using all 20 of the real high-grade glioma volumes, and the DICE value is a mean of fivefold cross-validation.
Symmetry 2018, 10, 627
7 of 9
additional classifiers as well. Adding a Relevance Vector Machine (RVM), for example, might bring an improvement over SVM [20]. Table 2. RF classifier comparison with similar studies. The classifier was trained using all 20 of the real high-grade glioma volumes, and the DICE value is a mean of fivefold cross-validation. Paper
Approach
DICE
This paper Geremia [21] Kapás [16] Bauer [22] Zikic [23] Festa [24]
RF Spatial decision forests with intrinsic hierarchy RF Integrated hierarchical RF Context-sensitive features with a decision tree ensemble RF using neighborhood and local context features
0.43 0.32 0.58 0.48 0.47 0.50
4. Conclusions The Supervised Segmentation Toolbox extension was presented as an addition to the 3D Slicer extension library. This extension allows the user to train and use three types of classifiers, with more to be added in the future. The usability of the extension was demonstrated on a brain-tumor segmentation use case. The effects of the training parameters of all classifiers on the final sensitivity and specificity of the classification were considered to provide an insight into usable parameter selection for future studies. A low γ in combination with softer margin terms resulted in a better performing classifier commonly for both SVM classifiers. This might be largely due to a limited training sample, and a broader dataset should be analyzed in order to generalize the results. The RF classifier performed best using no added randomization, a relatively large tree count, and a small node size. The possibility of reducing training vector size in order to reduce model complexity and decrease classification time is verified. A 20-fold increase of the number of unique training samples resulted, at best, in a 2% increase of specificity. All combinations of input images are considered as a training input for all classifiers, and the significance of adding more types of images is discussed. A combination of T1C and T2 images performed sufficiently for all classifiers. The addition of the FL image brought a slight improvement in sensitivity. Lastly, best-performing parameter combinations were listed and the corresponding results were compared. The RF classifier had the largest sensitivity and worst specificity, and C-SVM performed oppositely. The significance of these two metrics largely depends on the type of task for which the classifiers are used. All sensitivity and specificity data were obtained directly using the 3D Slicer extension. Supplementary Materials: The authors publicly release the source code of the Supervised Segmentation Toolbox. The source code can be found on GitHub [13]. The BraTS 2012 dataset is available on the challenge’s website. Author Contributions: J.M. conceived and designed the experiments; D.C. performed the experiments, programmed the extension, analyzed the data, and wrote the paper. Funding: This work was supported by the Ministry of Health of the Czech Republic, grant no. NV18-08-00459. Conflicts of Interest: The authors declare no conflict of interest.
Abbreviations The following abbreviations are used in this manuscript: BraTS FL LGG MCS OOB RVM RF SVM
Brain-Tumor Segmentation Fluid-Attenuated Inversion Recovery Low-Grade Glioma Multiple Classifier System Out Of Box Relevance Vector Machine Random Forest Support Vector Machine
Symmetry 2018, 10, 627
8 of 9
References 1.
2.
3.
4.
5.
6.
7. 8. 9. 10. 11. 12. 13.
14. 15. 16.
17.
18. 19. 20. 21. 22.
Fedorov, A.; Beichel, R.; Kalpathy-Cramer, J.; Finet, J.; Fillion-Robin, J.C.; Pujol, S.; Bauer, C.; Jennings, D.; Fennessy, F.; Sonka, M.; et al. 3D Slicer as an image computing platform for the Quantitative Imaging Network. Magn. Reson. Imaging 2012, 30, 1323–1341. [CrossRef] [PubMed] Mehrtash, A.; Sedghi, A.; Ghafoorian, M.; Taghipour, M.; Tempany, C.M.; Wells, W.M.; Kapur, T.; Mousavi, P.; Abolmaesumi, P.; Fedorov, A. Classification of Clinical Significance of MRI Prostate Findings Using 3D Convolutional Neural Networks. Proc. SPIE Int. Soc. Optical Eng. 2017, 10134. Zhang, Y.; Dong, Z.; Wang, S.; Ji, G.; Yang, J. Preclinical diagnosis of magnetic resonance (MR) brain images via discrete wavelet packet transform with Tsallis entropy and generalized eigenvalue proximal support vector machine (GEPSVM). Entropy 2015, 17, 1795–1813. [CrossRef] Zhang, Y.; Wang, S.; Phillips, P.; Dong, Z.; Ji, G.; Yang, J. Detection of Alzheimer’s disease and mild cognitive impairment based on structural volumetric MR images using 3D-DWT and WTA-KSVM trained by PSOTVAC. Biomed. Signal Process. Control 2015, 21, 58–73. [CrossRef] Wang, S.; Li, Y.; Shao, Y.; Cattani, C.; Zhang, Y.; Du, S. Detection of dendritic spines using wavelet packet entropy and fuzzy support vector machine. CNS Neurol. Dis.-Drug Targets 2017, 16, 116–121. [CrossRef] [PubMed] Zhang, Y.; Yang, Z.; Lu, H.; Zhou, X.; Phillips, P.; Liu, Q.; Wang, S. Facial emotion recognition based on biorthogonal wavelet entropy, fuzzy support vector machine, and stratified cross validation. IEEE Access 2016, 4, 8375–8385. [CrossRef] Zheng, Q.; Wu, Y.; Fan, Y. Integrating semi-supervised and supervised learning methods for label fusion in multi-atlas based image segmentation. Front. Neuroinform. 2018, 12, 69. [CrossRef] [PubMed] Amiri, S.; Mahjoub, M.A.; Rekik, I. Tree-based Ensemble Classifier Learning for Automatic Brain Glioma Segmentation. Neurocomputing 2018, 313, 135–142. [CrossRef] Cortes, C.; Vapnik, V. Support-vector networks. Mach. Learn. 1995, 20, 273–297. [CrossRef] Chang, C.C.; Lin, C.J. LIBSVM: A library for support vector machines. ACM Trans. Intell. Syst. Technol. 2011, 2, 1–27. [CrossRef] Schölkopf, B.; Smola, A.J.; Williamson, R.C.; Bartlett, P.L. New support vector algorithms. Neural Comput. 2000, 12, 1207–1245. [CrossRef] [PubMed] Ho, T.K. Random decision forests. In Proceedings of the 3rd International Conference on Document Analysis and Recognition, Montreal, QC, Canada, 14–16 August 1995; pp. 278–282. Chalupa, D. Supervised Segmentation Toolbox for 3D Slicer. Source Code Available Under the GNU General Public License v3.0. Available online: https://github.com/chalupaDaniel/slicerSupervisedSegmentation (accessed on 14 October 2018). King, D.E. Dlib-ml: A machine learning toolkit. J. Mach. Learn. Res. 2009, 10, 1755–1758. Igel, C.; Heidrich-Meisner, V.; Glasmachers, T. Shark. J. Mach. Learn. Res. 2008, 9, 993–996. Kapás, Z.; Lefkovits, L.; Szilágyi, L. Automatic Detection and Segmentation of Brain Tumor Using Random Forest Approach. In Modeling Decisions for Artificial Intelligence; Springer: New York, NY, USA, 2016; pp. 301–312. Menze, B.H.; Jakab, A.; Bauer, S.; Kalpathy-Cramer, J.; Farahani, K.; Kirby, J.; Burren, Y.; Porz, N.; Slotboom, J.; Wiest, R.; et al. The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS). IEEE Trans. Med. Imaging 2015, 34, 1993–2024. [CrossRef] [PubMed] Wo´zniak, M.; Graña, M.; Corchado, E. A survey of multiple classifier systems as hybrid systems. Inf. Fusion 2014, 16, 3–17. [CrossRef] Termenon, M.; Graña, M. A two stage sequential ensemble applied to the classification of Alzheimer’s disease based on MRI features. Neural Process. Lett. 2012, 35, 1–12. [CrossRef] Tipping, M.E. Sparse Bayesian learning and the relevance vector machine. J. Mach. Learn. Res. 2001, 1, 211–244. Geremia, E.; Menze, B.H.; Ayache, N. Spatial Decision Forests for Glioma Segmentation in Multi-Channel MR Images. MICCAI Chall. Multimodal Brain Tumor Segmentation 2012, 34. Bauer, S.; Nolte, L.P.; Reyes, M. Fully automatic segmentation of brain tumor images using support vector machine classification in combination with hierarchical conditional random field regularization. In Medical Image Computing and Computer-Assisted Intervention—MICCAI; Springer: Berlin/Heidelberg, Germany, 2011.
Symmetry 2018, 10, 627
23. 24.
9 of 9
Criminisi, A.; Zikic, D.; Glocker, B.; Shotton, J. Context-sensitive classification forests for segmentation of brain tumor tissues. Proc MICCAI-BraTS 2012, 1, 1–9. Festa, J.; Pereira, S.; Mariz, J.; Sousa, N.; Silva, C. Automatic brain tumor segmentation of multi-sequence MR images using random decision forests. Proc. NCI-MICCAI BRATS 2013, 1, 23–26. © 2018 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (http://creativecommons.org/licenses/by/4.0/).