Automatic brain Magnetic Resonance Imaging (MRI) tissue segmentation continues being an active research topic in medical image analysis since it provides doctors with meaningful and reliable quantitative information, such as tissue volume measurements. This information is widely used to diagnose brain diseases and to evaluate progression through regular MRI analysis over time (Rovira et al., 2015; Steenwijk et al., 2016; Filippi et al., 2016). Hence, MRI and its study contribute to a better comprehension of the nature of brain problems and the effectiveness of new treatments.
Several tissue segmentation algorithms have been proposed throughout the years. Many supervised machine learning methods existed before the Convolutional Neural Network (CNN) era. A clear example of that is the pipelines that participated in the MRBrainS13 challenge (Mendrik et al., 2015). Commonly, intensity-based methods assumed each tissue could be represented by its intensity values (Cardoso et al., 2013) (e.g. using GMM models). Since noise and intensity inhomogeneities degraded them, they were later equipped with spatial information (Clarke et al., 1995; Kapur et al., 1996; Liew & Yan, 2006; Valverde et al., 2015). Four main strategies were distinguished in the literature: (i) impose local contextual constraints using Markov Random Fields (MRF) (Zhang et al., 2001), (ii) include penalty terms accounting for neighbourhood similarity in clustering objective functions (Pham, 2001), (iii) use Gibbs prior to model spatial characteristics of the brain (Shattuck et al., 2001) and (iv) introduce spatial information using probabilistic atlases (Ashburner & Friston, 2005; Ashburner et al., 2012). It is important to remark that some of these methods, like FAST (Zhang et al., 2001) and SPM (Ashburner & Friston, 2005; Ashburner et al., 2012), are still being used in medical centres due to their robustness and adaptability (Valverde et al., 2017).
Nowadays, CNNs have become appealing to address this task in coming years since (i) they have achieved record-shattering performances in various fields in computer vision and (ii) they discover classification-suitable representations directly from the input data – unlike conventional machine-learning strategies. However, unlike traditional approaches, these methods still present two main issues when placed in real life scenarios: (i) lack of sufficiently labelled data and (ii) domain adaptation issues – also related to generalisation problems. Seminal work on CNN for brain tissue segmentation date back to when Zhang et al. (2015)
proposed a CNN to address infant brain tissue segmentation on MRI where tissue distributions overlap and, hence, the GMM assumption does not hold. The authors showed that their CNN was suitable for the problem and could outperform techniques, such as random forest, support vector machines, coupled level sets, and majority voting. From thereon, many more sophisticated proposals have been devised(Litjens et al., 2017; Bernal et al., 2017).
Former CNN strategies for tissue segmentation were trained to provide a single label given an input patch (Zhang et al., 2015; Moeskops et al., 2016; Chen et al., 2016). Naturally, both training and testing can be time-consuming and computationally demanding. Also, the relationship between neighbouring segmented voxels is not encoded – in principle – and, consequently, additional components on the architecture (such as in (Stollenga et al., 2015)) or post-processing may be needed to smooth results. These drawbacks can be diminished by adapting the network to perform dense prediction. The prevailing approach consists in replacing fully connected layers by convolutional layers – if processing 3D data. This particular group is known as Fully CNN (FCNN) (Long et al., 2015).
Regarding input dimensionality, three main streams are identified: 2D, 2.5D and 3D. At the beginning of the CNN era, most of the state-of-the-art CNN techniques were 2D, in part, due to (i) their initial usage on natural images, and (ii) computation limitations of processing 3D volumes directly. Surely, three independent 2D models can be arranged to handle patches from axial, sagittal and coronal at the same time, hence improving acquired contextual information. These architectures are referred in the literature as 2.5D (Lyksborg et al., 2015; Birenbaum & Greenspan, 2016; Kushibar et al., 2017). With advances in technology, more 3D approaches have been developed and attracted more researchers as they tend to outperform 2D architectures (Bernal et al., 2017). Intuitively, the improvement of 3D over 2D and 2.5D lies on the fact that more information from the three orthogonal planes is integrated into the network – i.e., more contextual knowledge is acquired. However, this does not hint they always perform better (Ghafoorian et al., 2017). It is important to highlight that, to the best of our knowledge, no 2.5D FCNN network has been created yet.
In this paper, we analyse quantitatively FCNN architectures for tissue segmentation on brain MRI. These networks, comprising 2D and 3D implementations, are inspired in four recent works (Çiçek et al., 2016; Dolz et al., 2017; Guerrero et al., 2017; Kamnitsas et al., 2017). The models are tested on three well-known datasets of infant and adult brain scans, with different spatial resolution, voxel spacing, and image modalities. In this paper, we aim to (i) compare different FCNN strategies for tissue segmentation; (ii) quantitatively analyse the effect of network’s dimensionality (2D or 3D) for tissue segmentation and the impact of fusing information from single or multiple modalities; and (iii) investigate the effects of extracting patches with a certain degree of overlap as a sampling strategy in both training and testing. To the best of our knowledge, this is the first work providing a comprehensive evaluation of FCNNs for the task mentioned above.
The rest of the paper is organised as follows. In Section 2, we present our evaluation framework: selected datasets, assessed networks, aspects to analyse, pipeline description and implementation details. Results are reported in Section 3 and analysed in Section 4. Final remarks are discussed in Section 5.
2 Materials and methods
2.1 Considered datasets
Public datasets are commonly used for assessing brain MRI tissue segmentation algorithms as they provide ground truth labels. In this work, we consider one publicly available repository and two challenges: Internet Brain Segmentation Repository 18 (IBSR18)111http://www.nitrc.org/projects/ibsr, MICCAI Multi-Atlas Labeling challenge 2012 (MICCAI 2012)222https://masi.vuse.vanderbilt.edu/workshop2012 and 6-month infant brain MRI segmentation (iSeg2017)333http://iseg2017.web.unc.edu/, respectively. The datasets were chosen since (i) they have been widely used in the literature to compare different methods and also (ii) they contain infants and adults data, with different voxel spacing and a varied number of scans. To use annotations of MICCAI 2012, we mapped all the labels to form the three tissue classes. We believe that these two factors allow us to see how robust, generalised and useful in different scenarios the algorithms can be. Specific details of these datasets are presented in Table 1.
|Number of scans|
2.2 FCNNs for brain MRI segmentation tasks
The proposed works using FCNN for brain MRI segmentation tasks are listed in Table 2. Proposals comprise single or multiple flows of information – referred in the literature as single-path and multi-path architectures, respectively. While single-path networks process input data faster than multi-path, knowledge fusion occurring in the latter strategy leads to better segmentation results: various feature maps from different interconnected modules and shallow layers are used to produce a final verdict (Moeskops et al., 2016). Under this scheme, a network is provided with contrast, fine-grained and implicit contextual information. Furthermore, proposals apply successive convolutions only or convolutions and de-convolutions in the so-called u-shaped models. The latter approach commonly considers connections from high-resolution layers to up-sampled ones to retain location and contextual information (Çiçek et al., 2016; Ronneberger et al., 2015; Milletari et al., 2016).
|Brosch et al. (2016)||3D multi-path [U]||Lesion|
|Kleesiek et al. (2016)||3D single-path||Skull stripping|
|Nie et al. (2016)||2D single-path [U]||Tissue|
|Shakeri et al. (2016)||2D single-path||Sub-cortical structure|
|Kamnitsas et al. (2017)||3D multi-path||Lesion/tumour|
|Dolz et al. (2017)||3D multi-path||Sub-cortical structure|
|Guerrero et al. (2017)||2D multi-path [U]||Lesion|
|Moeskops et al. (2017)||2D single-path||Structure|
From the papers indexed in Table 2, we built four multi-path architectures inspired by the works of Kamnitsas et al. (2017), Dolz et al. (2017), Çiçek et al. (2016) and Guerrero et al. (2017) (i.e. two convolution-only and two u-shaped architectures). The networks were implemented in 2D and 3D to investigate the effect of the network’s dimensionality in tissue segmentation. All these architectures were implemented from scratch following the architectural details given in the original work and are publicly available at our research website444https://github.com/NIC-VICOROB/tissue_segmentation_comparison. Although slight architectural differences may be observed, the core idea of the proposals is retained. More details of the networks are given in the following sections.
2.2.1 Networks incorporating multi-resolution information
Kamnitsas et al. (2017), proposed a two-path 3D FCNN applied to brain lesion segmentation. This approach achieved top performance on two public benchmarks, BRATS 2015 and ISLES 2015. By processing information of the targeted area from two different scales simultaneously, the network incorporated local and larger contextual information, providing a more accurate response (Moeskops et al., 2016). A high-level scheme of the architecture is depicted in Fig. 0(a). Initially, two independent feature extractor modules extracted maps from patches from normal and downscaled versions of an input volume. Each module was formed by eight convolutional layers using between and kernels. Afterwards, features maps were fused and mined by two intermediate convolutional layers with kernels. Finally, a classification layer (another convolutional layer) produced segmentation prediction using a softmax activation.
Dolz et al. (2017) presented a multi-resolution 3D FCNN architecture applied to sub-cortical structure tissue segmentation. A general illustration of the architecture is shown in Fig. 0(b). The network consisted of 13 convolutional layers: nine , and four
. Each one of these layers was immediately followed by a Parametric Rectified Linear Unit (PReLU) layer, except for the output layer which activation was softmax. Multi-resolution information was integrated into this architecture by concatenating feature maps from shallower layers to the ones resulting from the lastconvolutional layer. As explained by Hariharan et al. (2015), this kind of connections allows networks to learn semantic – coming from deeper layers – as well as fine-grained localisation information – coming from shallow layers.
2.2.2 U-shaped networks
In the u-shaped network construction scheme, feature maps from higher resolution layers are commonly merged to the ones on deconvolved maps to keep localisation information. Merging has been addressed in the literature through concatenation (Çiçek et al., 2016; Brosch et al., 2016) and addition (Guerrero et al., 2017). In this paper, we consider networks using both approaches. A general scheme of our implementations inspired in both works is displayed in Fig. 0(c).
Çiçek et al. (2016) proposed a 3D u-shaped FCNN, known as 3D u-net. The network is formed by four convolution-pooling layers and four deconvolution-convolution layers. The number of kernels ranged from in its bottommost layers to
Guerrero et al. (2017) designed a 2D u-shaped residual architecture applied on lesion segmentation, referred as u-ResNet. The building block of this network was the residual module which (i) added feature maps produced by - and -kernel convolution layers, (ii) normalised resulting features using batchnorm, and, finally, (iii) used a ReLU activation. The network consisted of three residual modules with , and kernels, each one followed by a max pooling operation. Then, a single residual module with kernels was applied. Afterwards, successive deconvolution-and-residual-module pairs were employed to enlarge the networks’ output size. The number of filters went from to in the layer before the prediction one. Maps from higher resolutions were merged with deconvolved maps through addition.
2.3 Evaluation measurement
To evaluate proposals, we used the Dice similarity coefficient (DSC) (Dice, 1945; Crum et al., 2006). The DSC is used to determine the extent of overlap between a given segmentation and the ground truth. Given an input volume , its corresponding ground truth , and obtained segmentation output , the DSC is mathematically expressed as
where represents the cardinality of the set. The values for DSC lay within , where the interval extremes correspond to null or exact similarity between the compared surfaces, respectively. Additionally, we consider the Wilcoxon signed-rank test to assess the statistical significance of differences among architectures.
2.4 Aspects to evaluate
As we mentioned previously, this paper aims to analyse (i) overlapping patch extraction in training and testing, (ii) single and multi-modality architectures and (iii) 2D and 3D strategies. Details on these three evaluation cornerstones are discussed in the following sections.
2.4.1 Overlapping sampling in training and testing
One of the drawbacks of networks performing dense-inference is that – under similar conditions – the number of parameters increases. This implies that more samples should be used in training to obtain acceptable results. The most common approach consists in augmenting the input data through transformations – e.g. translation, rotation, scaling. However, if the output dimension is not equal to the input size, other options can be considered. For instance, patches can be extracted from the input volumes with a certain extent of overlap and, thus, the same voxel would be seen several times in different neighbourhoods. An example of this technique can be observed in Fig. 2. Consequently, (i) more samples are gathered, and (ii) networks are provided with information that may improve spatial consistency as illustrated in Fig. 3 (a-d).
The sampling strategy aforementioned can be additionally enhanced by overlaying predicted patches. Unlike sophisticated post-processing techniques, the network itself is used to improve its segmentation. As depicted in Fig. 3 (e-h), the leading property of this post-processing technique is that small segmentation errors – e.g. holes and block boundary artefacts – are corrected. The consensus among outputs can be addressed through majority voting, for instance.
2.4.2 Input modalities
Depending on the number of modalities available in a dataset, approaches can be either single- or multi-modality. If many modalities were acquired, networks could be adapted to process them all at the same time either using different channels or various processing paths – also referred in the literature as early and late fusion schemes (Ghafoorian et al., 2017), respectively. Naturally, regarding computational resources the former strategy is desirable, but the latter may extract more valuable features. In this work, we consider the early fusion only. Regardless of the fusion scheme, merging different sources of information may provide models with complementary features which may lead to enhanced segmentation results (Zhang et al., 2015).
2.4.3 Network’s dimensionality
There are two streams of FCNN regarding its input dimensionality: 2D and 3D. On the one hand, 2D architectures are fast, flexible and scalable; however, they ignore completely data from neighbouring slices, i.e. implicit information is reduced compared to 3D approaches. On the other hand, 3D networks acquire valuable implicit contextual information from orthogonal planes. These strategies tend to lead to better performance than 2D – even if labelling is carried out slice-by-slice – but they are computationally demanding – exponential increase in parameters and resource consumption. Moreover, due to the increasing number of parameters, these models may require larger training sets. Therefore, depending on the data itself, one approach would be more suitable than the other.
2.5 Implementation details
General tissue segmentation pipelines contemplate four essential components: preprocessing, data preparation, classification and post-processing. Specific implementations of each one of these elements can be plugged and unplugged as required to achieve the best performance. First, preprocessing is carried out by (i) removing skull, and (ii) normalising intensities between scans. We use the ground truth masks to address the former tasks and standardise our data to have zero mean and unit variance. Second, data is prepared by extracting useful and overlapping patches – containing information from one of the three tissues. Third, classification takes place; segmentation of an input volume is provided through means of majority voting in case of overlapping predictions. Fourth, no post-processing technique was considered.
All the networks were trained maximum for 20 epochs. Training stopping criterium was overfitting, which was monitored using an early stopping policy with patience equal to. For each of the datasets, we split the training set into training and validation (80% and 20% of the volumes, respectively). Additionally, voxels laying on the background area were given a weight of zero to avoid considering them in the optimisation/training process.
All the architectures have been implemented from scratch in Python, using the Keras library. From here on, our implementations ofDolz et al. (2017), Kamnitsas et al. (2017), Çiçek et al. (2016), and Guerrero et al. (2017) are denoted by , , and , respectively. The number of parameters per architecture is listed in Table 3. All the experiments have been run on a GNU/Linux machine box running Ubuntu 16.04, with 128GB RAM. CNN training and testing have been carried out using a single TITAN-X PASCAL GPU (NVIDIA corp., United States) with GB RAM. The developed framework for this work is currently available to download at our research website. The source code includes architecture implementation and experimental evaluation scripts.
The evaluation conducted in this paper is three-fold. First, we investigate the effect of overlapping patches in both training and testing stages. Second, we assess the improvement of multi-modality architectures over single-modality ones. Third, we compare the different models on the three considered datasets. Note that, for the sake of simplicity, the network’ dimensionality is shown as subscript (e.g. UResNet denotes the 2D version of the UResNet architecture).
To evaluate the effect of extracting overlapping patches in training and testing, we run all the architectures on the three datasets contemplating three levels: null, medium and high (approximately 0%, 50% and 90%, respectively). On IBSR18 and iSeg2017, the evaluation was carried out using a leave-one-out cross-validation scheme. On MICCAI2012, the process consisted in using the given training and testing sets.
The first test consisted in quantifying improvement between networks trained with either null or high degrees of overlap on training. Resulting -values obtained on the three datasets are depicted in Fig. 4. In all the cases, the model trained with high overlap led to higher DSC values than when not. As it can be observed, in most of the cases ( out of ), the overlapping sampling led to statistically significant higher performance than when omitted. The two groups (convolutional-only and u-shaped) exhibited opposite behaviours. On the one hand, the highest improvements are noted in u-shaped networks. This is related to the fact that non-overlap may mean not enough samples. On the other hand, convolutional-only models evidenced the least increase. Since output patches are smaller, more data can be extracted and passed during training. Therefore, they can provide already accurate results. This fact is illustrated by the results of DM and KK.
The second test consisted in quantifying the improvement of extracting patches using combinations of the three considered degrees of overlap during training and testing. As mentioned previously, results were fused using a majority voting technique. We noted that the general trend is that the difference between results using null and high extends of overlap on prediction is not significant (-values ). Also, IQR remained the same regardless of the method or dataset. Nevertheless, the general trend was an improvement of mean DSC of at least % in the overlapping cases. Another important observation from our experiments is that zero impact or slight degradation of the DSC values was noted when training with null overlap and testing with high overlap. Naturally, this last outcome is a consequence of merging predictions of a poorly trained classifier.
Medium level of overlap patch extraction, in both training and testing, led to improvement w.r.t. null degree cases but yielded lower values than when using a high extent of overlap. That is to say, the general trend is: the more the extent of overlap, the higher the overall performance of the method. The price to pay for using farther levels of overlap is computational time and power since the number of samples to process increases proportionally. For example, given an input volume with dimensions and a network producing output size of , the number of possible patches to be extracted following the null, medium and high overlap policies are , and , respectively.
Since overlapping sampling proved useful, the results showed in following sections correspond to the ones obtained using high overlap in both training and testing.
3.2 Single and multiple modalities
We performed leave-one-out cross-validation on the iSeg2017 dataset using the implemented 2D and 3D architectures to assess the effect of single and numerous imaging sequences on the final segmentation. The results of this experiment are shown in Fig. 5.
As it can be observed, the more the input modalities, the more improved the segmentation. In this case, two modalities not only allowed the network to achieve higher mean but also to reduce the IQR, i.e. results are consistently better. This behaviour was evidenced regardless of architectural design or tissue type. For instance, while the best single modality strategy scored , and for CSF, GM and WM, respectively; its multi-modality analogue yielded , and for the same classes. Additionally, most of the strategies using both T1-w and T2-w obtained DSC values which were statistically higher than their single-modality counterparts ( out of the cases returned -values ).
3.3 Comparison of 2D and 3D FCNN architectures
The eight architectures were evaluated using their best parameters according to the previous sections on three different datasets: IBSR18, MICCAI2012 and iSeg2017. The DSC mean and standard deviation values are shown in Table4.
The networks performing the best on IBSR18, MICCAI2012 and iSeg2017 were UResNet and UNet, UNet, and DM. While for MICCAI2012 and iSeg2017, 3D approaches took the lead, 2D versions performed the best for IBSR18. Taking into account the information in Table 1, 3D architectures appear to be slightly more affected by differences in voxel spacing than 2D ones since the former set obtains similar or lower results than latter group – unlike in the other datasets. One of the reasons explaining this outcome could be the lack of sufficient data allowing 3D networks to understand variations in voxel spacing, i.e. 3D networks may be overfitting to one of the voxel spacing groups. Nevertheless, the increase of the 2D networks w.r.t. the 3D ones is not statistically significant overall.
Segmentation outputs obtained by the different methods on one of the volumes of the IBSR18 dataset are displayed in Fig. 6. Note that architectures using 2D information were trained with axial slices. As it can be seen in the illustration, since 2D architectures process each slice independently, the final segmentation is not necessarily accurate nor consistent: (i) subcortical structures exhibit unexpected shapes and holes, and (ii) sulci and gyri are not segmented finely. Thus, even if segmentation was carried out slice-by-slice, 3D approaches exhibit a smoother segmentation presumably since these methods exploit the 3D volumes directly.
Another thing to note in Fig. 6f is that segmentation provided by KK seems worse than the rest – even than its 2D analogue. The problem does not appear to be related to the number of parameters since KK requires the least amount of parameters in the 3D group, according to Table 3. This issue may be a consequence of the architectural design itself. Anisotropic voxels and heterogeneous spacing may be affecting the low-resolution path of the network considerably. Hence, the overall performance is degraded.
In comparison with state of the art, our methods showed similar or enhanced performance. First, the best DSC scores for IBSR18 were collected by Valverde et al. (2015). The highest values for CSF, GM and WM were , and ; while our best approach scored , and , for the same classes. Second, the best-known values for tissue segmentation using the MICCAI 2012 dataset, were reported by Moeskops et al. (2016). Their strategy – a multi-path CNN – obtained and for CSF and WM, respectively; while our best approach yielded and . In this case, we cannot establish a direct comparison of GM scores since in Moeskops’ case this class was subdivided into (a) cortical GM and (b) basal ganglia and thalami. Third, based on the results displayed in Table 4, our pipeline using DM led to the best segmentation results on the iSeg2017 leave-one-out cross-validation. Hence, we submitted our approach to the online challenge under the team name “nic_vicorob”555Results can be viewed at http://iseg2017.web.unc.edu/rules/results/. The mean DSC values were , and for CSF, GM and WM, correspondingly; and we also ranked top-5 in six of the nine evaluation scenarios (three classes, three measures).
In this paper, we analysed quantitatively eight FCNN architectures inspired by the literature of brain segmentation related tasks. The networks were assessed through three experiments studying the importance of (i) overlapping patch extraction, (ii) multiple modalities, and (iii) network’s dimensionality.
Our first experiment evaluated the impact of overlapping as sampling strategy at training and testing stages. This overlapping sampling is explored as a workaround to the commonly used data augmentation techniques. This procedure can be used in this case as none of these networks processes a whole volume at a time, but patches of it. Based on our results, the technique proved beneficial since most of the networks obtained significantly higher values than when not considered. In particular, the four u-shaped architectures exhibited remarkable influence of this approach, presumably since more samples are used during training and since the same area is seen with different neighbouring regions, enforcing spatial consistency. Overlapping sampling in testing acts as a de-noising technique. We observed that this already-incorporated tool led to better performance than when absent as it helps filling small holes in areas expected homogeneous. The improvement was found to be at least %. Naturally, the main drawback of this technique is the expertise of the classifier itself, since it may produce undesired outputs when poorly trained.
Our second experiment assessed the effect of single and multiple imaging sequences on the final segmentation. We observed that regardless of the segmentation network, the inclusion of various modalities led to significantly better segmentations that in the single modality approach. This situation may be a consequence of networks being able to extract valuable contrast information. Improvements were noted concerning the mean as well as the dispersion of the values yielded by the methods. Although this outcome is aligned with the literature (Zhang et al., 2015), additional trials on more datasets should be carried out to drawn stronger conclusions. Additionally, future work should consider evaluating tissue segmentation in the presence of pathologies and using more imaging sequences such as FLAIR and PD.
Our third experiment evaluated significant differences between 2D and 3D methods on the three considered datasets. In general, 3D architectures produced significantly higher performance than their 2D analogues. However, in one of our datasets, IBSR18, the results were quite similar between the two groups. Since the other two sets were re-sampled to obtain isotropic voxels, this outcome seems to be a consequence of the heterogeneity of the data in IBSR18, i.e. 2D methods seem to be more resilient to this kind of issues than 3D ones.
Regarding network design, we observed that networks using information from shallower layers in deeper ones achieved higher performance than those using features directly from the input volume. Note the difference is intensified in heterogeneous datasets, IBSR18, where the latter strategy performs worse on average. Although it is only two networks, KK and KK
, this situation may underline the importance of internal connections (e.g. skip connections, residual connections) and fusion of multi-resolution information to segment more accurately. No remarkable difference was seen between convolutional-only and u-shaped architectures, except for processing times. In both training and testing, u-shaped networks produce segmentation faster than convolutional-only networks: u-shaped models require extracting less number of patches and provide a more prominent output at a time. For instance, in testing time and using a high degree of overlap, UNetcan process a volume of size in around seconds while DM can take up to seconds.
Regarding general performance, two methods, DM and UNet, displayed the best results. It is important to remark that our specific implementation of the latter architecture (i) required % fewer parameters to be set than the former, and (ii) classifies K voxels more at a time which makes it appealing regarding computational speed. If the priority is time (training and testing), UResNet is a suitable option since it produces the same output size as the UNet, but the number of parameters is approximately half. Therefore, patch-based u-shaped networks are recommended to address tissue segmentation compared to convolutional-only approaches.
Taking into account results reported in the literature, we achieved top performance for IBSR18, MICCAI2012 and iSeg2017 with our implemented architectures. Three relevant things to note in this work. First, none of these networks has explicitly been tweaked to the scenarios, a typical pipeline has been used. Hence, it is possible to compare them under similar conditions. Approaches expressly tuned for challenges may win, but it does not imply they will work identically – using the same set-up – on real-life scenarios. Second, although these strategies have shown acceptable results, more development on domain adaptation and transfer learning (zero-shot or one-shot training) should be carried out to implement them in medical centres. Third, we do not intend to compare original works. Our implementations are inspired by the original works, but general pipelines are not taken into account in here. In short, our study focus on understanding architectural strengths and weaknesses of literature-like approaches.
In this paper, we have analysed quantitatively FCNN architectures, 2D and 3D, for tissue segmentation on brain MRI. These networks were implemented inspired by four recently proposed networks (Çiçek et al., 2016; Dolz et al., 2017; Guerrero et al., 2017; Kamnitsas et al., 2017). Among other characteristics, these methods comprised (i) convolutional-only and u-shaped architectures, (ii) single- and multi-modality inputs, (iii) 2D and 3D network dimensionality, (iv) varied implementation of multi-path schemes, and (v) different number of parameters.
The eight networks were tested using three different well-known datasets: IBSR18, MICCAI2012 and iSeg2017. These datasets were considered since they were acquired with different configuration parameters. Testing scenarios evaluated the impact of (i) overlapping sampling on both training and testing, (ii) multiple modalities, and (iii) 2D and 3D inputs on tissue segmentation. First, we observed that extracting patches with a certain degree of overlap among themselves led consistently to higher performance. The same approach on testing did not show a relevant improvement (around % in DSC). It is a de-noising tool that comes along with the trained network. Second, we noted that using multiple modalities – when available – can help the method to achieve significantly higher accuracy values. Third, based on our evaluation for tissue segmentation, 3D methods tend to outperform their 2D counterpart. However, it is relevant to recognise that 3D methods appear slightly more affected to variations in voxel spacing. Additionally, networks implemented in this paper were able to deliver state-of-the-art results on IBSR18 and MICCAI2012. Our best approach on the iSeg2017 ranked top-5 in most of the on-line testing scenarios.
To encourage other researchers to use the implemented evaluation framework and FCNN architectures, we have released a public version of it at our research website.
Jose Bernal and Kaisar Kushibar hold FI-DGR2017 grants from the Catalan Government with reference numbers 2017FI B00476 and 2017FI B00372 respectively. Mariano Cabezas holds a Juan de la Cierva - Incorporación grant from the Spanish Government with reference number IJCI-2016-29240. This work has been partially supported by La Fundació la Marató de TV3, by Retos de Investigació TIN2014-55710-R, TIN2015-73563-JIN and DPI2017-86696-R from the Ministerio de Ciencia y Tecnología. The authors gratefully acknowledge the support of the NVIDIA Corporation with their donation of the TITAN-X PASCAL GPU used in this research. The authors would like to thank the organisers of the iSeg2017 challenge for providing the data.
- Ashburner et al. (2012) Ashburner, J., Barnes, G., & Chen, C. (2012). SPM8 Manual. http://www.fil.ion.ucl.ac.uk/. [Online; accessed 18 May 2017].
- Ashburner & Friston (2005) Ashburner, J., & Friston, K. J. (2005). Unified segmentation. NeuroImage, 26, 839 – 851.
- Bernal et al. (2017) Bernal, J., Kushibar, K., Asfaw, D. S., Valverde, S., Oliver, A., Martí, R., & Lladó, X. (2017). Deep convolutional neural networks for brain image analysis on magnetic resonance imaging: a review. CoRR, abs/1712.03747. arXiv:1712.03747.
- Birenbaum & Greenspan (2016) Birenbaum, A., & Greenspan, H. (2016). Longitudinal multiple sclerosis lesion segmentation using multi-view convolutional neural networks. In International Workshop on Large-Scale Annotation of Biomedical Data and Expert Label Synthesis (pp. 58--67). Springer.
- Brosch et al. (2016) Brosch, T., Tang, L. Y., Yoo, Y., Li, D. K., Traboulsee, A., & Tam, R. (2016). Deep 3D convolutional encoder networks with shortcuts for multiscale feature integration applied to multiple sclerosis lesion segmentation. IEEE Transactions on Medical Imaging, 35, 1229--1239.
- Cardoso et al. (2013) Cardoso, M. J., Melbourne, A., Kendall, G. S., Modat, M., Robertson, N. J., Marlow, N., & Ourselin, S. (2013). AdaPT: an adaptive preterm segmentation algorithm for neonatal brain MRI. NeuroImage, 65, 97--108.
- Chen et al. (2016) Chen, H., Dou, Q., Yu, L., Qin, J., & Heng, P.-A. (2016). VoxResNet: Deep voxelwise residual networks for brain segmentation from 3D MR images. CoRR, abs/1608.05895. arXiv:1608.05895.
- Çiçek et al. (2016) Çiçek, Ö., Abdulkadir, A., Lienkamp, S. S., Brox, T., & Ronneberger, O. (2016). 3D u-net: learning dense volumetric segmentation from sparse annotation. In International Conference on Medical Image Computing and Computer-Assisted Intervention (pp. 424--432). Springer.
- Clarke et al. (1995) Clarke, L., Velthuizen, R., Camacho, M., Heine, J., Vaidyanathan, M., Hall, L., Thatcher, R., & Silbiger, M. (1995). MRI segmentation: methods and applications. Magnetic resonance imaging, 13, 343--368.
- Crum et al. (2006) Crum, W. R., Camara, O., & Hill, D. L. G. (2006). Generalized overlap measures for evaluation and validation in medical image analysis. IEEE Transactions on Medical Imaging, 25, 1451--1461.
- Dice (1945) Dice, L. R. (1945). Measures of the amount of ecologic association between species. Ecology, 26, 297--302.
- Dolz et al. (2017) Dolz, J., Desrosiers, C., & Ayed, I. B. (2017). 3D fully convolutional networks for subcortical segmentation in MRI: A large-scale study. NeuroImage, .
- Filippi et al. (2016) Filippi, M., Rocca, M. A., Ciccarelli, O., De Stefano, N., Evangelou, N., Kappos, L., Rovira, A., Sastre-Garriga, J., Tintorè, M., Frederiksen, J. L., Gasperini, C., Palace, J., Reich, D. S., Banwell, B., Montalban, X., & Barkhof, F. (2016). MRI criteria for the diagnosis of multiple sclerosis: MAGNIMS consensus guidelines. The Lancet Neurology, 15, 292 -- 303.
- Ghafoorian et al. (2017) Ghafoorian, M., Karssemeijer, N., Heskes, T., van Uden, I., Sanchez, C., Litjens, G., de Leeuw, F.-E., van Ginneken, B., Marchiori, E., & Platel, B. (2017). Location sensitive deep convolutional neural networks for segmentation of white matter hyperintensities. Nature Scientific Reports, 7.
- Guerrero et al. (2017) Guerrero, R., Qin, C., Oktay, O., Bowles, C., Chen, L., Joules, R., Wolz, R., Valdes-Hernandez, M., Dickie, D., Wardlaw, J. et al. (2017). White matter hyperintensity and stroke lesion segmentation and differentiation using convolutional neural networks. CoRR, abs/1706.00935. arXiv:1706.00935.
Hariharan et al. (2015)
Hariharan, B., Arbeláez, P.,
Girshick, R., & Malik, J.
Hypercolumns for object segmentation and fine-grained
Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition(pp. 447--456).
- Kamnitsas et al. (2017) Kamnitsas, K., Ledig, C., Newcombe, V. F., Simpson, J. P., Kane, A. D., Menon, D. K., Rueckert, D., & Glocker, B. (2017). Efficient multi-scale 3D CNN with fully connected CRF for accurate brain lesion segmentation. Medical Image Analysis, 36, 61--78.
- Kapur et al. (1996) Kapur, T., Grimson, W. E. L., Wells, W. M., & Kikinis, R. (1996). Segmentation of brain tissue from magnetic resonance images. Medical Image Analysis, 1, 109--127.
- Kleesiek et al. (2016) Kleesiek, J., Urban, G., Hubert, A., Schwarz, D., Maier-Hein, K., Bendszus, M., & Biller, A. (2016). Deep MRI brain extraction: a 3D convolutional neural network for skull stripping. NeuroImage, 129, 460--469.
- Kushibar et al. (2017) Kushibar, K., Valverde, S., González-Villà, S., Bernal, J., Cabezas, M., Oliver, A., & Lladó, X. (2017). Automated sub-cortical brain structure segmentation combining spatial and deep convolutional features. CoRR, abs/1709.09075. arXiv:1709.09075.
- Liew & Yan (2006) Liew, A. W.-C., & Yan, H. (2006). Current methods in the automatic tissue segmentation of 3D magnetic resonance brain images. Current medical imaging reviews, 2, 91--103.
- Litjens et al. (2017) Litjens, G., Kooi, T., Bejnordi, B. E., Setio, A. A. A., Ciompi, F., Ghafoorian, M., van der Laak, J. A., van Ginneken, B., & Sánchez, C. I. (2017). A survey on deep learning in medical image analysis. CoRR, abs/1702.05747. arXiv:1702.05747.
- Long et al. (2015) Long, J., Shelhamer, E., & Darrell, T. (2015). Fully convolutional networks for semantic segmentation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (pp. 3431--3440).
- Lyksborg et al. (2015) Lyksborg, M., Puonti, O., Agn, M., & Larsen, R. (2015). An ensemble of 2D convolutional neural networks for tumor segmentation. In Scandinavian Conference on Image Analysis (pp. 201--211). Springer.
- Mendrik et al. (2015) Mendrik, A. M., Vincken, K. L., Kuijf, H. J., Breeuwer, M., Bouvy, W. H., De Bresser, J., Alansary, A., De Bruijne, M., Carass, A., El-Baz, A. et al. (2015). MRBrainS challenge: online evaluation framework for brain image segmentation in 3T MRI scans. Computational intelligence and neuroscience, 2015, 1.
- Milletari et al. (2016) Milletari, F., Navab, N., & Ahmadi, S.-A. (2016). V-net: Fully convolutional neural networks for volumetric medical image segmentation. In on 3D Vision (3DV), 2016 Fourth International Conference (pp. 565--571).
- Moeskops et al. (2017) Moeskops, P., Veta, M., Lafarge, M. W., Eppenhof, K. A., & Pluim, J. P. (2017). Adversarial training and dilated convolutions for brain MRI segmentation. In Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support (pp. 56--64). Springer.
- Moeskops et al. (2016) Moeskops, P., Viergever, M. A., Mendrik, A. M., de Vries, L. S., Benders, M. J., & Išgum, I. (2016). Automatic segmentation of MR brain images with a convolutional neural network. IEEE Transactions on Medical Imaging, 35, 1252--1261.
- Nie et al. (2016) Nie, D., Wang, L., Gao, Y., & Sken, D. (2016). Fully convolutional networks for multi-modality isointense infant brain image segmentation. In Biomedical Imaging (ISBI), 2016 IEEE 13th International Symposium (pp. 1342--1345).
- Pham (2001) Pham, D. L. (2001). Robust fuzzy segmentation of magnetic resonance images. In Proceedings 14th IEEE Symposium on Computer-Based Medical Systems. CBMS 2001 (pp. 127--131).
- Ronneberger et al. (2015) Ronneberger, O., Fischer, P., & Brox, T. (2015). U-net: Convolutional networks for biomedical image segmentation. In International Conference on Medical Image Computing and Computer-Assisted Intervention (pp. 234--241). Springer.
- Rovira et al. (2015) Rovira, À., Wattjes, M. P., Tintoré, M., Tur, C., Yousry, T. A., Sormani, M. P., De Stefano, N., Filippi, M., Auger, C., Rocca, M. A. et al. (2015). Evidence-based guidelines: MAGNIMS consensus guidelines on the use of MRI in multiple sclerosis - clinical implementation in the diagnostic process. Nature Reviews Neurology, 11, 471--482.
- Shakeri et al. (2016) Shakeri, M., Tsogkas, S., Ferrante, E., Lippe, S., Kadoury, S., Paragios, N., & Kokkinos, I. (2016). Sub-cortical brain structure segmentation using F-CNN’s. In Biomedical Imaging (ISBI), 2016 IEEE 13th International Symposium (pp. 269--272).
- Shattuck et al. (2001) Shattuck, D. W., Sandor-Leahy, S. R., Schaper, K. A., Rottenberg, D. A., & Leahy, R. M. (2001). Magnetic resonance image tissue classification using a partial volume model. NeuroImage, 13, 856--876.
- Steenwijk et al. (2016) Steenwijk, M. D., Geurts, J. J. G., Daams, M., Tijms, B. M., Wink, A. M., Balk, L. J., Tewarie, P. K., Uitdehaag, B. M. J., Barkhof, F., Vrenken, H. et al. (2016). Cortical atrophy patterns in multiple sclerosis are non-random and clinically relevant. Brain, 139, 115--126.
- Stollenga et al. (2015) Stollenga, M. F., Byeon, W., Liwicki, M., & Schmidhuber, J. (2015). Parallel Multi-dimensional LSTM, with Application to Fast Biomedical Volumetric Image Segmentation. In Proceedings of the 28th International Conference on Neural Information Processing Systems NIPS’15 (pp. 2998--3006). Cambridge, MA, USA: MIT Press.
- Valverde et al. (2015) Valverde, S., Oliver, A., Cabezas, M., Roura, E., & Lladó, X. (2015). Comparison of 10 brain tissue segmentation methods using revisited IBSR annotations. Journal of Magnetic Resonance Imaging, 41, 93--101.
- Valverde et al. (2017) Valverde, S., Oliver, A., Roura, E., González-Villà, S., Pareto, D., Vilanova, J. C., Ramió-Torrentà, L., Rovira, À., & Lladó, X. (2017). Automated tissue segmentation of MR brain images in the presence of white matter lesions. Medical Image Analysis, 35, 446--457.
- Zhang et al. (2015) Zhang, W., Li, R., Deng, H., Wang, L., Lin, W., Ji, S., & Shen, D. (2015). Deep convolutional neural networks for multi-modality isointense infant brain image segmentation. NeuroImage, 108, 214 -- 224.
Zhang et al. (2001)
Zhang, Y., Brady, M., &
Smith, S. (2001).
Segmentation of brain MR images through a hidden Markov random field model and the expectation-maximization algorithm.IEEE Transactions on Medical Imaging, 20, 45--57.