MRI in NeurosciencesView this Special Issue
Label Fusion Strategy Selection
Label fusion is used in medical image segmentation to combine several different labels of the same entity into a single discrete label, potentially more accurate, with respect to the exact, sought segmentation, than the best input element. Using simulated data, we compared three existing label fusion techniques—STAPLE, Voting, and Shape-Based Averaging (SBA)—and observed that none could be considered superior depending on the dissimilarity between the input elements. We thus developed an empirical, hybrid technique called SVS, which selects the most appropriate technique to apply based on this dissimilarity. We evaluated the label fusion strategies on two- and three-dimensional simulated data and showed that SVS is superior to any of the three existing methods examined. On real data, we used SVS to perform fusions of 10 segmentations of the hippocampus and amygdala in 78 subjects from the ICBM dataset. SVS selected SBA in almost all cases, which was the most appropriate method overall.
Label fusion is a process used in medical image segmentation. Its aim is to produce a single, discrete element or label from a combination of multiple independent inputs. The merged result is potentially more accurate, with respect to the exact, sought segmentation, than each individual input due to the reduction of uncorrelated errors. Labels can be obtained by combining inputs from different raters or automated segmentations [1, 2].
A long-term goal of our research program is to obtain accurate, automated segmentations of neuroanatomical structures, primarily the hippocampus (HC). Our primary motivation stems from our work in Alzheimer’s disease, for which HC volume and atrophy measurements are putative disease markers (see reviews in [3–6]). Of the multiple HC segmentation approaches available (see  for review), novel template-based paradigms propose the use of template libraries . In such approaches, a single label is found by combining multiple individually segmented HC through label fusion .
To reach our goal, we thus decided to investigate different fusion processes. To suit our research context, we restricted our analysis to techniques that depend solely on given input labels. We disregarded techniques that depend on intensity images [9, 10], since these images may sometimes be unavailable or noisy. We also ignored techniques that depend on object-specific training, i.e. that have geometric or topological prior.
Our first objective was to characterize applicable label fusion strategies. The first approach is the Vote method (or sum rule), which has been widely used and described by virtue of its simplicity [1, 9, 11–13]. The second is also a well-known technique called Simultaneous Truth and Performance Level Estimation (STAPLE), initially proposed by Warfield et al. [14, 15], and used in a variety of studies [9, 16]. The third approach is referred to as Shape-Based Averaging (SBA), which incorporates spatial information .
While testing the implementations of these three approaches on simulated data, we observed that the technique with a result closest to the ground truth was not the same depending on the dissimilarity between raters’ input labels, as detailed below. Therefore, the second objective of our study was to propose an empirical, hybrid STAPLE-Vote-SBA (SVS) technique that automatically selects the right label fusion approach based on this dissimilarity.
We report results of comparison tests on the four label fusion methods for simulated two-dimensional (2D) and three-dimensional (3D) data as well as HC and amygdala (AG) labels obtained from magnetic resonance images (MRI). All images used in this study were binary. For the real data, we performed label fusion on HC and AG independently.
2. Materials and Methods
2.1. Mathematical Notation
Our mathematical notation is as follows. We consider an image of pixels or voxels () for which raters () each produces a binary label segmentation . To each element of , i.e. each pixel/voxel , is assigned a label equal to 0 or 1, for background and segmented object, respectively. A decision matrix is formed with all the vectors, with size , and fed to a label fusion algorithm to obtain an estimate of the true segmentation .
For evaluating the performance of SVS with respect to STAPLE, Vote and SBA, our data consisted of 2D and 3D simulated data as well as real data.
2.2.1. Two-Dimensional (2D) Simulated Data
We created two simulated 2D data sets: one for training SVS and one for testing the label fusion approaches. The SVS version trained with 2D data is hereafter referred to as SVS-2D.
The data consisted of multiple binary images created from a ground-truth object, shown in Figure 1(a), which was an ellipse geometry defined by eight control points interpolated with cubic splines.
(a) Ground truth
(d) Ground truth
We generated individual, simulated rater images by moving the control points of the ground-truth ellipse and reinterpolating with cubic splines. We moved the control points in random directions, following a uniform distribution, with random distances from their original coordinates. The random distance followed a normal distribution of zero mean with a standard deviation adjusted so that it could be modified by a normalized deformation factor (between 0 and 1) to create images with a relative difference area ranging from 0% to 50%, where is given by where corresponds to the area in pixels of the ground-truth ellipse. represents the number of pixels in the image that are different between decision of rater and the ground truth :
For each of the training and testing sets, we created 625 label fusion tests, each consisting of 10 deformed images, for a total of 6,250 images in the training set and 6,250 different images in the testing set. Each test was created by varying of the test images according to a given Gaussian distribution. For each test, different mean and standard deviation were used for , ranging both from 0 to 1 with 25 linearly spaced points each, making a total of 625 Gaussian distributions, one for each test. Negative values of and values higher than 1 were clamped to 0 and 1, respectively. We performed the label fusion of the 10 deformed images in each of the 625 tests of the testing set.
2.2.2. Three-Dimensional (3D) Simulated Data
As for the 2D case, we created two simulated 3D sets: one for training SVS and one for testing the label fusion techniques. The SVS version trained with 3D data is hereafter referred as SVS-3D. An SVS version was also trained with the combination of 2D and 3D training sets. It is referred as SVS-2D&3D.
The 3D data consisted of binary volume images created from a ground-truth ellipsoid. To produce the ground truth, we first created a cubic regular grid volume. This volume was then warped along each axis by dividing each voxel coordinate by its corresponding ground-truth ellipsoid radius, creating a warped grid. By applying this warping transformation, the ellipsoidal space became a spherical space. A ground-truth sphere was created by regularly sampling the angles and in the spherical-coordinate space , giving a set of 26 control points .
To produce the ground-truth image, the control points were projected into a Cartesian space with the following axes: , , and . We transformed the warped grid into spherical coordinates and performed a cubic interpolation of on to find at each point . For each grid voxel, if , the voxel was considered inside the sphere and was labeled accordingly. The warped grid (spherical space) was then unwarped into the regular grid (ellipsoidal space) to give the desired ground-truth ellipsoid image shown in Figure 1(d).
While appearing complex, this process in fact simplified the creation of the deformed ellipsoid images. We randomly moved the control points of the ground-truth sphere along , modifying , reinterpolated to find for the warped grid, performed the labeling by thresholding (i.e. ), and unwarped the grid to obtain the deformed ellipsoid.
As for the 2D sets, the random distance followed a normal distribution of zero mean. The standard deviation was adjusted so that it could be modified by to create deformed ellipsoids with relative difference in volume, (1), ranging between 0% and 50% with respect to the ground truth.
Figures 1(e) and 1(f) show two examples of deformed images with of 25% and 50%, respectively. As for the 2D data, we produced a training set and a testing set, each consisting of 625 label fusion tests. Each test was created as previously described and comprised 10 deformed images. Each of the training and testing sets thus consisted of 6,250 images. We performed the label fusion of the 10 deformed images in each of the 625 tests of the testing set.
2.2.3. Real MRI Data
The real MRI data consisted of intensity images and segmented left and right HC and AG labels of 78 young, neurologically healthy subjects part of the ICBM database . Subjects were scanned in Montréal (Québec, Canada) on a Philips Gyroscan 1.5T scanner (Philips Medical, Best, Netherlands) using a T1-weighted fast gradient echo sequence (sagittal acquisition, TR = 18 ms, TE = 10 ms, 1-mm3 voxels, flip angle = 30°).
The ground truth consisted of left and right HC and AG manual labels, presented in a previous study , with a reported intraclass reliability coefficient of 0.900 and 0.925 for interrater and intrarater reliability, respectively.
The labels available for fusion were obtained using a template-based segmentation algorithm . In this approach, each subject’s image is compared in turn to a library of other such images; the 10 images with highest match (e.g., highest normalized mutual information) are selected and then nonlinearly aligned with the original subject image. Given that each image in the library has an associated label, inverse warping allows the transfer of label in the original subject’s space, where they must be fused to provide a single object. In our dataset, we received 10 labels for each subject, obtained with this technique, for each of the four following regions: left HC, right HC, left AG, and right AG. Label fusions were then performed independently for each region, giving a total of 312 label fusions (78 subjects × 4 regions). We assessed the performance of the fusions using the manual segmentations as “ground truths”.
2.3. Label Fusion Strategies
The next sections present the three existing label fusion strategies that we used in this study: STAPLE, Vote, and SBA. We implemented all label fusion methods, including SVS, in MATLAB (MathWorks, Natick, MA, USA).
It is important to note that all approaches were applied to the disputed pixels/voxels only. Pixels/voxels for which all the raters unanimously agreed on their label were not considered; the label was automatically assigned. Working with only disputed pixels/voxels speeded up computation for all methods and significantly improved the results given by STAPLE (see ).
STAPLE is an expectation-maximization (EM) algorithm that iteratively estimates (1) the true segmentation from the raters’ performance (E-step) and (2) the raters’ performance (sensitivity and specificity) from this true segmentation estimate (M-step). We implemented STAPLE following the mathematical description in .
The Vote method consists of summing for each pixel/voxel and label , the occurrences of label among the raters, and assigning the most occurring label to .
SBA is a voting scheme where each vote is weighted by the signed Euclidean distance computed for each input label. In this study, SBA is the only method that incorporates spatial information in the label fusion process. We implemented this method following the mathematical description in .
2.4. Label Fusion Strategy Selection: SVS
SVS is a strategy that selects the most appropriate method among STAPLE, Vote, and SBA, based solely on the input labels and their dissimilarity. We point out that SVS is not limited to these three label fusion methods. It could easily be extended to include further methods.
2.4.1. Experimental Observations
We developed SVS after observing, during our simulations, that the performance of STAPLE, Vote, and SBA was dependent on the distribution of in the input labels of each label fusion test. This can be observed in the scatter plots of Figure 2 obtained for the 2D (a, b, and c) and 3D training sets (d, e, and f). The scatter plots show centered on the Vote’s values, i.e. (), after label fusions performed with STAPLE (red), Vote (blue), and SBA (green), as a function of the mean (a, d), standard deviation (b, e), and coefficient of variation (c, f) of , calculated over the input labels for each test.
We note that and give an idea of how differently the raters perform between themselves, while measures how bad the raters are overall. These measures thus describe, in a way, the dissimilarity in the raters’ input labels.
As can be seen, none of STAPLE, Vote, and SBA can be considered superior to the others. The choice of the best method seems to depend on the distribution of . For low values of , which better discriminates the label fusion methods than , SBA seems better (i.e. with lower values of after label fusion), while, for higher values, STAPLE would be a better choice. Focusing on the results with respect to , STAPLE seems better at lower values, and SBA, at higher values. We also observe that in none of the cases does Vote clearly outperform the other methods.
These observations thus suggested that and could be used to determine the appropriate label fusion method.
2.4.2. Dissimilarity Factors
The measures and cannot be used in practice since the computation of depends on (1) and , and thus requires to know the ground truth, which is what we try to estimate with label fusion. We thus needed to find estimates for and .
We overcame this problem by using the following scheme. For , we first computed the frequency of occurrence , between 0 and 1, of each label for each pixel/voxel over all raters:
We then computed, for each rater and each pixel/voxel , the estimated probability that rater misclassifies pixel/voxel , i.e. that the assigned label was a false positive or a false negative:
For each estimated rater’s probability , we then performed a Bernouilli trial with experiments to compute the probability that a majority of “virtual” raters misclassified pixel/voxel , according to :
This last equation corresponds to a cumulative sum of the upper half of the probability mass function of a binomial distribution. In this study, we used so that ranged from 50 to 99. An odd number for was used to separate the binomial probability mass function equally into a lower and an upper part, the latter corresponding to a clear majority.
From (5), we were able to compute an estimate of by summing over all pixels/voxels:
To estimate , we used (3) in a similar Bernouilli trial approach. For each pixel/voxel , we computed a probability that a majority of “virtual” raters classifies pixel/voxel as being part of the segmented region, i.e. with label 1, according to :
We then summed over all pixels/voxels to obtain an estimate of :
From and , we defined two empirical factors: the dissimilarity coefficient , estimating , and the dissimilarity ratio , estimating . These factors are respectively given by
In Figure 3, we demonstrate the performance of these estimates by showing that (a, c) and (b, d) match, with a quasi-one-to-one relationship, their theoretical values and , respectively, for both the 2D (a, b) and 3D (c, d) training sets.
2.4.3. SVS Training
To perform its selection, SVS finds a score , from the dissimilarity factors and , for each of STAPLE, Vote, and SBA, i.e. , and , and selects the label fusion method that gives the highest score. The following training procedure was used to determine the scoring functions , , and .(1)For each label fusion test of a given training set, we computed and , according to the approach presented in the last section.(2)After performing label fusion with STAPLE, Vote, and SBA, we first summed, for each label fusion method and test , the number of pixels/voxels that were different between the label fusion result and the ground truth , i.e. the number of false positives and false negatives: For each test , we assigned a score of 1 to the label fusion method with the lowest , corresponding to the best method, 0 to the method with the highest , corresponding to the poorest method, and we linearly interpolated the score value for the remaining method.(3)Following the last two steps of the training procedure, we had, for each test , the five following values: , , , , and . To obtain the continuous scoring functions , , and , we finally fitted, for each method , a surface using locally weighted linear regression (MATLAB Curve Fitting Toolbox, MathWorks, Natick, MA, USA).
This procedure was performed for each of the 2D and 3D training data sets as well as the combination of both sets resulting in three versions of SVS: SVS-2D (trained with 2D data), SVS-3D (trained with 3D data), and SVS-2D&3D (trained with 2D and 3D data). We note that using this scheme, other label fusion methods could be incorporated in SVS, increasing only the number of scoring functions .
Figure 4 presents, for SVS-2D (a), SVS-3D (b), and SVS-2D&3D (c), the scoring surface functions in the space as well as the selection regions in the space , where each method gives the highest score. The latter images thus correspond to the top views of the firsts. We observe that the three versions of SVS give very similar delimitations between the methods. Interestingly, with SVS-2D&3D, the border between STAPLE and SBA is almost linear in the region of covered by the label fusion tests.
2.4.4. SVS Selection
We can now describe the SVS method as follows.(1)Compute the dissimilarity coefficient and the dissimilarity ratio from the raters’ input labels, as described in Section 2.4.2.(2)Find the score for each label fusion method using its corresponding scoring surface function.(3)Select the label fusion method corresponding to the highest score.
In case of two or more equal scores, which do not imply identical label fusions, a weighted vote “meta fusion” of the label fusion results, obtained with STAPLE, Vote, and SBA, is performed using the scores as weights. In practice, this situation is uncommon. We point out that, besides the SVS versions presented here, this “meta fusion” approach, i.e. performing a label fusion of STAPLE, Vote, and SBA, has also been tested (results not presented), using each of STAPLE, Vote, and SBA as “meta fusion” method with and without score weights for the two latter methods. However, no “meta fusion” outperformed the versions of SVS presented in this study.
We also point out that and depend on the decision matrix only, i.e. the input labels. Effectively, this ensures that there are no external parameters to the input data that may affect the sensitivity of the technique. Moreover, since and are normalized values, we believe that the technique should not be sensitive to the training data. In fact, we observe in Figure 4 that the different training sets gave similar regions.
2.5. Performance Measure
To measure the performance of the label fusion techniques, we computed , as well as the Dice similarity coefficient , an established measure widely reported in the field [1, 2, 9, 11, 12, 15], between each label fusion image and the ground truth. is given by where is the area or volume of the segmented region .
To further characterize our testing sets and insure the deformation factor reflected its initial intent, we computed the between each deformed image and its ground truth. Figures 5(a), 5(b), 5(d), and 5(e) show the relationship between , along with , and the deformation factor for the 2D (a, b) and 3D (d, e) testing sets. Figures 5(c) and 5(f) show the quasilinear relationship between and .
3.1. 2D Simulated Data
The three existing techniques (STAPLE, Vote, and SBA) as well as the three versions of SVS (SVS-2D, SVS-3D, and SVS-2D&3D) were used to perform the label fusion of the 10 images of each of the 625 tests of the 2D testing set. Figures 6(A)–6(D) show boxplots of (A, C) and (B, D), centered on the Vote values, obtained with the six fusion methods. To see the improvement brought by SVS (methods d, e, and f), the boxplots have been separated in two groups, Group STAPLE and Group SBA, determined by the selection performed by SVS-2D&3D on the testing set (see Figure 7(a), right). The data in Group STAPLE and Group SBA are the tests for which SVS-2D&3D selected STAPLE and SBA, respectively. We see that the SVS boxplots, matching the selected method’s, give in both groups higher and lower , while each of STAPLE (method a) and SBA (method c) is outperformed in its counterpart group. Regarding Vote (method b), it gives better performance than SBA in Group STAPLE but seems to be the worse method in Group SBA. We also see that the three versions of SVS are similar despite the different training sets. Figure 7(a) presents the distribution of the pairs for all the tests of the testing set among the regions of SVS-2D, SVS-3D, and SVS-2D&3D.
3.2. 3D Simulated Data
The experiment described in the last section was also performed on the 3D testing set. and boxplots are presented in Figures 6(E, G) and 6(F, H), respectively. The results are very similar to the 2D testing set’s; the three versions of SVS give in both groups higher and lower . Figure 7(b) shows the distribution of the pairs among the selection regions.
3.3. Real Data
Figure 8 presents the (A–D) and (E–H) boxplots, respectively, obtained for each of HC and AG, left and right. Since the three versions of SVS (methods d, e, and f) selected SBA for nearly all label fusions, as shown in Figure 7(c), the boxplots are almost identical to SBA’s. We see that SBA/SVS overall gives the highest DSC and the lowest . This is also shown in Figure 9, which presents scatter plots of (a, b) and (c, d), centered on the Vote values, as a function of (a, c) and (b, d) for all the 312 label fusion cases. SBA/SVS is overall superior to STAPLE and Vote, with and respectively above and below STAPLE and Vote means for the majority of the label fusion cases.
We showed on a large set of different simulated data that the label fusion method giving the label closest to the ground truth was not the same depending on the dissimilarity among the raters.
Regarding robustness, we showed that SVS outperformed any single method among STAPLE, Vote, and SBA, regardless of the training set. Applying SVS-2D (trained with 2D data) and SVS-3D (trained with 3D data) on 3D and 2D data, respectively, we still obtained better performance than STAPLE, Vote, and SBA. Effectively, the three versions of SVS showed similar results, explained by similar selection regions (Figure 4). This suggests that SVS is independent of the type of training set, 2D or 3D, and that the delimitations of the selecting regions with SVS-2D&3D could represent what we should really expect since there are more training tests.
We also demonstrated that with real data, Vote was not necessarily the method of choice; in our study, SBA was better than Vote and STAPLE. To our knowledge, SBA has not been widely used in the literature, and it might have been underestimated.
The first and obvious limitation of the SVS technique is that it is upper-bound limited to the best technique (either STAPLE, Vote, or SBA) in each case.
Secondly, we used and in this study as the criteria for assessing the label fusion methods, the first being commonly used in the literature. However, we think that gives a better indication of the difference between a rater image and the ground truth. This is demonstrated in Figure 8 for HC left, HC right, and AG right. For these regions, while STAPLE’s medians are higher (better) than Vote’s, medians are higher (worse), meaning that there are more false positives and/or negatives. Also, in Figures 5(c) and 5(f), we show that compared to a point with given and , a neighbor point with a higher (more false positives and/or negatives) can still give a higher (better) or similar , especially for high . This difference between and might be explained by the fact that normalizes by the mean area/volume of the label fusion and ground truth, while normalizes by the area/volume of the ground truth only. Therefore, the denominator in remains constant, while the denominator in varies between label fusions. The comparison is thus not performed on the same basis. Although we could argue on which measure is the most appropriate, this questions the validity of as a performance measure for label fusion if the ground truth is available. We thus keep in mind for future work that is not necessarily the best criterion in this case and that should be used instead.
Thirdly, we did not assess the influence of the number and the selection of input labels on the performance of the label fusion strategies. While these two aspects are important, as reported in some studies [2, 12], our objectives were primarily to characterize three existing label fusion strategies and to propose a selection method based on our observations. We will confront these aspects in future work.
We proposed a method that automatically selects the most appropriate label fusion method based on the dissimilarity of input labels. Overall, the SVS technique performed better with simulated data compared to either individual technique among STAPLE, Vote, and SBA. For real data, SVS selected SBA for almost all cases, which was overall superior to STAPLE and Vote.
|DSC:||Dice similarity coefficient|
|MRI:||Magnetic resonance imaging|
|STAPLE:||Simultaneous Truth and Performance Level Estimation|
The authors thank Dr. J. C. Pruessner and Dr. D. L. Collins (McGill University, Montréal, Canada), and the International Consortium for Brain Mapping for access to label and MRI data. This work was supported by an operating grant from the Ministère du Développement Économique, de l’Innovation, et de l’Exportation du Québec.
K. Kantarci and C. R. Jack, “Neuroimaging in Alzheimer disease: an evidence-based review,” Neuroimaging Clinics of North America, vol. 13, no. 2, pp. 197–209, 2003.View at: Google Scholar
M. Chupin, A. R. Mukuna-Bantumbakulu, D. Hasboun et al., “Anatomically constrained region deformation for the automated segmentation of the hippocampus and the amygdala: method and validation on controls and patients with Alzheimer's disease,” NeuroImage, vol. 34, no. 3, pp. 996–1019, 2007.View at: Publisher Site | Google Scholar
S. K. Warfield, K. H. Zou, and M. Wells III, “Validation of image segmentation and expert quality with an expectation-maximization algorithm,” in Proceedings of the 5th International Conference on Medical Imaging Computing and Computer Assisted Interventions (MICCAI '02), pp. 298–306, Springer, Tokyo, Japan, September 2002.View at: Google Scholar
T. Rohlfing, D. B. Russakoff, and C. R. Maurer Jr., “Expectation maximization strategies for multi-atlas multi-label segmentation,” Inf Process Med Imaging, vol. 18, pp. 210–221, 2003.View at: Google Scholar
J. C. Pruessner, D. L. Collins, M. Pruessner, and A. C. Evans, “Age and gender predict volume decline in the anterior and posterior hippocampus in early adulthood,” Journal of Neuroscience, vol. 21, no. 1, pp. 194–200, 2001.View at: Google Scholar