Hybrid Deep Learning Gaussian Process for Diabetic Retinopathy Diagnosis and Uncertainty Quantification

Abstract

La retinopatía diabética (RD) es una de las complicaciones microvasculares de la diabetes mellitus, que sigue siendo una de las principales causas de ceguera en todo el mundo. Los modelos computacionales basados ​​en redes neuronales convolucionales representan el estado del arte para la detección automática de RD utilizando imágenes de fondo de ojo. La mayor parte del trabajo actual aborda este problema como una tarea de clasificación binaria. Sin embargo, incluir la estimación de leyes y la cuantificación de la incertidumbre de las predicciones puede aumentar potencialmente la solidez del modelo. En este artículo, se presenta un método de proceso híbrido de aprendizaje profundo y gaussiano para el diagnóstico de RD y la cuantificación de la incertidumbre. Este método combina el poder de representación del aprendizaje profundo con la capacidad de generalizar a partir de pequeños conjuntos de datos de modelos de procesos gaussianos. Los resultados muestran que la cuantificación de la incertidumbre en las predicciones mejora la interpretabilidad del método como herramienta de apoyo al diagnóstico

Diabetic retinopathy (DR) is one of the microvascular complications of diabetes mellitus, which remains a leading cause of blindness worldwide. Computational models based on convolutional neural networks represent the state of the art for automatic detection of DR using fundus images. Most of the current work addresses this problem as a binary classification task. However, including law estimation and quantification of prediction uncertainty can potentially increase model robustness. In this paper, a hybrid deep learning and Gaussian process method for DR diagnosis and uncertainty quantification is presented. This method combines the representational power of deep learning with the ability to generalize from small data sets of Gaussian process models. The results show that the quantification of uncertainty in the predictions improves the interpretability of the method as a diagnostic support tool.

Translated with www.DeepL.com/Translator (free version)