Browse > Article
http://dx.doi.org/10.9718/JBER.2019.40.2.48

Validation Data Augmentation for Improving the Grading Accuracy of Diabetic Macular Edema using Deep Learning  

Lee, Tae Soo (Department of Biomedical Engineering, College of Medicine and Medical Research Institute, Chungbuk National University)
Publication Information
Journal of Biomedical Engineering Research / v.40, no.2, 2019 , pp. 48-54 More about this Journal
Abstract
This paper proposed a method of validation data augmentation for improving the grading accuracy of diabetic macular edema (DME) using deep learning. The data augmentation technique is basically applied in order to secure diversity of data by transforming one image to several images through random translation, rotation, scaling and reflection in preparation of input data of the deep neural network (DNN). In this paper, we apply this technique in the validation process of the trained DNN, and improve the grading accuracy by combining the classification results of the augmented images. To verify the effectiveness, 1,200 retinal images of Messidor dataset was divided into training and validation data at the ratio 7:3. By applying random augmentation to 359 validation data, $1.61{\pm}0.55%$ accuracy improvement was achieved in the case of six times augmentation (N=6). This simple method has shown that the accuracy can be improved in the N range from 2 to 6 with the correlation coefficient of 0.5667. Therefore, it is expected to help improve the diagnostic accuracy of DME with the grading information provided by the proposed DNN.
Keywords
deep learning; Validation data augmentation; Diagnostic accuracy; Diabetic macular edema;
Citations & Related Records
연도 인용수 순위
  • Reference
1 Gye HJ and Song SJ, "New Modalities for the Diagnosis and Treatment of Diabetic Retinopathy," Korean J Med, vol. 89, no. 3, pp. 271-276, 2015.   DOI
2 Jee D, Lee WK and Kang S, "Prevalence and risk factors for diabetic retinopathy: the Korea National Health and Nutrition Examination Survey 2008-2011," Invest Ophthalmol Vis Sci, vol. 54, no. 10, pp. 6827-33, 2013.   DOI
3 Abramoff MD, Lou Y, Erginay A, Clarida W, Amelon R, "Folk JC and Niemeijer M. Improved Automated Detection of Diabetic Retinopathy on a Publicly Available Dataset Through Integration of Deep Learning," Invest Ophthalmol Vis Sci, vol. 57, no. 13, pp. 5200-5206, 2016.   DOI
4 Abramoff MD, Lavin PT, Birch M, Shah N and Folk JC, "Pivotal trial of an autonomous AI-based diagnostic system for detection of diabetic retinopathy in primary care offices," npj Digital Medicine, vol. 1, no. 39, pp. 1-8, 2018.   DOI
5 van der Heijden AA, Abramoff MD, Verbraak F, van Hecke MV, Liem A and Nijpels G, "Validation of automated screening for referable diabetic retinopathy with the IDx-DR device in the Hoorn Diabetes Care System," Acta Ophthalmol, vol. 96, no. 1, pp. 63-68, 2018.   DOI
6 Messidor. Available at: http://www.adcis.net/en/DownloadThirdParty/Messidor.html. Accessed on Sep. 1, 2017.
7 Gulshan V, Peng L, Coram M, Stumpe MC, Wu D, Narayanaswamy A, Venugopalan S, Widner K, Madams T, Cuadros J, Kim R, Raman R, Nelson PC, Mega JL and Webster DR, "Development and Validation of a Deep Learning Algorithm for Detection of Diabetic Retinopathy in Retinal Fundus Photographs," JAMA, vol. 316, no. 22, pp. 2402-2410, 2016.   DOI
8 Al-Bander B, Al-Nuaimy W, Al-Taee MA, Williams BM and Zheng Y, "Diabetic Macular Edema Grading Based on Deep Neural Networks," Proceedings of the Ophthalmic Medical Image Analysis International Workshop OMIA, vol. 2016, no. 1, pp. 121-128, 2016.
9 Kaggle, Inc. Diabetic Retinopathy Detection. Available at: https://www.kaggle.com/c/diabetic-retinopathydetection. Accessed on Sep. 1, 2016.
10 Wang Z and Yang J, "Diabetic Retinopathy Detection via Deep Convolutional Networks for Discriminative Localization and Visual Explanation," arXiv, vol. 1703, no. 10757, pp. 1-9, 2017.
11 Lam C, Darvin Yi D, Guo M and Lindsey T, "Automated Detection of Diabetic Retinopathy using Deep Learning," AMIA Jt Summits Transl Sci Proc, vol. 2017, pp. 147-155, 2018.
12 Pratt H, Coenen F, Broadbent DM, Harding SP and Zheng Y, "Convolutional Neural Networks for Diabetic Retinopathy," Procedia Computer Science, vol. 90, pp. 200-205, 2016.   DOI
13 Ting DSW, Cheung CY, Lim G, Tan GSW, Quang ND, Gan A, Hamzah H, Garcia-Franco R, San Yeo IY, Lee SY, Wong EYM, Sabanayagam C, Baskaran M, Ibrahim F, Tan NC, Finkelstein EA, Lamoureux EL, Wong IY, Bressler NM, Sivaprasad S, Varma R, Jonas JB, He MG, Cheng CY, Cheung GCM, Aung T, Hsu W, Lee ML and Wong TY, "Development and Validation of a Deep Learning System for Diabetic Retinopathy and Related Eye Diseases Using Retinal Images From Multiethnic Populations With Diabetes," JAMA, vol. 318, no. 22, pp. 2211-2223, 2017.   DOI
14 Rajalakshmi R, Subashini R, Anjana RM and Mohan V, "Automated diabetic retinopathy detection in smartphone-based fundus photography using artificial intelligence," Eye, vol. 32, no. 6, pp. 1138-1144, 2018.   DOI