A Multiorgan Segmentation Model for CT Volumes via Full Convolution-Deconvolution Network
We propose a model with two-stage process for abdominal segmentation on CT volumes. First, in order to capture the details of organs, a full convolution-deconvolution network (FCN-DecNet) is constructed with multiple new unpooling, deconvolutional, and fusion layers. Then, we optimize the coarse segmentation results of FCN-DecNet by multiscale weights probabilistic atlas (MS-PA), which uses spatial and intensity characteristic of atlases. Our coarse-fine model takes advantage of intersubject variability, spatial location, and gray information of CT volumes to minimize the error of segmentation. Finally, using our model, we extract liver, spleen, and kidney with Dice index of 90.1 ± 1%, 89.0 ± 1.6%, and 89.0 ± 1.3%, respectively.
Precision medicine was firstly raised in 2011 and has continued developing the new medical model: biologically informed therapies  and computer-aided diagnosis. As we know, precise organ segmentation relies on computer-aided diagnosis necessarily. Due to the fact that segmenting organ manually is time-consuming, material-consuming, and labor-intensive, many methods are proposed for the auto-segmentation of individual organs in previous work [2–7]. However, when it comes to multiorgan segmentation, the problems left to address are not only the variability of the shape and position of abdominal organs, but also the complicated interrelations among the organs, which make segmentation become a challenge.
In order to segment multiple organs accurately, several approaches have been proposed. Okada et al. analyzed organ correlations and prior information of shapes to improve accuracy of multiorgan segmentation . Shimizu et al. used an abdominal cavity standardization process and atlas guided segmentation with parameters estimated by EM algorithm . Oda et al. selected similar atlas as the input image for segmentation and refine the results based on graph cut . Chu et al. proposed an automated multiorgan segmentation method based on a scale hierarchical probabilistic atlas . However, because of the variety of organs among patients, the important sophisticated information of special atlas may be ignored in the above methods.
Recently, Convolutional Neural Network (CNN) proves itself at visual recognition [12–14]. Simonyan and Zisserman increased depth of networks, which showed a significant improvement for image recognition . Additionally, CNN can generate a good deal of valuable consequence in semantic segmentations [16–19]. Hong et al. raised a novel decoupled architecture with heterogeneous annotations to classification and segmentation networks separately . Recent approaches in CT images segmentation are mainly driven by CNN. Roth et al. used multilevel deep convolutional networks for pancreas segmentation . Zhen et al. employed a structure including multiscale deep networks and random forests for direct estimation of cardiac ventricular volumes . However, these works segmented organs simply based on local features, each layer may lose information (especially the location of the organs) after convoluting, and the size of input images is limited as well. Fully convolutional network (FCN) converts an existing CNN architecture by changing all fully connected layers to convolutions and adding deconvolutional layers in network [23–26]. Compared to the previous CNNs researches, they accept any size of images as input and the labels are predicted in pixel level. Although these networks can extract the overall shape of objects, intensity and spatial features are ignored when segmenting organs.
Therefore, in order to take into account variability information and deeper correlations of abdominal organs, we employ a new model for multiorgan segmentation. Our contributions are as follows:(1)We propose a full convolution-deconvolution network (FCN-DecNet) with new architecture which includes deconvolutional layer, new unpooling layer, and fusion layer. So, in multiorgan segmentation process, our network can extract different scales of shape details and record location information within the feature maps.(2)To overcome the lack of spatial information and gray values of the segmentation regions in FCN-DecNet, we construct a multiscale weights probabilistic atlas (MS-PA), which integrates atlas information as well as rough segmentation results from FCN-DecNet.(3)We propose a method to optimize FCN-DecNet by combining segmentation results of FCN-DecNet with MS-PA and take the ensemble as the final segmentation results.
The rest of this paper is organized as follows. In Section 2, we describe our new multiorgan segmentation model with the details of the FCN-DecNet and MS-PA, respectively. To verify our method, experimental results are compared and a general discussion regarding our approach is done in Section 3. Lastly, conclusions are summarized in Section 4.
In this paper, we construct a network for multiorgan segmentation, further optimized with MS-PA. A brief summary of the basic flow is described in Figure 1. Before the whole segmentation process, we divide CT volumes into sequence slices as the input of FCN-DecNet. In FCN-DecNet segmentation process, a network with multiple layers is trained. Probability of each pixel is obtained from output and taken as the rough segmentation results. In optimization process, we first take CT volumes and rough segmentation results from FCN-DecNet as the input of MS-PA to generate a multiscale weights probabilistic atlas. Then, probability of each voxel is calculated through Bayes-based estimation and used to refine rough segmentation results; in this way, final segmentation is gotten.
2.2. FCN-DecNet Network for Segmentation
Our network is particularly composed of five parts: convolutional layer, pooling layer, deconvolutional layer, new unpooling layer, and fusion layer. Figure 2 shows the architecture of FCN-DecNet in this paper. The network is initialized from the VGG-16 net , we discard the final classifier layer and convert all fully connected layers to convolutional layers, and then we append a convolution with channel dimension 4 to predict scores of liver, spleen, kidney, and background after pool 3, pool 4, pool 5, and fc7. Since unpooling and deconvolution have different contributions to reconstructing the original size of segmentation regions, unpooling records the original locations; deconvolution tends to capture detail shapes. 4 rounds of unpooling and deconvolution in FCN-DecNet are followed for upsampling the coarse outputs. Based on the enlarged coarse outputs in first round, the second and third rounds reconstruct the location and shape of the organs. Meanwhile, from the first to the third round, the outputs are fused with the corresponding prediction scores at pool 3, pool 4, and pool 5. Considering the little improvement for the even lower layer fusion, in the fourth round, we directly deconvolve and get the output probability with 8 pixels’ stride at final layer.
2.2.2. New Unpooling
Unpooling is an approximate inverse of max pooling operation by recording the value and location of the maximum activation during pooling and filling maximum value back as well as setting other activation to zero within each pooling region. However, zero values make difference becoming large between max activation and other activation within each pooling region, which would increase intersubject variance in local segmentation regions. To deal with this issue, our new unpooling fills values at other activation after comparing maximum activation value with mean value of the whole map instead of just filling zero. Therefore, the generated segmentation organs from the above layer are placed into the appropriate locations and the shapes of organs in this layer are also preserved at the same time. The mean value of the whole map is calculated when yielding pooled maps and filled in vacancy during unpooling. The mathematical form is defined aswhere is filling function; , denote mean value of the whole map and maximum activation value during pooling, respectively.
The detail of unpooling is illustrated in Figure 3, where . The output of an unpooled map is first enlarged to after filling maximum activation values , , , in the same location. Then, for each of the maximum activation values, if they are larger than , other activation within corresponding pooling region is filled with ; otherwise it is filled with 0. In this way, is changed to and taken as the final output of the new unpooling. Meanwhile, deconvolution operation is employed to densify because vacancy still exists sparsely when .
2.2.3. Analysis of FCN-DecNet
(A) Net Visualization. Net visualization represents the input pattern that stimulates the given feature map in the model accurately . We can judge the reasonableness of design structure and understand specific roles of deconvolution and unpooling processes by observing corresponding activation maps, as shown in Figure 4. From round 1 to round 4, coarse to fine segmentation regions are reconstructed through unpooling and deconvolution. Unpooling reconstructs the rough shapes of the organs at appropriate location. Deconvolution learns multiple filters to densify segmentation regions obtained from the previous unpooling layer. Furthermore, through a series of fuse operations, class-specific patterns are found from lower layers to higher layers.
(B) Net Comparison. There are three kinds of output from FCN fusing information from layers and upsampling with different strides: FCN-8s, FCN-16s, and FCN-32s. And FCN-8s net shows the highest performance among them . So we make comparison with FCN-8s and get segmentation results in Figure 5. Discovering our net not only fills vacancy area of FCN-8s but also improves the accuracy when reconstructing the boundary of organ. However, small scattered regions are still not belonging to organs. So the MS-PA is applied to FCN-DecNet for more precise results.
2.3. FCN-DecNet Optimization
2.3.1. MS-PA Construction
Our method performs segmentation of three abdominal organs (liver, spleen, and kidney) in target CT volume. Before the optimization, MS-PA is first constructed to obtain organ regions, which provides a priori information about the intensity and spatial information of the target CT volume. All atlases are aligned by nonlinear registration (affine transformation and B spline transformation) to reduce location and shape variation of organs among different patients. The label spaces of organs are manually marked by doctors, denoted as (represent background, liver, spleen, and kidney, resp.).
In order to segment different organs, more comprehensive information is needed. We give a hierarchical weight model, including three scales (image-wise, organ-wise, and voxel-wise). Here, we take the similarity between reference atlases in ( is the number of atlases; is one of the atlases in ) and target CT volume as image-wise weight and the similarity between organs in and as organ-wise weight . Then, in each voxel at the same position between and , we calculate the similarity of region composed of 242 adjacent voxels ( in previous slice, in next slice, and in current slice) as voxel-wise weight . is used to select the most similar atlas in after registration, so we use Pearson correlation coefficient to calculate. Meanwhile, we divide into two parts: one part is to compute rough overlapping proportions between organs in and ; the other part is to further rectify inaccurate places. Therefore, Jaccard index and Pearson correlation coefficient are used to calculate, respectively. As for , we use normalized Euclid distance directly to calculate intensity relevance among voxels. They are successively defined bywhere and are the organ regions in and corresponding positions in ; particularly, is obtained from the rough segmentations of FCN-DecNet; is the region in with same position in . and are adjacent regions of voxel with same position in and . , , and indicate Pearson correlation coefficient, Jaccard index, and Euclid distance, respectively, and are defined bywhere is the covariance of and . and are the variances of and . And the probability of each voxel belonging to organ is calculated by
The voxel level probability result of MS-PA is obtained through Bayes-based estimation that is defined bywhere intensity distribution of each organ is a normal with parameters estimated by EM algorithm, forming the Gaussian mixture distribution of the whole atlases. Based on the joint probability of all organs, is calculated.
Figure 6 shows the segmentation results from MS-PA, which highly depend on the spatial information and gray values of the organs. However, since the limitation number of atlases in , irrelevant voxels especially around the boundary of organs may be classified in error.
2.3.2. Optimization through MS-PA
Our FCN-DecNet obtains finer segmentation results than FCN-8s. Yet, inaccurate segmentation regions still appeared. So in optimization step of FCN-DecNet, a Bayes-based estimation through MS-PA is deployed to refine the rough segmentation. All pixel level probabilities of FCN-DecNet in one volume are first stacked into voxel level probabilities. Then we calculate the mean value of both voxel probabilities in FCN-DecNet and MS-PA. In this way, the best matching label is estimated by maximum mean voxel probabilities. By the amelioration of MS-PA, our model considers high-scale shapes and locations information of organs in FCN-DecNet as well as the spatial information and gray values of the segmentation regions in optimization step.
3. Results and Discussion
We evaluate our segmentation algorithm with 3D abdominal CT volumes of 12 patients, and each volume contains altogether about 70 abdominal slices from the same CT scanner. To offset insufficient data and select available model, 4-fold cross validation is applied with the data randomly split into training and testing sets. All the organ boundaries are manually defined by us and approved by a doctor. CT volumes are divided into slices with fixed-size 512 × 512 for training FCN-DecNet, and then volumes themselves are taken as input in MS-PA construction as well. In addition, we reconstruct organs using the sequence of segmentation results. Considering the best results from the FCN-8s architecture in , Figure 7 gives the segmentation results from FCN-8s, FCN-DecNet, and FCN-DecNet + MS-PA. Compared to FCN-8s, the vacancy in organs is filled and the boundary is smoothed through FCN-DecNet, which not only builds overall shape of an object, but also corrects wrong segmentation results from FCN-8s. Moreover, the training time is decreasing based on our efficient net. Although our results are better than FCN-8s on account of holding the shapes and locations of the organ, inaccurate segmentations still exist. So we incorporate with spatial information as well as gray values and can find that the false segmentation regions are eliminated and the overshapes of reconstructed organs are close to ground truth through FCN-DecNet + MS-PA in Figure 7. Figure 8 shows 3D view of the segmentation results compared with ground truth.
Then, we make comparison between FCN-DecNet and FCN-8s in Table 1 via indicators defined in . We see that the pixel and mean IU accuracy of FCN-DecNet seem close to FCN-8s, but mean accuracy is improved substantially in FCN-DecNet.
In order to further evaluate our method, Table 2 reports maximum segmentation accuracies in 4-fold cross validation using Dice similarity index [29, 30], Precision, and Recall rate, defined bywhere , , and denote the number of voxels relevantly classified, the number of voxels irrelevantly classified, and the number of voxels in ground truth that is ignored. Accuracies are rising via cross validation. This further explains that FCN-DecNet + MS-PA may be highly generalizable in cross-dataset assessment.
We use the Dice similarity index which measures the volume overlap between the ground truth and segmentation results. In comparison with state-of-the-art methods as reported in Table 3, our method performs better. Specifically, the segmentation accuracy is improved compared to Shimizu et al. , Liu et al. , and Okada et al.  with similar training samples. For others, our maximum segment Dice values of organs are close to Chu et al.  and Wolz et al.  but lower than them in general. This is because our cases are much less than Chu et al.  and Wolz et al. , which approximately equal 100 cases. What is more, data variation in nonlinear registration may bring influence to the effectiveness of the results. For instance, the size of abdominal cavity of some patients is too big or too small.
This paper proposes a new coarse-fine model for multiorgan abdominal segmentation (liver, spleen, and kidney). A new fully convolutional network (FCN-DecNet) is trained and refined in optimization step. For our FCN-DecNet, new unpooling layer records locations of organs, deconvolutional layer reconstructs detailed shapes of organ, and fusion layer is employed to make local predictions which make use of correlations among regional pixels. The learning process of FCN-DecNet is shown via visualizing the corresponding activation maps. For optimization step, three scales (image-wise, organ-wise, and voxel-wise) are considered in construction with multiscale weights probabilistic atlas (MS-PA), and Bayes-based estimation through MS-PA is employed to improve the coarse segmentation results of FCN-DecNet further. The proposed method shows better segmentation performance compared to other methods with similar number of cases. Further perfection will be done in future work, including evaluating more organs and using large number of cases to train our model.
Conflicts of Interest
The authors declare that they have no conflicts of interest.
This work was supported by the National Natural Science Foundation of China (no. 61472073).
H. Ling, S. K. Zhou, Y. Zheng, B. Georgescu, M. Suehling, and D. Comaniciu, “Hierarchical, learning-based automatic liver segmentation,” in Proceedings of the 26th IEEE Conference on Computer Vision and Pattern Recognition (CVPR '08), pp. 1–8, Anchorage, Alaska, USA, June 2008.View at: Publisher Site | Google Scholar
A. M. Ali, A. A. Farag, and A. S. El-Baz, “Graph cuts framework for kidney segmentation with prior shape constraints,” in Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, pp. 384–392, Springer, Berlin, Germany, 2007.View at: Google Scholar
Z. Xu, B. Li, S. Panda et al., “Shape-constrained multi-atlas segmentation of spleen,” in Proceedings of the International Society for Optics and Photonics, SPIE Medical Imaging, 2014.View at: Google Scholar
M. Erdt, M. Kirschner, K. Drechsler, S. Wesarg, M. Hammon, and A. Cavallaro, “Automatic pancreas segmentation in contrast enhanced CT data using learned spatial anatomy and texture descriptors,” in Proceedings of the 2011 8th IEEE International Symposium on Biomedical Imaging: From Nano to Macro, ISBI'11, pp. 2076–2082, Chicago, IL, USA, April 2011.View at: Publisher Site | Google Scholar
T. Okada, M. G. Linguraru, Y. Yoshida et al., “Abdominal multi-organ segmentation of CT images based on hierarchical spatial modeling of organ interrelations,” Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), vol. 7029, pp. 173–180, 2012.View at: Publisher Site | Google Scholar
M. Oda, T. Nakaoka, T. Kitasaka et al., “Organ segmentation from 3D abdominal CT images based on atlas selection and graph cut,” Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), vol. 7029, pp. 181–188, 2012.View at: Publisher Site | Google Scholar
C. Chu, M. Oda, T. Kitasaka et al., “Multi-organ segmentation based on spatially-divided probabilistic atlas from 3D abdominal CT images,” in Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, pp. 165–172, Springer, Berlin, Germany, 2013.View at: Google Scholar
A. Krizhevsky, I. Sutskever, and E. Hinton G, “Imagenet classification with deep convolutional neural networks,” in Proceedings of the Advances in Neural Information Processing Systems, pp. 1097–1105.View at: Google Scholar
J. Donahue, Y. Jia, O. Vinyals et al., “DeCAF: a deep convolutional activation feature for generic visual recognition,” in Proceedings of the ICML, pp. 647–655, 2014.View at: Google Scholar
A. Karpathy, G. Toderici, S. Shetty, T. Leung, R. Sukthankar, and F.-F. Li, “Large-scale video classification with convolutional neural networks,” in Proceedings of the 27th IEEE Conference on Computer Vision and Pattern Recognition, (CVPR '14), pp. 1725–1732, Columbus, OH, USA, June 2014.View at: Publisher Site | Google Scholar
D. Eigen and R. Fergus, “Predicting depth, surface normals and semantic labels with a common multi-scale convolutional architecture,” in Proceedings of the 15th IEEE International Conference on Computer Vision, ICCV 2015, pp. 2650–2658, Santiago, Chile, December 2015.View at: Publisher Site | Google Scholar
S. Hong, H. Noh, and B. Han, “Decoupled deep neural network for semi-supervised semantic segmentation,” in Proceedings of the Advances in Neural Information Processing Systems, pp. 1495–1503, 2015.View at: Google Scholar
H. R. Roth, L. Lu, A. Farag et al., “Deeporgan: Multi-level deep convolutional networks for automated pancreas segmentation,” Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), vol. 9349, pp. 556–564, 2015.View at: Publisher Site | Google Scholar
M. D. Zeiler and R. Fergus, “Visualizing and understanding convolutional networks,” in Computer Vision—ECCV 2014: 13th European Conference, Zurich, Switzerland, September 6–12, 2014, Proceedings, Part I, vol. 8689 of Lecture Notes in Computer Science, pp. 818–833, Springer, 2014.View at: Publisher Site | Google Scholar
X. Liu, M. G. Linguraru, J. Yao, and R. M. Summers, “Organ pose distribution model and an MAP framework for automated abdominal multi-organ localization,” Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), vol. 6326, pp. 393–402, 2010.View at: Publisher Site | Google Scholar
R. Wolz, C. Chu, K. Misawa et al., “Multi-organ abdominal CT segmentation using hierarchically weighted subject-specific atlases,” in Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, pp. 10–17, Springer, Berlin, Germany, 2012.View at: Google Scholar
L. R. Dice, “Measures of the amount of ecologic association between species,” Ecology, vol. 26, no. 3, pp. 297–302, 1945.View at: Google Scholar
S. Pereira, A. Pinto, V. Alves et al., “Deep convolutional neural networks for the segmentation of gliomas in multi-sequence MRI,” in Proceedings of the International Workshop on Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries, pp. 131–143, Springer International Publishing, 2015.View at: Google Scholar