Stochastic variational deep kernel learning based diabetic retinopathy severity grading

Marlin Siebert*, Nikolay Tesmer, Philipp Rostalski

*Corresponding author for this work
1 Citation (Scopus)

Abstract

The retinal disease Diabetic retinopathy (DR) is one of the most probable causes of blindness. Automatic detection of DR is mostly done using convolutional neural networks (CNNs) on colour retinal images. This work in contrast uses stochastic variational deep kernel learning (SVDKL) for DR grading, combining a deep CNN with Gaussian processes (GPs) into a single end-to-end trainable model, which promises to provide predictions with a reliable uncertainty estimate exploiting approximate Bayesian inference. Evaluating the performance and uncertainty calibration of SVDKL on DR grading compared to a plain CNN, the EfficientNet-B0, preliminary results on a subset of the Kaggle DR dataset show a naturally enhanced uncertainty calibration for SVDKL over the plain CNN as well as a good diagnostic performance. Despite SVDKL achieving a slightly reduced accuracy, incorrect predictions were in closer proximity to the target stages, which is beneficial for clinical diagnosis due to minimizing the cost related to severe misclassifications.

Original languageEnglish
JournalCurrent Directions in Biomedical Engineering
Volume8
Issue number2
Pages (from-to)408-411
Number of pages4
ISSN2364-5504
DOIs
Publication statusPublished - 01.08.2022

Research Areas and Centers

  • Academic Focus: Biomedical Engineering

DFG Research Classification Scheme

  • 409-06 Information Systems, Process and Knowledge Management

Cite this