用部分标注数据训练多疾病眼底病筛查模型,抗标签缺失与域偏移。
PSScreen V2: Partially Supervised Multiple Retinal Disease Screening
- 三分支架构:教师生成伪标签,学生用低频扰动增强鲁棒性
- 在多个眼底数据集上达最优性能,跨域泛化能力显著提升
- 适配DINOv2等主流模型,可扩展至胸部X光等其他医学影像
本文提出PSScreen V2,一种用于多疾病眼底病筛查的部分监督自训练框架。不同于依赖全标注或单一领域数据集的以往方法,PSScreen V2能从多个标注不全、分布各异的数据集中学习,同时应对标签缺失与域偏移问题。其采用三分支结构:教师分支通过弱增强图像生成伪标签以补全缺失标签;两个学生分支引入创新特征增强策略——低频丢弃(LF-Dropout),随机剔除与域相关的低频成分以提升域鲁棒性;以及低频不确定性(LF-Uncert),通过对抗学习的高斯扰动估计低频统计的不确定域变异性。在多个内部与外部域的眼底数据集上的大量实验表明,PSScreen V2实现当前最优性能,并具备出色的域泛化能力。此外,与多种主干网络(包括视觉基础模型DINOv2)的兼容性测试,以及在胸部X光数据集上的评估,进一步验证了该框架的通用性与适应性。代码已开源:https://github.com/boyiZheng99/PSScreen_V2。
原文摘要 · Abstract (English)
In this work, we propose PSScreen V2, a partially supervised self-training framework for multiple retinal disease screening. Unlike previous methods that rely on fully labelled or single-domain datasets, PSScreen V2 is designed to learn from multiple partially labelled datasets with different distributions, addressing both label absence and domain shift challenges. To this end, PSScreen V2 adopts a three-branch architecture with one teacher and two student networks. The teacher branch generates pseudo labels from weakly augmented images to address missing labels, while the two student branches introduce novel feature augmentation strategies: Low-Frequency Dropout (LF-Dropout), which enhances domain robustness by randomly discarding domain-related low-frequency components, and Low-Frequency Uncertainty (LF-Uncert), which estimates uncertain domain variability via adversarially learned Gaussian perturbations of low-frequency statistics. Extensive experiments on multiple in-domain and out-of-domain fundus datasets demonstrate that PSScreen V2 achieves state-of-the-art performance and superior domain generalization ability. Furthermore, compatibility tests with diverse backbones, including the vision foundation model DINOv2, as well as evaluations on chest X-ray datasets, highlight the universality and adaptability of the proposed framework. The codes are available at https://github.com/boyiZheng99/PSScreen_V2.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。