Joint optic disc and cup segmentation using feature fusion and attention

Abstract

Currently, glaucoma is one of the leading causes of irreversible vision loss. So far, glaucoma is incurable, but early treatment can stop the progression of the condition and slow down the speed and extent of vision loss. Early detection and treatment are crucial to prevent glaucoma from developing into blindness. It is an effective method for glaucoma diagnosis to measure Cup to Disc Ratio (CDR) by the segmentation of Optic Disc (OD) and Optic Cup (OC). Compared with OD segmentation, OC segmentation still faces difficulties in segmentation accuracy. In this paper, a deep learning architecture named FAU-Net (feature fusion and attention U-Net) is proposed for the joint segmentation of OD and OC. It is an improved architecture based on U-Net. By adding a feature fusion module in U-Net, information loss in feature extraction can be reduced. The channel and spatial attention mechanisms are combined to highlight the important features related to the segmentation task and suppress the expression of irrelevant regional features. Finally, a multi-label loss is used to generate the final joint segmentation of OD and OC. Experimental results show that the proposed FAU-Net outperforms the state-of-the-art segmentation of OD and OC on Drishti-GS1, REFUGE, RIM-ONE and ODIR datasets.

Keywords: Attention; Deep learning; Glaucoma screening; OD and OC segmentation; U-Net.