Abstract
Segmentation of the kidney and kidney tumors using computed tomography (CT) is a crucial step in related surgical procedures. Although many deep learning models have been constructed to solve this problem, most of them ignore the boundary information. In this paper, we propose a boundary-aware network (BA-Net) for kidney and kidney tumor segmentation. This model consists of a shared 3D encoder, a 3D boundary decoder, and a 3D segmentation decoder. In contrast to existing boundary-involved methods, we first introduce the skip connections from the boundary decoder to the segmentation decoder, incorporating the boundary prior as the attention that indicates the error-prone regions into the segmentation process, and then define the consistency loss to push both decoders towards producing the same result. Besides, we also use the strategies of multi-scale input and deep supervision to extract hierarchical structural information, which can alleviate the issues caused by variable tumor sizes. We evaluated the proposed BA-Net on the kidney tumor segmentation challenge (KiTS19) dataset. The results suggest that the boundary decoder and consistency loss used in our model are effective and the BA-Net is able to produce relatively accurate segmentation of the kidney and kidney tumors.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Bray, F., Ferlay, J., Soerjomataram, I., Siegel, R.L., Torre, L.A., Jemal, A.: Global cancer statistics 2018: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries. CA: Cancer J. Clin. 68(6), 394–424 (2018)
Capitanio, U., Montorsi, F.: Renal cancer. Lancet 387(10021), 894–906 (2016)
Çiçek, Ö., Abdulkadir, A., Lienkamp, S.S., Brox, T., Ronneberger, O.: 3D u-net: learning dense volumetric segmentation from sparse annotation. In: Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), pp. 424–432 (2016)
Fang, X., Yan, P.: Multi-organ segmentation over partially labeled datasets with multi-scale feature abstraction. IEEE Transactions on Medical Imaging, pp. 1–1 (2020)
Ficarra, V., et al.: Preoperative aspects and dimensions used for an anatomical (padua) classification of renal tumours in patients who are candidates for nephron-sparing surgery. Eur. Urol. 56(5), 786–793 (2009)
Gao, S., Cheng, M.M., Zhao, K., Zhang, X.Y., Yang, M.H., Torr, P.H.: Res2net: a new multi-scale backbone architecture. In: IEEE Transactions on Pattern Analysis and Machine Intelligence, (2019)
Heller, N., et al.: The state of the art in kidney and kidney tumor segmentation in contrast-enhanced CT imaging: results of the kits19 challenge. arXiv preprint, (2019) arXiv:1912.01054
Heller, N., et al.: The kits19 challenge data: 300 kidney tumor cases with clinical context, ct semantic segmentations, and surgical outcomes. arXiv preprint arXiv:1904.00445 (2019)
Hou, X., et al.: A triple-stage self-guided network for kidney tumor segmentation. In: Proceedings of the IEEE International Symposium on Biomedical Imaging (ISBI), pp. 341–344. IEEE (2020)
Jia, H., Song, Y., Huang, H., Cai, W., Xia, Y.: HD-Net: hybrid discriminative network for prostate segmentation in MR images. In: Shen, D., et al. (eds.) MICCAI 2019. LNCS, vol. 11765, pp. 110–118. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-32245-8_13
Milletari, F., Navab, N., Ahmadi, S.A.: V-net: Fully convolutional neural networks for volumetric medical image segmentation. In: Proceedings of the International Conference on 3D Vision (3DV). pp. 565–571 (2016)
Myronenko, A., Hatamizadeh, A.: 3D kidneys and kidney tumor semantic segmentation using boundary-aware networks. arXiv preprint, (2019) arXiv:1909.06684
Paszke, A., et al.: Pytorch: An imperative style, high-performance deep learning library. In: Proceedings of the Advances in Neural Information Processing Systems, pp. 8024–8035 (2019)
Wu, Y., et al.: Vessel-Net: retinal vessel segmentation under multi-path supervision. In: Shen, D., et al. (eds.) MICCAI 2019. LNCS, vol. 11764, pp. 264–272. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-32239-7_30
Xie, Y., Lu, H., Zhang, J., Shen, C., Xia, Y.: Deep segmentation-emendation model for gland instance segmentation. In: Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), Springer. pp. 469–477 (2019) https://doi.org/10.1007/978-3-030-32239-7_52
Yu, Q., Shi, Y., Sun, J., Gao, Y., Zhu, J., Dai, Y.: Crossbar-net: a novel convolutional neural network for kidney tumor segmentation in CT images. IEEE Trans. Image Process. 28(8), 4060–4074 (2019)
Zhang, J., Xie, Y., Zhang, P., Chen, H., Xia, Y., Shen, C.: Light-weight hybrid convolutional network for liver tumor segmentation. In: Proceedings of the International Joint Conference on Artificial Intelligence (IJCAI), pp. 4271–4277 (2019)
Zhou, S., Nie, D., Adeli, E., Yin, J., Lian, J., Shen, D.: High-resolution encoder-decoder networks for low-contrast medical image segmentation. IEEE Trans. Image Process. 29, 461–475 (2019)
Acknowledgment
This work was supported in part by the Science and Technology Innovation Committee of Shenzhen Municipality, China, under Grants JCYJ20180306171334997, in part by the National Natural Science Foundation of China under Grants 61771397, and in part by the Project for Graduate Innovation team of Northwestern Polytechnical University.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2020 Springer Nature Switzerland AG
About this paper
Cite this paper
Hu, S., Zhang, J., Xia, Y. (2020). Boundary-Aware Network for Kidney Tumor Segmentation. In: Liu, M., Yan, P., Lian, C., Cao, X. (eds) Machine Learning in Medical Imaging. MLMI 2020. Lecture Notes in Computer Science(), vol 12436. Springer, Cham. https://doi.org/10.1007/978-3-030-59861-7_20
Download citation
DOI: https://doi.org/10.1007/978-3-030-59861-7_20
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-59860-0
Online ISBN: 978-3-030-59861-7
eBook Packages: Computer ScienceComputer Science (R0)
