Daily Arxiv

This is a page that curates AI-related papers published worldwide.
All content here is summarized using Google Gemini and operated on a non-profit basis.
Copyright for each paper belongs to the authors and their institutions; please make sure to credit the source when sharing.

Uncertainty-Guided Coarse-to-Fine Tumor Segmentation with Anatomy-Aware Post-Processing

Created by
  • Haebom

Author

Ilkin Sevgi Isler, David Mohaisen, Curtis Lisle, Damla Turgut, Ulas Bagci

Outline

In this paper, we propose an uncertainty-based coarse-to-fine segmentation framework, recognizing that reliable tumor segmentation in chest CT suffers from boundary ambiguity, class imbalance, and anatomical variation. The framework combines full-volume tumor localization with fine segmentation of regions of interest (ROIs), and is enhanced by anatomically-aware postprocessing. A first-stage model generates coarse predictions and performs anatomically-informed filtering based on lung overlap, proximity to the lung surface, and component size. The resulting ROIs are segmented by a second-stage model trained with an uncertainty-aware loss function to improve accuracy and boundary correction in the ambiguous region. Experimental results on private and public datasets demonstrate improved Dice and Hausdorff scores, reduced false positives, and improved spatial interpretability. These results highlight the importance of combining uncertainty modeling and anatomical priors in a cascaded segmentation pipeline to generate robust and clinically meaningful tumor contours. On the Orlando dataset, the proposed framework improves the Dice score of Swin UNETR from 0.4690 to 0.6447, and the reduction in erroneous components is strongly correlated with the improvement in segmentation performance, demonstrating the value of anatomically informed postprocessing.

Takeaways, Limitations

Takeaways:
We demonstrate that a cascaded segmentation pipeline combining uncertainty modeling and anatomical prior information can improve the accuracy and reliability of thoracic CT tumor segmentation.
We demonstrate that anatomically informed postprocessing is effective in reducing false positives and improving spatial interpretability.
Significantly improves the performance of existing models such as Swin UNETR (Dice score 0.4690 -> 0.6447 on the Orlando dataset).
Limitations:
The datasets used are private and public datasets, so specific information is limited.
Further research is needed on the generalization performance of the proposed method.
Further studies are needed to determine its applicability to other types of tumors or other imaging techniques.
👍