Particle Swarm Optimization for Great Enhancement in Semi-supervised Retinal Vessel Segmentation with Generative Adversarial Networks

Qiang Huo,Geyu Tang,Feng Zhang
DOI: https://doi.org/10.1007/978-3-030-33327-0_14
2019-01-01
Abstract:Retinal vessel segmentation based on deep learning requires a lot of manual labeled data. That’s time-consuming, laborious and professional. In this paper, we propose a data-efficient semi-supervised learning framework, which effectively combines the existing deep learning network with generative adversarial networks (GANs) and self-training ideas. In view of the difficulty of tuning hyper-parameters of semi-supervised learning, we propose a method for hyper-parameters selection based on particle swarm optimization (PSO) algorithm. This work is the first demonstration that combines intelligent optimization with semi-supervised learning for achieving the best performance. Under the collaboration of adversarial learning, self-training and PSO, we obtain the performance of retinal vessel segmentation approximate to or even better than representative supervised learning using only one tenth of the labeled data from DRIVE.
What problem does this paper attempt to address?