Feasibility study to improve deep learning in OCT diagnosis of rare retinal diseases with few-shot classification

Med Biol Eng Comput. 2021 Feb;59(2):401-415. doi: 10.1007/s11517-021-02321-1. Epub 2021 Jan 25.

Abstract

Deep learning (DL) has been successfully applied to the diagnosis of ophthalmic diseases. However, rare diseases are commonly neglected due to insufficient data. Here, we demonstrate that few-shot learning (FSL) using a generative adversarial network (GAN) can improve the applicability of DL in the optical coherence tomography (OCT) diagnosis of rare diseases. Four major classes with a large number of datasets and five rare disease classes with a few-shot dataset are included in this study. Before training the classifier, we constructed GAN models to generate pathological OCT images of each rare disease from normal OCT images. The Inception-v3 architecture was trained using an augmented training dataset, and the final model was validated using an independent test dataset. The synthetic images helped in the extraction of the characteristic features of each rare disease. The proposed DL model demonstrated a significant improvement in the accuracy of the OCT diagnosis of rare retinal diseases and outperformed the traditional DL models, Siamese network, and prototypical network. By increasing the accuracy of diagnosing rare retinal diseases through FSL, clinicians can avoid neglecting rare diseases with DL assistance, thereby reducing diagnosis delay and patient burden.

Keywords: Deep learning; Few-shot learning; Generative adversarial network; Optical coherence tomography; Rare diseases.

MeSH terms

  • Deep Learning*
  • Feasibility Studies
  • Humans
  • Neural Networks, Computer
  • Rare Diseases
  • Retinal Diseases* / diagnostic imaging
  • Tomography, Optical Coherence