Uncertainty-aware retinal layer segmentation in OCT through probabilistic signed distance functions

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Islam, Mohammad Mohaiminul, de Vente, Coen, Liefers, Bart, Klaver, Caroline, Bekkers, Erik J, Sánchez, Clara I.
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916510826496000
author Islam, Mohammad Mohaiminul
de Vente, Coen
Liefers, Bart
Klaver, Caroline
Bekkers, Erik J
Sánchez, Clara I.
author_facet Islam, Mohammad Mohaiminul
de Vente, Coen
Liefers, Bart
Klaver, Caroline
Bekkers, Erik J
Sánchez, Clara I.
contents In this paper, we present a new approach for uncertainty-aware retinal layer segmentation in Optical Coherence Tomography (OCT) scans using probabilistic signed distance functions (SDF). Traditional pixel-wise and regression-based methods primarily encounter difficulties in precise segmentation and lack of geometrical grounding respectively. To address these shortcomings, our methodology refines the segmentation by predicting a signed distance function (SDF) that effectively parameterizes the retinal layer shape via level set. We further enhance the framework by integrating probabilistic modeling, applying Gaussian distributions to encapsulate the uncertainty in the shape parameterization. This ensures a robust representation of the retinal layer morphology even in the presence of ambiguous input, imaging noise, and unreliable segmentations. Both quantitative and qualitative evaluations demonstrate superior performance when compared to other methods. Additionally, we conducted experiments on artificially distorted datasets with various noise types-shadowing, blinking, speckle, and motion-common in OCT scans to showcase the effectiveness of our uncertainty estimation. Our findings demonstrate the possibility to obtain reliable segmentation of retinal layers, as well as an initial step towards the characterization of layer integrity, a key biomarker for disease progression. Our code is available at \url{https://github.com/niazoys/RLS_PSDF}.
format Preprint
id arxiv_https___arxiv_org_abs_2412_04935
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Uncertainty-aware retinal layer segmentation in OCT through probabilistic signed distance functions
Islam, Mohammad Mohaiminul
de Vente, Coen
Liefers, Bart
Klaver, Caroline
Bekkers, Erik J
Sánchez, Clara I.
Image and Video Processing
Artificial Intelligence
Computer Vision and Pattern Recognition
In this paper, we present a new approach for uncertainty-aware retinal layer segmentation in Optical Coherence Tomography (OCT) scans using probabilistic signed distance functions (SDF). Traditional pixel-wise and regression-based methods primarily encounter difficulties in precise segmentation and lack of geometrical grounding respectively. To address these shortcomings, our methodology refines the segmentation by predicting a signed distance function (SDF) that effectively parameterizes the retinal layer shape via level set. We further enhance the framework by integrating probabilistic modeling, applying Gaussian distributions to encapsulate the uncertainty in the shape parameterization. This ensures a robust representation of the retinal layer morphology even in the presence of ambiguous input, imaging noise, and unreliable segmentations. Both quantitative and qualitative evaluations demonstrate superior performance when compared to other methods. Additionally, we conducted experiments on artificially distorted datasets with various noise types-shadowing, blinking, speckle, and motion-common in OCT scans to showcase the effectiveness of our uncertainty estimation. Our findings demonstrate the possibility to obtain reliable segmentation of retinal layers, as well as an initial step towards the characterization of layer integrity, a key biomarker for disease progression. Our code is available at \url{https://github.com/niazoys/RLS_PSDF}.
title Uncertainty-aware retinal layer segmentation in OCT through probabilistic signed distance functions
topic Image and Video Processing
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.04935