Patch-based Output Space Adversarial Learning for Joint Optic Disc and Cup Segmentation

Shujun Wang
Lequan Yu
Xin Yang
Chi-Wing Fu
Pheng-Ann Heng
The Chinese University of Hong Kong

IEEE TMI 2019 [Paper]
Code [GitHub]




Abstract

Glaucoma is a leading cause of irreversible blindness. Accurate segmentation of the optic disc (OD) and cup (OC) from fundus images is beneficial to glaucoma screening and diagnosis. Recently, convolutional neural networks demonstrate promising progress in joint OD and OC segmentation. However, affected by the domain shift among different datasets, deep networks are severely hindered in generalizing across different scanners and institutions. In this paper, we present a novel patch-based Output Space Adversarial Learning framework (\textit{p}OSAL) to jointly and robustly segment the OD and OC from different fundus image datasets. We first devise a light-weight and efficient segmentation network as a backbone. Considering the specific morphology of OD and OC, a novel morphology-aware segmentation loss is proposed to guide the network to generate accurate and smooth segmentation. Our \textit{p}OSAL framework then exploits unsupervised domain adaptation to address the domain shift challenge by encouraging the segmentation in the target domain to be similar to the source ones. Since the whole-segmentation-based adversarial loss is insufficient to drive the network to capture segmentation details, we further design the \textit{p}OSAL in a patch-based fashion to enable fine-grained discrimination on local segmentation details. We extensively evaluate our \textit{p}OSAL framework and demonstrate its effectiveness in significantly improving segmentation on three public retinal fundus image datasets, \ie, Drishti-GS, RIM-ONE-r3 and REFUGE. Furthermore, our \textit{p}OSAL framework achieved the first place in the OD and OC segmentation tasks in \textit{MICCAI 2018 Retinal Fundus Glaucoma Challenge}.


Overview

Pipeline of pOSAL


Segmentation Network Architecture



Segmentation results on Drishti-GS dataset.



More Results Download Links

Datasets: [Drishti-GS] [RIM-ONE-r3] [REFUGE test]




Acknowledgements