U2-Net: A Bayesian U-Net model with epistemic uncertainty feedback for photoreceptor layer segmentation in pathological OCT scans
U2-Net: A Bayesian U-Net model with epistemic uncertainty feedback for photoreceptor layer segmentation in pathological OCT scans
José Ignacio Orlando Philipp Seeböck Hrvoje Bogunović Sophie Klinscha Christoph Grechenig Sebastian Waldstein Bianca S. Gerendas Ursula Schmidt-Erfurth

Abstract
In this paper, we introduce a Bayesian deep learning based model for segmenting the photoreceptor layer in pathological OCT scans. Our architecture provides accurate segmentations of the photoreceptor layer and produces pixel-wise epistemic uncertainty maps that highlight potential areas of pathologies or segmentation errors. We empirically evaluated this approach in two sets of pathological OCT scans of patients with age-related macular degeneration, retinal vein oclussion and diabetic macular edema, improving the performance of the baseline U-Net both in terms of the Dice index and the area under the precision/recall curve. We also observed that the uncertainty estimates were inversely correlated with the model performance, underlying its utility for highlighting areas where manual inspection/correction might be needed.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| image-matting-on-aim-500 | U2NET | Conn.: 82.14 Grad.: 51.02 MAD: 0.0493 MSE: 0.0348 SAD: 83.46 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.