Paper
29 October 2018 Automatic classification of diabetic retinopathy based on convolutional neural networks
Xingming Zhang, Wanwan Zhang, Mingchao Fang, Jiale Xue, Lifeng Wu
Author Affiliations +
Proceedings Volume 10836, 2018 International Conference on Image and Video Processing, and Artificial Intelligence; 1083608 (2018) https://doi.org/10.1117/12.2503883
Event: 2018 International Conference on Image, Video Processing and Artificial Intelligence, 2018, Shanghai, China
Abstract
In this paper, we propose a novel classification algorithm based on convolutional neural networks (CNNs) to diagnose the severity of diabetic retinopathy (DR). We adopt a series of preprocessing operations to improve the quality of dataset. In addition, data augmentation is implemented on the training data to tackle the problem of imbalanced dataset. We design a CNNs model named DR-Net with a new Adaptive Cross-Entropy Loss, which emphasizes the difference of the penalty when training data are misclassified into different intervals. We train DR-Net on the publicly available Kaggle dataset. Experimental results show that our DR-Net achieves an accuracy of 0.821 and a kappa score of 0.663 on 3338 testing images.
© (2018) COPYRIGHT Society of Photo-Optical Instrumentation Engineers (SPIE). Downloading of the abstract is permitted for personal use only.
Xingming Zhang, Wanwan Zhang, Mingchao Fang, Jiale Xue, and Lifeng Wu "Automatic classification of diabetic retinopathy based on convolutional neural networks", Proc. SPIE 10836, 2018 International Conference on Image and Video Processing, and Artificial Intelligence, 1083608 (29 October 2018); https://doi.org/10.1117/12.2503883
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Convolutional neural networks

Data modeling

Algorithm development

Detection and tracking algorithms

Image classification

Retina

Diagnostics

Back to Top