Data augmentation for deep learning of non-mydriatic screening retinal fundus images
Por:
Moya-Sánchez E.U., Sánchez A., Zapata M., Moreno J., Garcia-Gasulla D., Parrés F., Ayguadé E., Labarta J., Cortés U.
Publicada:
1 ene 2019
Resumen:
Fundus image is an effective and low-cost tool to screen for common retinal diseases. At the same time, Deep Learning (DL) algorithms have been shown capable of achieving similar or even better performance accuracies than physicians in certain image classification tasks. One of the key aspects to improve the performance of DL models is to use data augmentation techniques. Data augmentation reduces the impact of overfitting and improves the generalization capacity of the models. However, the most appropriate data augmentation methodology is highly dependant on the nature of the problem. In this work, we propose a data augmentation and image enhancement algorithm for the task of classifying non-mydriatic fundus images of pigmented abnormalities in the macula. For training, fine tuning and data augmentation, we used the Barcelona Supercomputing Centre cluster CTE IBM Power8+ and Marenostrum IV. The parallelization and optimization of the algorithms were performed using Numba, and Python-Multiprocessing, made compatible with the underlying DL framework used for training the model. We propose and trained a specific DL model from scratch. Our main results are an increase in the number of input images up to a factor of, and report the information of quality images for. As a result, our data augmentation approach results in an increase of up to 9% in classification accuracy. © Springer Nature Switzerland AG 2019.
Filiaciones:
Moya-Sánchez E.U.:
Barcelona Supercomputing Center, Barcelona, Spain
Posgrado en Ciencias Computacionales, Universidad Autónoma de Guadalajara, Guadalajara, Mexico
Sánchez A.:
Posgrado en Ciencias Computacionales, Universidad Autónoma de Guadalajara, Guadalajara, Mexico
Zapata M.:
Ophthalmology, Hospital Vall d’Hebron, Barcelona, Barcelona, Spain
Moreno J.:
Barcelona Supercomputing Center, Barcelona, Spain
Garcia-Gasulla D.:
Barcelona Supercomputing Center, Barcelona, Spain
Parrés F.:
Barcelona Supercomputing Center, Barcelona, Spain
Ayguadé E.:
Barcelona Supercomputing Center, Barcelona, Spain
Universitat Politèctnica de Catalunya, Barcelona Tech, Barcelona, Spain
Labarta J.:
Barcelona Supercomputing Center, Barcelona, Spain
Universitat Politèctnica de Catalunya, Barcelona Tech, Barcelona, Spain
Cortés U.:
Barcelona Supercomputing Center, Barcelona, Spain
Universitat Politèctnica de Catalunya, Barcelona Tech, Barcelona, Spain
|