CheXmask-U: Quantifying uncertainty in landmark-based anatomical segmentation for X-ray images

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Cosarinsky, Matias, Gaggion, Nicolas, Echeveste, Rodrigo, Ferrante, Enzo
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914340965187584
author Cosarinsky, Matias
Gaggion, Nicolas
Echeveste, Rodrigo
Ferrante, Enzo
author_facet Cosarinsky, Matias
Gaggion, Nicolas
Echeveste, Rodrigo
Ferrante, Enzo
contents In this work, we study uncertainty estimation for anatomical landmark-based segmentation on chest X-rays. Inspired by hybrid neural network architectures that combine standard image convolutional encoders with graph-based generative decoders, and leveraging their variational latent space, we derive two complementary measures: (i) latent uncertainty, captured directly from the learned distribution parameters, and (ii) predictive uncertainty, obtained by generating multiple stochastic output predictions from latent samples. Through controlled corruption experiments we show that both uncertainty measures increase with perturbation severity, reflecting both global and local degradation. We demonstrate that these uncertainty signals can identify unreliable predictions by comparing with manual ground-truth, and support out-of-distribution detection on the CheXmask dataset. More importantly, we release CheXmask-U (huggingface.co/datasets/mcosarinsky/CheXmask-U), a large scale dataset of 657,566 chest X-ray landmark segmentations with per-node uncertainty estimates, enabling researchers to account for spatial variations in segmentation quality when using these anatomical masks. Our findings establish uncertainty estimation as a promising direction to enhance robustness and safe deployment of landmark-based anatomical segmentation methods in chest X-ray. A fully working interactive demo of the method is available at huggingface.co/spaces/matiasky/CheXmask-U and the source code at github.com/mcosarinsky/CheXmask-U.
format Preprint
id arxiv_https___arxiv_org_abs_2512_10715
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle CheXmask-U: Quantifying uncertainty in landmark-based anatomical segmentation for X-ray images
Cosarinsky, Matias
Gaggion, Nicolas
Echeveste, Rodrigo
Ferrante, Enzo
Computer Vision and Pattern Recognition
In this work, we study uncertainty estimation for anatomical landmark-based segmentation on chest X-rays. Inspired by hybrid neural network architectures that combine standard image convolutional encoders with graph-based generative decoders, and leveraging their variational latent space, we derive two complementary measures: (i) latent uncertainty, captured directly from the learned distribution parameters, and (ii) predictive uncertainty, obtained by generating multiple stochastic output predictions from latent samples. Through controlled corruption experiments we show that both uncertainty measures increase with perturbation severity, reflecting both global and local degradation. We demonstrate that these uncertainty signals can identify unreliable predictions by comparing with manual ground-truth, and support out-of-distribution detection on the CheXmask dataset. More importantly, we release CheXmask-U (huggingface.co/datasets/mcosarinsky/CheXmask-U), a large scale dataset of 657,566 chest X-ray landmark segmentations with per-node uncertainty estimates, enabling researchers to account for spatial variations in segmentation quality when using these anatomical masks. Our findings establish uncertainty estimation as a promising direction to enhance robustness and safe deployment of landmark-based anatomical segmentation methods in chest X-ray. A fully working interactive demo of the method is available at huggingface.co/spaces/matiasky/CheXmask-U and the source code at github.com/mcosarinsky/CheXmask-U.
title CheXmask-U: Quantifying uncertainty in landmark-based anatomical segmentation for X-ray images
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2512.10715