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

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.