Abstract
Kidney cancer is one of the most common cancers worldwide. Automated segmentation of kidneys, tumors, and cysts from CT images is an important pathway to assist doctors in diagnosis. However, the diverse morphologies of tumors and cysts pose challenges regarding their difficult identification and unpredictable behavior. This paper proposes a two-stage segmentation method, proceeding from coarse to fine. In the first stage, we obtain the kidney region of interest (ROI) based on nnU-Net as input for the second stage. In the second stage, we design a parallel encoder structure. It employs a dual-stream end-to-end training approach, simultaneously monitoring and segmenting boundary information and targets. In particular, a residual channel attention mechanism was incorporated with the boundary prediction branch, highlighting the most relevant feature channels. The method has been experimentally demonstrated to be significantly superior to the baseline nnU-Net. On the official test set, our Kidney + Masses Dice and Tumor Dice are 0.936 and 0.670, respectively, ranking 14th on the leaderboard.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Chow, W.H., Dong, L.M., Devesa, S.S.: Epidemiology and risk factors for kidney cancer. Nat. Rev. Urol. 7(5), 245–257 (2010)
Li, X., Liu, L., Heng, P.A.: H-Denseunet for kidney and tumor segmentation from CT scans (2019)
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)
Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. In: Navab, N., Hornegger, J., Wells, W.M., Frangi, A.F. (eds.) MICCAI 2015. LNCS, vol. 9351, pp. 234–241. Springer, Cham (2015). https://doi.org/10.1007/978-3-319-24574-4_28
Isensee, F., Jaeger, P.F., Kohl, S.A., Petersen, J., Maier-Hein, K.H.: nnu-net: a self-configuring method for deep learning-based biomedical image segmentation. Nat. Methods 18(2), 203–211 (2021)
Zhao, Z., Chen, H., Wang, L.: A coarse-to-fine framework for the 2021 kidney and kidney tumor segmentation challenge. In: Heller, N., Isensee, F., Trofimova, D., Tejpaul, R., Papanikolopoulos, N., Weight, C. (eds.) KiTS 2021. LNCS, vol. 13168, pp. 53–58. Springer, Cham (2022). https://doi.org/10.1007/978-3-030-98385-7_8
George, Y.: A coarse-to-fine 3D U-Net network for semantic segmentation of kidney CT scans. In: Heller, N., Isensee, F., Trofimova, D., Tejpaul, R., Papanikolopoulos, N., Weight, C. (eds.) KiTS 2021. LNCS, vol. 13168, pp. 137–142. Springer, Cham (2022). https://doi.org/10.1007/978-3-030-98385-7_18
Zhao, Z., Chen, H., Li, J., Wang, L.: Boundary attention u-net for kidney and kidney tumor segmentation. In: 2022 44th Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC), pp. 1540–1543. IEEE (2022)
Fang, Y., Chen, C., Yuan, Y., Tong, K.: Selective feature aggregation network with area-boundary constraints for polyp segmentation. In: Shen, D., et al. (eds.) MICCAI 2019. LNCS, vol. 11764, pp. 302–310. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-32239-7_34
Murugesan, B., Sarveswaran, K., Shankaranarayana, S.M., Ram, K., Joseph, J., Sivaprakasam, M.: Psi-net: shape and boundary aware joint multi-task deep network for medical image segmentation. In: 2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), pp. 7223–7226. IEEE (2019)
Zhou, T., et al.: Cross-level feature aggregation network for polyp segmentation. Pattern Recogn. 140, 109555 (2023)
Jin, Q., Meng, Z., Sun, C., Cui, H., Su, R.: Ra-UNet: A hybrid deep attention-aware network to extract liver and tumor in CT scans. Front. Bioeng. Biotechnol. 8, 605132 (2020)
Oktay, O., et al.: Attention u-net: learning where to look for the pancreas. arXiv preprint arXiv:1804.03999 (2018)
Li, C., et al.: Attention unet++: a nested attention-aware u-net for liver CT image segmentation. In: 2020 IEEE International Conference on Image Processing (ICIP), pp. 345–349. IEEE (2020)
Luu, H.M., Park, S.H.: Extending nn-UNet for brain tumor segmentation. In: Crimi, A., Bakas, S. (eds.) BrainLes 2021. LNCS, vol. 1263, pp. 173–186. Springer, Cham (2021). https://doi.org/10.1007/978-3-031-09002-8_16
Gu, R., et al.: Ca-net: comprehensive attention convolutional neural networks for explainable medical image segmentation. IEEE Trans. Med. Imaging 40(2), 699–711 (2020)
Acknowledgment
This work was supported in part by the National Natural Science Foundation of China (Grant No. 61976126). We want to express our gratitude to the organizers of KiTS2023 and the nnU-Net team for their assistance.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Li, Z., Peng, Y., Zhang, Z. (2024). Two-Stage Segmentation Framework with Parallel Decoders for the Kidney and Kidney Tumor Segmentation. In: Heller, N., et al. Kidney and Kidney Tumor Segmentation. KiTS 2023. Lecture Notes in Computer Science, vol 14540. Springer, Cham. https://doi.org/10.1007/978-3-031-54806-2_12
Download citation
DOI: https://doi.org/10.1007/978-3-031-54806-2_12
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-54805-5
Online ISBN: 978-3-031-54806-2
eBook Packages: Computer ScienceComputer Science (R0)