用贝叶斯方法给视网膜OCT图像分割加不确定性评估,提升临床可信度。
Bayesian Deep Learning Approaches for Uncertainty-Aware Retinal OCT Image Segmentation for Multiple Sclerosis
- 采用贝叶斯卷积网络生成分割结果和不确定性图
- 整体Dice分数达95.65%,可识别噪声或校准错误的异常样本
- 适合需要可信诊断依据的神经科与眼科医生使用
光学相干断层扫描(OCT)因其高分辨率、横断面视网膜图像,在眼科、心脏病学和神经病学中具有重要价值。视网膜层的精确分割对眼科医生至关重要,但人工操作耗时且易受主观偏差影响。以往深度学习方法因缺乏不确定性估计,常出现‘自信错误’的幻觉结果,限制了临床采纳。本研究采用贝叶斯卷积神经网络(BCNN)对公开的OCT数据集(含35例健康对照与多发性硬化患者)进行分割。结果表明,该方法可生成分割不确定性图,用于识别存在噪声或校准问题的高不确定样本。同时支持对关键次级指标(如层厚度)的不确定性估计,具有医学意义。相比同类工作,整体Dice得分提升至95.65%,显著增强临床适用性、统计稳健性与性能。
原文摘要 · Abstract (English)
Optical Coherence Tomography (OCT) provides valuable insights in ophthalmology, cardiology, and neurology due to high-resolution, cross-sectional images of the retina. One critical task for ophthalmologists using OCT is delineation of retinal layers within scans. This process is time-consuming and prone to human bias, affecting the accuracy and reliability of diagnoses. Previous efforts to automate delineation using deep learning face challenges in uptake from clinicians and statisticians due to the absence of uncertainty estimation, leading to "confidently wrong" models via hallucinations. In this study, we address these challenges by applying Bayesian convolutional neural networks (BCNNs) to segment an openly available OCT imaging dataset containing 35 human retina OCTs split between healthy controls and patients with multiple sclerosis. Our findings demonstrate that Bayesian models can be used to provide uncertainty maps of the segmentation, which can further be used to identify highly uncertain samples that exhibit recording artefacts such as noise or miscalibration at inference time. Our method also allows for uncertainty-estimation for important secondary measurements such as layer thicknesses, that are medically relevant for patients. We show that these features come in addition to greater performance compared to similar work over all delineations; with an overall Dice score of 95.65%. Our work brings greater clinical applicability, statistical robustness, and performance to retinal OCT segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。