Deep Small Bowel Segmentation with Cylindrical Topological Constraints
- 5k Downloads
We present a novel method for small bowel segmentation where a cylindrical topological constraint based on persistent homology is applied. To address the touching issue which could break the applied constraint, we propose to augment a network with an additional branch to predict an inner cylinder of the small bowel. Since the inner cylinder is free of the touching issue, a cylindrical shape constraint applied on this augmented branch guides the network to generate a topologically correct segmentation. For strict evaluation, we achieved an abdominal computed tomography dataset with dense segmentation ground-truths. The proposed method showed clear improvements in terms of four different metrics compared to the baseline method, and also showed the statistical significance from a paired t-test.
KeywordsSmall bowel segmentation Topological constraint Persistent homology Inner cylinder Abdominal computed tomography
This research was supported by the National Institutes of Health, Clinical Center and National Cancer Institute.
- 1.Small bowel obstruction (2019). https://my.clevelandclinic.org/health/diseases/15850-small-bowel-obstruction
- 4.Çiçek, Ö., Abdulkadir, A., Lienkamp, S.S., Brox, T., Ronneberger, O.: 3D U-Net: learning dense volumetric segmentation from sparse annotation. In: Ourselin, S., Joskowicz, L., Sabuncu, M.R., Unal, G., Wells, W. (eds.) MICCAI 2016. LNCS, vol. 9901, pp. 424–432. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-46723-8_49CrossRefGoogle Scholar
- 5.Clough, J.R., Oksuz, I., Byrne, N., Zimmer, V.A., Schnabel, J.A., King, A.P.: A topological loss function for deep-learning based image segmentation using persistent homology. arXiv preprint arXiv:1910.01877 (2019)
- 7.Gabrielsson, R.B., Nelson, B.J., Dwaraknath, A., Skraba, P., Guibas, L.J., Carlsson, G.E.: A topology layer for machine learning. arXiv preprint arXiv:1905.12200 (2019)
- 8.Hu, X., Li, F., Samaras, D., Chen, C.: Topology-preserving deep image segmentation. Adv. Neural Inf. Process. Syst. 32, 5657–5668 (2019)Google Scholar
- 9.Loshchilov, I., Hutter, F.: Decoupled weight decay regularization. In: International Conference on Learning Representations (2019)Google Scholar
- 11.Oda, H., et al.: Visualizing intestines for diagnostic assistance of ileus based on intestinal region segmentation from 3D CT images. arXiv preprint arXiv:2003.01290 (2020)
- 14.Sudre, C.H., Li, W., Vercauteren, T., Ourselin, S., Jorge Cardoso, M.: Generalised dice overlap as a deep learning loss function for highly unbalanced segmentations. In: Cardoso, M.J., et al. (eds.) DLMIA/ML-CDS -2017. LNCS, vol. 10553, pp. 240–248. Springer, Cham (2017). https://doi.org/10.1007/978-3-319-67558-9_28CrossRefGoogle Scholar
- 15.Wu, Y., He, K.: Group normalization. arXiv preprint arXiv:1803.08494 (2018)
- 16.Zhang, W., et al.: Mesenteric vasculature-guided small bowel segmentation on 3-D CT. IEEE Trans. Med. Imaging 32(11), 2006–2021 (2013). https://doi.org/10.1109/TMI.2013.2271487