Towards an extended EfficientNet-based U-Net framework for joint optic disc and cup segmentation in the fundus image

Jinke Wang,Xiang Li,Yuanzhi Cheng
DOI: https://doi.org/10.1016/j.bspc.2023.104906
IF: 5.1
2023-04-01
Biomedical Signal Processing and Control
Abstract:Purpose Automatic and accurate segmentation of the optic disc (OD) and optic cup (OC) is crucial for clinical glaucoma screening. However, the segmentation accuracy cannot meet the requirements of clinical diagnosis yet. This paper aims to propose a novel joint segmentation framework to improve accuracy. Methods We suggested an extended EfficientNet-based U-Net, named EE-UNet. First, we use the EfficientNet to extract distinguishable features under different scales of receptive fields. Second, we employ the Conditional Random Field as a Recurrent Neural Network (CRF-RNN) to extend the U-Net framework, maintaining an end-to-end mode. Third, the Ranger optimizer is suggested for faster and more stable convergence with minimal computational cost. Finally, we designed a multi-label loss function to balance the foreground and background pixels. Results Extensive experiments were trained and verified on the datasets REFUGE and GAMMA and tested on Drishti-GS1 and RIM-ONE-v3. The proposed EE-UNet achieved a high DICE score in OD/OC segmentation (0.9624/0.9228, respectively), outperforming the state-of-the-art (SOTA) methods. Conclusion Quantitative and qualitative results verified the superior performance of the proposed method in OC and OD segmentation, thus proving that it can be a promising tool in the early screening of large-scale glaucoma.
engineering, biomedical
What problem does this paper attempt to address?