Abstract
Accurate medical image segmentation is the foundation of clinical imaging diagnosis and 3D image reconstruction. However, medical images often have low contrast between target objects, greatly affected by organ movement, and suffer from limited annotated samples. To address these issues, we propose a few-shot medical image segmentation network with boundary category correction named Boundary Category Correction Network (BCC-Net). Of overall medical few-shot learning framework, we first propose the Prior Mask Generation Module (PRGM) and Multi-scale Feature Fusion Module (MFFM). PRGM can better localize the query target, while MFFM can adaptively fuse the support set prototype, the prior mask and the query set features at different scales to solve the problem of the spatial inconsistency between the support set and the query set. To improve segmentation accuracy, we construct an additional base-learning branch, which, together with the meta-learning branch, forms the Boundary Category Correction Framework (BCCF). It corrects the boundary category of the meta-learning branch prediction mask by predicting the region of the base categories in the query set. Experiments are conducted on the mainstream ABD-MR and ABD-CT medical image segmentation public datasets. Comparative analysis and ablation experiments are performed with a variety of existing state-of-the-art few-shot segmentation methods. The results demonstrate that the effectiveness of the proposed method with significant enhance the segmentation performance on medical images.
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
Similar content being viewed by others
References
Chen, L.C., Papandreou, G., et al.: DeepLab: semantic image segmentation with deep convolutional nets, atrous convolution, and fully connected CRFs. IEEE Trans. Pattern Anal. Mach. Intell. 40, 834–848 (2016)
Finn, C., Abbeel, P., Levine, S.: Model-agnostic meta-learning for fast adaptation of deep networks. arXiv arXiv:1703.03400 (2017)
He, K., et al.: Deep residual learning for image recognition. In: 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 770–778 (2015)
Heidari, M., Kazerouni, A., et al.: HiFormer: hierarchical multi-scale representations using transformers for medical image segmentation. In: 2023 IEEE/CVF Winter Conference on Applications of Computer Vision (WACV), pp. 6191–6201 (2022)
Kavur, A.E., Gezer, N.S., et al.: CHAOS challenge - combined (CT-MR) healthy abdominal organ segmentation. Med. Image Anal. 69, 101950 (2020)
Koch, G.R.: Siamese neural networks for one-shot image recognition (2015)
Landman, B., Xu, Z., Igelsias, J., Styner, M., et al.: MICCAI multi-atlas labeling beyond the cranial vault-workshop and challenge. In: Proceedings of the MICCAI Multi-Atlas Labeling Beyond Cranial Vault-Workshop Challenge, vol. 5, p. 12 (2015)
Lang, C., Cheng, G., Tu, B., Han, J.: Learning what not to segment: a new perspective on few-shot segmentation. In: 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 8047–8057 (2022)
Li, G., Jampani, V., Sevilla-Lara, L., Sun, D., Kim, J., Kim, J.: Adaptive prototype learning and allocation for few-shot segmentation. In: 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 8330–8339 (2021)
Lin, T.-Y., et al.: Microsoft COCO: common objects in context. In: Fleet, D., Pajdla, T., Schiele, B., Tuytelaars, T. (eds.) ECCV 2014. LNCS, vol. 8693, pp. 740–755. Springer, Cham (2014). https://doi.org/10.1007/978-3-319-10602-1_48
Ouyang, C., Biffi, C., Chen, C., Kart, T., Qiu, H., Rueckert, D.: Self-supervision with superpixels: training few-shot medical image segmentation without annotation. In: Vedaldi, A., Bischof, H., Brox, T., Frahm, J.-M. (eds.) ECCV 2020. LNCS, vol. 12374, pp. 762–780. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-58526-6_45
Roy, A.G., Siddiqui, S., et al.: ‘squeeze & excite’ guided few-shot segmentation of volumetric images. Med. Image Anal. 59, 101587 (2019)
Shen, X., Zhang, G., Lai, H., et al.: PoissonSeg: semi-supervised few-shot medical image segmentation via Poisson learning. In: 2021 IEEE International Conference on Bioinformatics and Biomedicine (BIBM), pp. 1513–1518 (2021)
Su, X., et al.: Amotivation, career engagement, and the moderating role of career adaptability of youth not in education, employment, or training (2020)
Sun, L., Li, C., Ding, X., Huang, Y., Wang, G., Yu, Y.: Few-shot medical image segmentation using a global correlation network with discriminative embedding. Comput. Biol. Med. 140, 105067 (2020)
Tang, H., Liu, X., Sun, S., Yan, X., Xie, X.: Recurrent mask refinement for few-shot medical image segmentation. In: 2021 IEEE/CVF International Conference on Computer Vision (ICCV), pp. 3898–3908 (2021)
Tian, Z., Zhao, H., Shu, M., Yang, Z., Li, R., Jia, J.: Prior guided feature enrichment network for few-shot segmentation. IEEE Trans. Pattern Anal. Mach. Intell. 44, 1050–1065 (2020)
Vinyals, O., Blundell, C., Lillicrap, T.P., Kavukcuoglu, K., Wierstra, D.: Matching networks for one shot learning. arXiv arXiv:1606.04080 (2016)
Wang, K., Liew, J.H., Zou, Y., Zhou, D., Feng, J.: PANet: few-shot image semantic segmentation with prototype alignment. In: 2019 IEEE/CVF International Conference on Computer Vision (ICCV), pp. 9196–9205 (2019)
Ye, Z., Zhang, W.: A dynamic few-shot learning framework for medical image stream mining based on self-training. EURASIP J. Adv. Sig. Process. 2023, 1–19 (2023)
Yu, Q., Dang, K., Tajbakhsh, N., Terzopoulos, D., Ding, X.: A location-sensitive local prototype network for few-shot medical image segmentation. In: 2021 IEEE 18th International Symposium on Biomedical Imaging (ISBI), pp. 262–266 (2021)
Zhang, C., Lin, G., et al.: CANet: class-agnostic segmentation networks with iterative refinement and attentive few-shot learning. In: 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 5212–5221 (2019)
Zhao, H., Shi, J., et al.: Pyramid scene parsing network. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 6230–6239 (2016)
Funding
This work is supported by the National Natural Science Foundation of China (Nos. (82071876, 6217010009, 82372043, 82371904).
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 Singapore Pte Ltd.
About this paper
Cite this paper
Xu, Z., Jia, X., Guo, X., Wang, L., Zheng, Y. (2024). A Few-Shot Medical Image Segmentation Network with Boundary Category Correction. In: Liu, Q., et al. Pattern Recognition and Computer Vision. PRCV 2023. Lecture Notes in Computer Science, vol 14434. Springer, Singapore. https://doi.org/10.1007/978-981-99-8549-4_31
Download citation
DOI: https://doi.org/10.1007/978-981-99-8549-4_31
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-8548-7
Online ISBN: 978-981-99-8549-4
eBook Packages: Computer ScienceComputer Science (R0)