학술논문

Bayesian deep learning for reliable oral cancer image classification.
Document Type
article
Source
Biomedical Optics Express. 12(10)
Subject
Dental/Oral and Craniofacial Disease
Rare Diseases
Cancer
Basic Behavioral and Social Science
Behavioral and Social Science
Optical Physics
Materials Engineering
Language
Abstract
In medical imaging, deep learning-based solutions have achieved state-of-the-art performance. However, reliability restricts the integration of deep learning into practical medical workflows since conventional deep learning frameworks cannot quantitatively assess model uncertainty. In this work, we propose to address this shortcoming by utilizing a Bayesian deep network capable of estimating uncertainty to assess oral cancer image classification reliability. We evaluate the model using a large intraoral cheek mucosa image dataset captured using our customized device from high-risk population to show that meaningful uncertainty information can be produced. In addition, our experiments show improved accuracy by uncertainty-informed referral. The accuracy of retained data reaches roughly 90% when referring either 10% of all cases or referring cases whose uncertainty value is greater than 0.3. The performance can be further improved by referring more patients. The experiments show the model is capable of identifying difficult cases needing further inspection.