The Library
Uncertainty-aware deep learning methods for robust diabetic retinopathy classification
Tools
Jaskari, Joel, Sahlsten, Jaakko, Damoulas, Theodoros, Knoblauch, Jeremias, Sarkka, Simo, Karkkainen, Leo, Hietala, Kustaa and Kaski, Kimmo K. (2022) Uncertainty-aware deep learning methods for robust diabetic retinopathy classification. IEEE Access, 10 . pp. 76669-76681. doi:10.1109/ACCESS.2022.3192024 ISSN 2169-3536.
|
PDF
WRAP-Uncertainty-aware-deep-learning-methods-robust-diabetic-retinopathy-22.pdf - Published Version - Requires a PDF viewer. Available under License Creative Commons Attribution 4.0. Download (2055Kb) | Preview |
Official URL: http://dx.doi.org/10.1109/ACCESS.2022.3192024
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.
Item Type: | Journal Article | ||||||
---|---|---|---|---|---|---|---|
Subjects: | R Medicine > RE Ophthalmology | ||||||
Divisions: | Faculty of Science, Engineering and Medicine > Science > Computer Science Faculty of Science, Engineering and Medicine > Science > Statistics |
||||||
Library of Congress Subject Headings (LCSH): | Diabetic retinopathy, Diabetic retinopathy -- Classification, Retina -- Diseases -- Diagnosis, Medical screening, Bayesian statistical decision theory -- Data processing, Neural networks (Computer science) | ||||||
Journal or Publication Title: | IEEE Access | ||||||
Publisher: | IEEE | ||||||
ISSN: | 2169-3536 | ||||||
Official Date: | 18 July 2022 | ||||||
Dates: |
|
||||||
Volume: | 10 | ||||||
Page Range: | pp. 76669-76681 | ||||||
DOI: | 10.1109/ACCESS.2022.3192024 | ||||||
Status: | Peer Reviewed | ||||||
Publication Status: | Published | ||||||
Access rights to Published version: | Open Access (Creative Commons) | ||||||
Date of first compliant deposit: | 21 September 2022 | ||||||
Date of first compliant Open Access: | 21 September 2022 | ||||||
RIOXX Funder/Project Grant: |
|
Request changes or add full text files to a record
Repository staff actions (login required)
View Item |
Downloads
Downloads per month over past year