Abstract
Retinal vessels usually serve as biomarkers for early diagnosis and treatment of ophthalmic and systemic diseases. However, collecting and labeling these clinical images require extensive costs and thus existing models are commonly based on extremely limited labeled data for supervised segmentation of retinal vessels, which may hinder the effectiveness of deep learning methods. In this paper, we propose a novel point consistency-based semi-supervised (PCS) framework for retinal vessel segmentation, which can be trained both on annotated and unannotated fundus images. It consists of two modules, one of which is the segmentation module predicting the pixel-wise vessel segmentation map like a common segmentation network. Otherwise, considering that retinal vessels present tubular structures and hence the point set representation enjoys its prediction flexibility and consistency, a point consistency (PC) module is introduced to learn and express vessel skeleton structure adaptively. It inputs high-level features from the segmentation module and produces the point set representation of vessels simultaneously, facilitating supervised segmentation. Meanwhile, we design a consistency regularization between point set predictions and directly predicted segmentation results to explore the inherent segmentation perturbation of the point consistency, contributing to semi-supervised learning. We validate our method on a typical public dataset DRIVE and provide a new large-scale dataset (TR160, including 160 labeled and 120 unlabeled images) for both supervised and semi-supervised learning. Extensive experiments demonstrate that our method is superior to the state-of-the-art methods.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Fraz, M.M., Remagnino, P., Hoppe, A., Uyyanonvara, B., Rudnicka, A.R., Owen, C.G., Barman, S.A.: Blood vessel segmentation methodologies in retinal images-a survey. Comput. Methods Programs Biomed. 108(1), 407–433 (2012)
Fu, H., Xu, Y., Lin, S., Kee Wong, D.W., Liu, J.: Deepvessel: Retinal vessel segmentation via deep learning and conditional random field. In: MICCAI. pp. 132–139. Springer (2016)
Grandvalet, Y., Bengio, Y.: Semi-supervised learning by entropy minimization. Advances in neural information processing systems 17 (2004)
Guo, C., Szemenyei, M., Hu, Y., Wang, W., Zhou, W., Yi, Y.: Channel attention residual u-net for retinal vessel segmentation. In: ICASSP. pp. 1185–1189. IEEE (2021)
Hu, J., Wang, H., Wang, J., Wang, Y., He, F., Zhang, J.: Sa-net: A scale-attention network for medical image segmentation. PLoS ONE 16(4), e0247388 (2021)
Lee, D.H., et al.: Pseudo-label: The simple and efficient semi-supervised learning method for deep neural networks. In: Workshop on challenges in representation learning, ICML. vol. 3, p. 896 (2013)
Lin, T.Y., Goyal, P., Girshick, R., He, K., Dollár, P.: Focal loss for dense object detection. In: Proceedings of the IEEE International Conference on Computer Vision. pp. 2980–2988 (2017)
Liu, Y., Shen, J., Yang, L., Yu, H., Bian, G.: Wave-net: A lightweight deep network for retinal vessel segmentation from fundus images. Comput. Biol. Med. 152, 106341 (2023)
Lokhande, V.S., Tasneeyapant, S., Venkatesh, A., Ravi, S.N., Singh, V.: Generating accurate pseudo-labels in semi-supervised learning and avoiding overconfident predictions via hermite polynomial activations. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. pp. 11435–11443 (2020)
Luo, X., Chen, J., Song, T., Wang, G.: Semi-supervised medical image segmentation through dual-task consistency. In: Proceedings of the AAAI Conference on Artificial Intelligence. vol. 35, pp. 8801–8809 (2021)
Ronneberger, O., Fischer, P., Brox, T.: U-net: Convolutional networks for biomedical image segmentation. In: MICCAI. pp. 234–241. Springer (2015)
Souly, N., Spampinato, C., Shah, M.: Semi supervised semantic segmentation using generative adversarial network. In: Proceedings of the IEEE international conference on computer vision. pp. 5688–5696 (2017)
Staal, J., Abrà moff, M.D., Niemeijer, M., Viergever, M.A., Van Ginneken, B.: Ridge-based vessel segmentation in color images of the retina. IEEE Trans. Med. Imaging 23(4), 501–509 (2004)
Tarvainen, A., Valpola, H.: Mean teachers are better role models: Weight-averaged consistency targets improve semi-supervised deep learning results. Advances in neural information processing systems 30 (2017)
Verma, V., Kawaguchi, K., Lamb, A., Kannala, J., Solin, A., Bengio, Y., Lopez-Paz, D.: Interpolation consistency training for semi-supervised learning. Neural Networks 145, 90–106 (2022)
Vu, T.H., Jain, H., Bucher, M., Cord, M., Pérez, P.: Advent: Adversarial entropy minimization for domain adaptation in semantic segmentation. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. pp. 2517–2526 (2019)
Wang, D., Zhang, Z., Zhao, Z., Liu, Y., Chen, Y., Wang, L.: Pointscatter: Point set representation for tubular structure extraction. In: Computer Vision-ECCV 2022: 17th European Conference, Tel Aviv, Israel, October 23–27, 2022, Proceedings, Part XXI. pp. 366–383. Springer (2022)
Wang, Y., Wang, H., Shen, Y., Fei, J., Li, W., Jin, G., Wu, L., Zhao, R., Le, X.: Semi-supervised semantic segmentation using unreliable pseudo-labels. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. pp. 4248–4257 (2022)
Wong, T.Y., Klein, R., Klein, B.E., Tielsch, J.M., Hubbard, L., Nieto, F.J.: Retinal microvascular abnormalities and their relationship with hypertension, cardiovascular disease, and mortality. Surv. Ophthalmol. 46(1), 59–80 (2001)
Wu, H., Prasad, S.: Semi-supervised deep learning using pseudo labels for hyperspectral image classification. IEEE Trans. Image Process. 27(3), 1259–1270 (2017)
Wu, L., Li, J., Wang, Y., Meng, Q., Qin, T., Chen, W., Zhang, M., Liu, T.Y., et al.: R-drop: Regularized dropout for neural networks. Advances in Neural Information Processing Systems 34, 10890–10905 (2021)
Wu, Y., Xia, Y., Song, Y., Zhang, Y., Cai, W.: Multiscale network followed network model for retinal vessel segmentation. In: MICCAI. pp. 119–126. Springer (2018)
Yau, J.W., Rogers, S.L., Kawasaki, R., Lamoureux, E.L., Kowalski, J.W., Bek, T., Chen, S.J., Dekker, J.M., Fletcher, A., Grauslund, J., et al.: Global prevalence and major risk factors of diabetic retinopathy. Diabetes Care 35(3), 556–564 (2012)
Yu, L., Wang, S., Li, X., Fu, C.W., Heng, P.A.: Uncertainty-aware self-ensembling model for semi-supervised 3d left atrium segmentation. In: International Conference on Medical Image Computing and Computer-Assisted Intervention. pp. 605–613. Springer (2019)
Zhang, Y., Chung, A.C.: Deep supervision with additional labels for retinal vessel segmentation task. In: Medical Image Computing and Computer Assisted Intervention-MICCAI 2018: 21st International Conference, Granada, Spain, September 16–20, 2018, Proceedings, Part II 11. pp. 83–91. Springer (2018)
Zhang, Y., Yang, L., Chen, J., Fredericksen, M., Hughes, D.P., Chen, D.Z.: Deep adversarial networks for biomedical image segmentation utilizing unannotated images. In: MICCAI. pp. 408–416. Springer (2017)
Zhang, Z., Liu, Q., Wang, Y.: Road extraction by deep residual u-net. IEEE Geosci. Remote Sens. Lett. 15(5), 749–753 (2018)
Zhou, Z., Rahman Siddiquee, M.M., Tajbakhsh, N., Liang, J.: Unet++: A nested u-net architecture for medical image segmentation. In: MICCAI. pp. 3–11. Springer (2018)
Zhu, X., Goldberg, A.B.: Introduction to semi-supervised learning. Synthesis lectures on artificial intelligence and machine learning 3(1), 1–130 (2009)
Acknowledgement
This work was supported by the Beijing Natural Science Foundation under Grant Z200024, in part by Hefei Innovation Research Institute, Beihang University, and in part by the University Synergy Innovation Program of Anhui Province under Grant GXXT-2019-044.
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
Hu, J., Qiu, L., Wang, H., Zhang, J. (2024). Semi-supervised Retinal Vessel Segmentation Through Point Consistency. In: Liu, Q., et al. Pattern Recognition and Computer Vision. PRCV 2023. Lecture Notes in Computer Science, vol 14437. Springer, Singapore. https://doi.org/10.1007/978-981-99-8558-6_13
Download citation
DOI: https://doi.org/10.1007/978-981-99-8558-6_13
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-8557-9
Online ISBN: 978-981-99-8558-6
eBook Packages: Computer ScienceComputer Science (R0)