KFWC: A Knowledge-Driven Deep Learning Model for Fine-grained Classification of Wet-AMD.

BACKGROUND AND

OBJECTIVES

Automated diagnosis using deep neural networks can help ophthalmologists detect the blinding eye disease wet Age-related Macular Degeneration (AMD). Wet-AMD has two similar subtypes, Neovascular AMD and Polypoidal Choroidal Vasculopathy (PCV). However, due to the difficulty in data collection and the similarity between images, most studies have only achieved the coarse-grained classification of wet-AMD rather than a fine-grained one of wet-AMD subtypes. Therefore, designing and building a deep learning model to diagnose neovascular AMD and PCV is a great challenge.

METHODS

To solve this issue, in this paper, we propose a Knowledge-driven Fine-grained Wet-AMD Classification Model (KFWC) to enhance the model's accuracy in the fine-grained disease classification with insufficient data. We innovatively introduced a two-stage method. In the first stage, we present prior knowledge of 10 lesion signs through pre-training; in the second stage, the model implements the classification task with the help of human knowledge. With the pre-training of priori knowledge of 10 lesion signs from input images, KFWC locates the powerful image features in the fine-grained disease classification task and therefore achieves better classification.

RESULTS

To demonstrate the effectiveness of KFWC, we conduct a series of experiments on a clinical dataset collected in cooperation with a Grade III Level A ophthalmology hospital in China. The AUC score of KFWC reaches 99.71%, with 6.69% over the best baseline and 4.14% over ophthalmologists. KFWC can also provide good interpretability and effectively alleviate the pressure of data collection and annotation in the field of fine-grained disease classification for wet-AMD.

CONCLUSIONS

The model proposed in this paper effectively solves the difficulties of small data volume and high image similarity in the wet-AMD fine-grained classification task through a knowledge-driven approach. Besides, this method effectively relieves the pressure of data collection and annotation in the field of fine-grained classification. In the diagnosis of wet-AMD, KFWC is superior to previous work and human ophthalmologists.