Jaskari, Joel;
Sahlsten, Jaakko;
Damoulas, THEODOROS;
Knoblauch, Jeremias;
Sarkka, Simo;
Karkkainen, Leo;
Hietala, Kustaa;
(2022)
Uncertainty-Aware Deep Learning Methods for Robust Diabetic Retinopathy Classification.
IEEE Access
, 10
pp. 76669-76681.
10.1109/ACCESS.2022.3192024.
Preview |
Text
Uncertainty-Aware_Deep_Learning_Methods_for_Robust_Diabetic_Retinopathy_Classification.pdf - Published Version Download (2MB) | Preview |
Abstract
Automatic classification of diabetic retinopathy from retinal images has been increasingly studied using deep neural networks with impressive results. However, there is clinical need for estimating uncertainty in the classifications, a shortcoming of modern neural networks. Recently, approximate Bayesian neural networks (BNNs) have been proposed for this task, but previous studies have only considered the binary referable/non-referable diabetic retinopathy classification applied to benchmark datasets. We present novel results for 9 BNNs by systematically investigating a clinical dataset and 5-class classification scheme, together with benchmark datasets and binary classification scheme. Moreover, we derive a connection between entropy-based uncertainty measure and classifier risk, from which we develop a novel uncertainty measure.We observe that the previously proposed entropy-based uncertainty measure improves performance on the clinical dataset for the binary classification scheme, but not to such an extent as on the benchmark datasets. It improves performance in the clinical 5-class classification scheme for the benchmark datasets, but not for the clinical dataset. Our novel uncertainty measure generalizes to the clinical dataset and to one benchmark dataset. Our findings suggest that BNNs can be utilized for uncertainty estimation in classifying diabetic retinopathy on clinical data, though proper uncertainty measures are needed to optimize the desired performance measure. In addition, methods developed for benchmark datasets might not generalize to clinical datasets.
| Type: | Article |
|---|---|
| Title: | Uncertainty-Aware Deep Learning Methods for Robust Diabetic Retinopathy Classification |
| Open access status: | An open access version is available from UCL Discovery |
| DOI: | 10.1109/ACCESS.2022.3192024 |
| Publisher version: | https://doi.org/10.1109/ACCESS.2022.3192024 |
| Language: | English |
| Additional information: | This work is licensed under a Creative Commons Attribution 4.0 International License. The images or other third-party material in this article are included in the Creative Commons license, unless indicated otherwise in the credit line; if the material is not included under the Creative Commons license, users will need to obtain permission from the license holder to reproduce the material. To view a copy of this license, visit http://creativecommons.org/licenses/by/4.0/ |
| Keywords: | Science & Technology, Technology, Computer Science, Information Systems, Engineering, Electrical & Electronic, Telecommunications, Computer Science, Engineering, Uncertainty, Retinopathy, Diabetes, Neural networks, Training, Measurement uncertainty, Deep learning, Approximate Bayesian neural networks, deep learning, diabetic retinopathy, reject option classification, uncertainty estimation, NEURAL-NETWORKS, RETINAL IMAGES, VALIDATION |
| UCL classification: | UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Maths and Physical Sciences UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Maths and Physical Sciences > Dept of Statistical Science UCL > Provost and Vice Provost Offices > UCL BEAMS UCL |
| URI: | https://discovery.ucl.ac.uk/id/eprint/10153621 |
Archive Staff Only
 |
View Item |
