UP2D: Uncertainty-aware Progressive Pseudo-label Denoising for Source-Free Domain Adaptive Medical Image Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bui-Tran, Quang-Khai, Nguyen, Thanh-Huy, Ho, Manh D., Lam, Thinh B., Vu, Vi, Nguyen, Hoang-Thien, Huynh, Phat, Bagci, Ulas
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917051042365440
author Bui-Tran, Quang-Khai
Nguyen, Thanh-Huy
Ho, Manh D.
Lam, Thinh B.
Vu, Vi
Nguyen, Hoang-Thien
Huynh, Phat
Bagci, Ulas
author_facet Bui-Tran, Quang-Khai
Nguyen, Thanh-Huy
Ho, Manh D.
Lam, Thinh B.
Vu, Vi
Nguyen, Hoang-Thien
Huynh, Phat
Bagci, Ulas
contents Medical image segmentation models face severe performance drops under domain shifts, especially when data sharing constraints prevent access to source images. We present a novel Uncertainty-aware Progressive Pseudo-label Denoising (UP2D) framework for source-free domain adaptation (SFDA), designed to mitigate noisy pseudo-labels and class imbalance during adaptation. UP2D integrates three key components: (i) a Refined Prototype Filtering module that suppresses uninformative regions and constructs reliable class prototypes to denoise pseudo-labels, (ii) an Uncertainty-Guided EMA (UG-EMA) strategy that selectively updates the teacher model based on spatially weighted boundary uncertainty, and (iii) a quantile-based entropy minimization scheme that focuses learning on ambiguous regions while avoiding overconfidence on easy pixels. This single-stage student-teacher framework progressively improves pseudo-label quality and reduces confirmation bias. Extensive experiments on three challenging retinal fundus benchmarks demonstrate that UP2D achieves state-of-the-art performance across both standard and open-domain settings, outperforming prior UDA and SFDA approaches while maintaining superior boundary precision.
format Preprint
id arxiv_https___arxiv_org_abs_2510_26826
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle UP2D: Uncertainty-aware Progressive Pseudo-label Denoising for Source-Free Domain Adaptive Medical Image Segmentation
Bui-Tran, Quang-Khai
Nguyen, Thanh-Huy
Ho, Manh D.
Lam, Thinh B.
Vu, Vi
Nguyen, Hoang-Thien
Huynh, Phat
Bagci, Ulas
Image and Video Processing
Medical image segmentation models face severe performance drops under domain shifts, especially when data sharing constraints prevent access to source images. We present a novel Uncertainty-aware Progressive Pseudo-label Denoising (UP2D) framework for source-free domain adaptation (SFDA), designed to mitigate noisy pseudo-labels and class imbalance during adaptation. UP2D integrates three key components: (i) a Refined Prototype Filtering module that suppresses uninformative regions and constructs reliable class prototypes to denoise pseudo-labels, (ii) an Uncertainty-Guided EMA (UG-EMA) strategy that selectively updates the teacher model based on spatially weighted boundary uncertainty, and (iii) a quantile-based entropy minimization scheme that focuses learning on ambiguous regions while avoiding overconfidence on easy pixels. This single-stage student-teacher framework progressively improves pseudo-label quality and reduces confirmation bias. Extensive experiments on three challenging retinal fundus benchmarks demonstrate that UP2D achieves state-of-the-art performance across both standard and open-domain settings, outperforming prior UDA and SFDA approaches while maintaining superior boundary precision.
title UP2D: Uncertainty-aware Progressive Pseudo-label Denoising for Source-Free Domain Adaptive Medical Image Segmentation
topic Image and Video Processing
url https://arxiv.org/abs/2510.26826