Label Uncertainty for Ultrasound Segmentation
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866908497324539904 |
|---|---|
| author | Shivaram, Malini Gare, Gautam Rajendrakumar Hutchins, Laura Duplantis, Jacob Deiss, Thomas Gomes, Thales Nogueira Tran, Thong Patel, Keyur H. Fox, Thomas H Krishnan, Amita Ramanan, Deva DeBoisblanc, Bennett Rodriguez, Ricardo Galeotti, John |
| author_facet | Shivaram, Malini Gare, Gautam Rajendrakumar Hutchins, Laura Duplantis, Jacob Deiss, Thomas Gomes, Thales Nogueira Tran, Thong Patel, Keyur H. Fox, Thomas H Krishnan, Amita Ramanan, Deva DeBoisblanc, Bennett Rodriguez, Ricardo Galeotti, John |
| contents | In medical imaging, inter-observer variability among radiologists often introduces label uncertainty, particularly in modalities where visual interpretation is subjective. Lung ultrasound (LUS) is a prime example-it frequently presents a mixture of highly ambiguous regions and clearly discernible structures, making consistent annotation challenging even for experienced clinicians. In this work, we introduce a novel approach to both labeling and training AI models using expert-supplied, per-pixel confidence values. Rather than treating annotations as absolute ground truth, we design a data annotation protocol that captures the confidence that radiologists have in each labeled region, modeling the inherent aleatoric uncertainty present in real-world clinical data. We demonstrate that incorporating these confidence values during training leads to improved segmentation performance. More importantly, we show that this enhanced segmentation quality translates into better performance on downstream clinically-critical tasks-specifically, estimating S/F oxygenation ratio values, classifying S/F ratio change, and predicting 30-day patient readmission. While we empirically evaluate many methods for exposing the uncertainty to the learning model, we find that a simple approach that trains a model on binarized labels obtained with a (60%) confidence threshold works well. Importantly, high thresholds work far better than a naive approach of a 50% threshold, indicating that training on very confident pixels is far more effective. Our study systematically investigates the impact of training with varying confidence thresholds, comparing not only segmentation metrics but also downstream clinical outcomes. These results suggest that label confidence is a valuable signal that, when properly leveraged, can significantly enhance the reliability and clinical utility of AI in medical imaging. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2508_15635 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Label Uncertainty for Ultrasound Segmentation Shivaram, Malini Gare, Gautam Rajendrakumar Hutchins, Laura Duplantis, Jacob Deiss, Thomas Gomes, Thales Nogueira Tran, Thong Patel, Keyur H. Fox, Thomas H Krishnan, Amita Ramanan, Deva DeBoisblanc, Bennett Rodriguez, Ricardo Galeotti, John Image and Video Processing Artificial Intelligence Computer Vision and Pattern Recognition Machine Learning In medical imaging, inter-observer variability among radiologists often introduces label uncertainty, particularly in modalities where visual interpretation is subjective. Lung ultrasound (LUS) is a prime example-it frequently presents a mixture of highly ambiguous regions and clearly discernible structures, making consistent annotation challenging even for experienced clinicians. In this work, we introduce a novel approach to both labeling and training AI models using expert-supplied, per-pixel confidence values. Rather than treating annotations as absolute ground truth, we design a data annotation protocol that captures the confidence that radiologists have in each labeled region, modeling the inherent aleatoric uncertainty present in real-world clinical data. We demonstrate that incorporating these confidence values during training leads to improved segmentation performance. More importantly, we show that this enhanced segmentation quality translates into better performance on downstream clinically-critical tasks-specifically, estimating S/F oxygenation ratio values, classifying S/F ratio change, and predicting 30-day patient readmission. While we empirically evaluate many methods for exposing the uncertainty to the learning model, we find that a simple approach that trains a model on binarized labels obtained with a (60%) confidence threshold works well. Importantly, high thresholds work far better than a naive approach of a 50% threshold, indicating that training on very confident pixels is far more effective. Our study systematically investigates the impact of training with varying confidence thresholds, comparing not only segmentation metrics but also downstream clinical outcomes. These results suggest that label confidence is a valuable signal that, when properly leveraged, can significantly enhance the reliability and clinical utility of AI in medical imaging. |
| title | Label Uncertainty for Ultrasound Segmentation |
| topic | Image and Video Processing Artificial Intelligence Computer Vision and Pattern Recognition Machine Learning |
| url | https://arxiv.org/abs/2508.15635 |