Oriel — AI & technology worth noticing, every day.

Pseudo-Label Foreground Quality Weighting for Domain-Generalized Fundus Image Segmentation

Paper · https://arxiv.org/abs/2610.23911

arXiv:2610.23911v1 Announce Type: new Abstract: Optic disc (OD) and optic cup (OC) segmentation is central to automated glaucoma assessment, but models trained on limited labeled fundus datasets often degrade under domain shift. Semi-supervised Teacher-Student learning can improve generalization by leveraging unlabeled data, yet standard Noisy Student training treats all pseudo-labels as equally reliable. In this paper we investigate a targeted extension of Noisy Student training in which OD- and OC-specific pseudo-label supervision is modulated using a mask-derived reliability estimate. A multitask Dice predictor estimates OD- and OC-specific pseudo-label quality from Teacher-generated masks, and these estimates are used to weight the pseudo-label loss during Student training. This preserves the baseline contribution of low-scoring foreground pseudo-labels while assigning greater relative emphasis to pseudo-labels predicted to be more reliable. Experiments across known and unseen domains show improvements over the Noisy Student baseline, particularly for OC segmentation, while maintaining competitive generalization performance.

First seen by Oriel on .

Aliases

SourceExternal IDURL
arXiv AI Papersoai:arXiv.org:2610.23911v1https://arxiv.org/abs/2610.23911