利用不自信的无标签数据提升半监督域泛化性能
Unlocking the Potential of Unlabeled Data in Semi-Supervised Domain Generalization
- 将不自信无标签样本作为负样本构建对比学习
- 通过混淆类别集生成不自信样本的正样本对
- 无需领域标签,可无缝集成到现有方法中
我们研究半监督域泛化(SSDG)问题,即训练与测试数据分布不同,且仅少量标注数据和大量无标注数据可用。现有方法仅利用模型预测置信度高的无标签样本,限制了无标签数据的充分利用。据我们所知,首次探索了在SSDG中引入此前被忽略的不自信无标签样本。为此,提出UPCSC方法,包含两个模块:1)无标签代理对比学习(UPC),将不自信无标签样本视为额外负样本;2)代理类别学习(SC),利用其混淆类别集生成不自信无标签样本的正样本对。两个模块可即插即用,无需领域标签,能轻松集成至现有方法。在四个常用SSDG基准上的实验表明,该方法在基线基础上持续提升性能,优于其他即插即用方法。分析显示,该方法增强了类别判别力并缓解了域间差异。代码已公开于https://github.com/dongkwani/UPCSC。
原文摘要 · Abstract (English)
We address the problem of semi-supervised domain generalization (SSDG), where the distributions of train and test data differ, and only a small amount of labeled data along with a larger amount of unlabeled data are available during training. Existing SSDG methods that leverage only the unlabeled samples for which the model's predictions are highly confident (confident-unlabeled samples), limit the full utilization of the available unlabeled data. To the best of our knowledge, we are the first to explore a method for incorporating the unconfident-unlabeled samples that were previously disregarded in SSDG setting. To this end, we propose UPCSC to utilize these unconfident-unlabeled samples in SSDG that consists of two modules: 1) Unlabeled Proxy-based Contrastive learning (UPC) module, treating unconfident-unlabeled samples as additional negative pairs and 2) Surrogate Class learning (SC) module, generating positive pairs for unconfident-unlabeled samples using their confusing class set. These modules are plug-and-play and do not require any domain labels, which can be easily integrated into existing approaches. Experiments on four widely used SSDG benchmarks demonstrate that our approach consistently improves performance when attached to baselines and outperforms competing plug-and-play methods. We also analyze the role of our method in SSDG, showing that it enhances class-level discriminability and mitigates domain gaps. The code is available at https://github.com/dongkwani/UPCSC.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。