On OCT Image Classification via Deep Learning

• Main Authors: Depeng Wang, Liejun Wang
• Format: Article
• Language: English
• Published: IEEE 2019-01-01
• Series: IEEE Photonics Journal
• Subjects: Deep learning; optical coherence tomography; diabetic macular edema; age-related macular degeneration automated diagnosis; computer-aided diagnosis
• Online Access: https://ieeexplore.ieee.org/document/8794616/

Computer-aided diagnosis of retinopathy is a research hotspot in the field of medical image classification. Diabetic macular edema (DME) and age-related macular degeneration (AMD) are two common ocular diseases that can result in partial or complete loss of vision. Optical coherence tomography imaging (OCT) is widely applied to the diagnosis of ocular diseases including DME and AMD. In this paper, an automatic method based on deep learning is proposed to detect AME and AMD lesions, in which two publicly available OCT datasets of retina were adopted and a network model with effective feature of reuse feature was applied to solve the problem of small datasets and enhance the adaptation to the difference of different datasets of the approach. Several network models with effective feature of reusable feature were compared and the transfer learning on networks with pre-trained models was realized. CliqueNet achieves better, classification results compared with other network models with a more than 0.98 accuracy and 0.99 of area under the curve (AUC) value finally.