딥러닝을 이용한 당뇨성황반부종 등급 분류의 정확도 개선을 위한 검증 데이터 증강 기법
Validation Data Augmentation for Improving the Grading Accuracy of Diabetic Macular Edema using Deep Learning
2
조회
0
다운로드
0
인용
0
참고문헌
키워드
초록
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 ± 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.
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 ± 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.
참고문헌 (14)
- 계효정, "당뇨망막병증 최신 지견", 대한내과학회지, 89(3), pp.271~276 (2015), 대한내과학회[DOI]
- Jee D, "Prevalence and risk factors for diabetic retinopathy : the Korea National Health and Nutrition Examination Survey 2008-2011", Invest Ophthalmol Vis Sci, 54(10), pp.6827~6833 (2013)
- Abramoff MD, "Improved Automated Detection of Diabetic Retinopathy on a Publicly Available Dataset Through Integration of Deep Learning", Invest Ophthalmol Vis Sci, 57(13), pp.5200~5206 (2016)
- Abramoff MD, "Pivotal trial of an autonomous AI-based diagnostic system for detection of diabetic retinopathy in primary care offices", npj Digital Medicine, 1(39), pp.1~8 (2018)
- van der Heijden AA, "Validation of automated screening for referable diabetic retinopathy with the IDx-DR device in the Hoorn Diabetes Care System", Acta Ophthalmol, 96(1), pp.63~68 (2018)
- "Messidor"[URL]
- Gulshan V, "Development and Validation of a Deep Learning Algorithm for Detection of Diabetic Retinopathy in Retinal Fundus Photographs", JAMA, 316(22), pp.2402~2410 (2016)
- Al-Bander B, "Diabetic Macular Edema Grading Based on Deep Neural Networks", Proceedings of the Ophthalmic Medical Image Analysis International Workshop OMIA, 2016(1), pp.121~128 (2016)
- "Diabetic Retinopathy Detection"[URL]
- Wang Z, "Diabetic Retinopathy Detection via Deep Convolutional Networks for Discriminative Localization and Visual Explanation", arXiv, vol. 1703, no. 10757, 1~9 (2017)
- Lam C, "Automated Detection of Diabetic Retinopathy using Deep Learning", AMIA Jt Summits Transl Sci Proc, 2017, pp.147~155 (2018)
- Pratt H, "Convolutional Neural Networks for Diabetic Retinopathy", Procedia Computer Science, 90, pp.200~205 (2016)
- Ting DSW, "Development and Validation of a Deep Learning System for Diabetic Retinopathy and Related Eye Diseases Using Retinal Images From Multiethnic Populations With Diabetes", JAMA, 318(22), pp.2211~2223 (2017)
- Rajalakshmi R, "Automated diabetic retinopathy detection in smartphonebased fundus photography using artificial intelligence", Eye, 32(6), pp.1138~1144 (2018)