arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2201.09042cs.CVcs.LG

Uncertainty-aware deep learning methods for robust diabetic retinopathy classification

Department of Computer Science, Aalto University(阿尔托大学计算机科学系);The Alan Turing Institute(艾伦图灵研究所);University of Warwick(华威大学);University College London(伦敦大学学院);Department of Electrical Engineering and Automation, Aalto University(阿尔托大学电气工程与自动化系);Central Finland Health Care District(芬兰中部医疗保健区)

Joel Jaskari, Jaakko Sahlsten, Theodoros Damoulas, Jeremias Knoblauch, Simo Särkkä, Leo Kärkkäinen, Kustaa Hietala, Kimmo Kaski

英文摘要:

Automatic classification of diabetic retinopathy from retinal images has been widely studied using deep neural networks with impressive results. However, there is a clinical need for estimation of the uncertainty in the classifications, a shortcoming of modern neural networks. Recently, approximate Bayesian deep learning methods have been proposed for the task but the studies have only considered the binary referable/non-referable diabetic retinopathy classification applied to benchmark datasets. We present novel results by systematically investigating a clinical dataset and a clinically relevant 5-class classification scheme, in addition to benchmark datasets and the binary classification scheme. Moreover, we derive a connection between uncertainty measures and classifier risk, from which we develop a new uncertainty measure. We observe that the previously proposed entropy-based uncertainty measure generalizes to the clinical dataset on the binary classification scheme but not on the 5-class scheme, whereas our new uncertainty measure generalizes to the latter case.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑