Abstract
Age-related macular degeneration (AMD) is one of the main causes of visual impairment in elderly people, with drusen and choroidal neovascularization (CNV) being two characterizing types of lesions. Based on optical coherence tomography (OCT), image classification can be used in AMD diagnosis, while image segmentation is necessary for quantitative assessment of the lesion area. In this paper, we propose a deep learning framework exploiting dual guidance between the two tasks. Firstly, a complementary mask guided convolutional neural network (CM-CNN) is proposed to perform classification of OCT B-scans with drusen or CNV from normal ones, where the guiding mask is generated by the auxiliary segmentation task. Secondly, a class activation map guided UNet (CAM-UNet) is proposed to achieve segmentation of drusen and CNV lesions, using CAM output from the CM-CNN. Tested on a subset of the public UCSD dataset, and compared with five classification networks, four segmentation networks, and three multi-task networks, the proposed dual guidance network has achieved higher accuracy both in classification and segmentation. The classification accuracy reaches 96.93% and the Dice coefficient for segmentation reaches 77.51%. Results on an extra dataset for detection of macular edema and segmentation of retinal fluids further show the generalizability of the proposed model.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.