针对开放集域泛化中的标签噪声问题,提出新方法提升模型鲁棒性。
Mitigating Label Noise using Prompt-Based Hyperbolic Meta-Learning in Open-Set Domain Generalization

- 用双曲原型与可学习提示进行元学习,感知标签噪声
- 在PACS和DigitsDG数据集上显著优于现有方法
- 适合处理真实场景中含噪声的开放集识别任务
开放集域泛化(OSDG)要求模型在未见领域中准确预测已知类别,同时对未知类别保持低置信度以有效拒绝。尽管该领域已有显著进展,但真实数据集中普遍存在的标签噪声影响却鲜受关注。标签噪声会误导模型优化,加剧未知领域的识别挑战。本文首次系统研究开放集域泛化下的标签噪声问题(OSDG-NL),基于PACS和DigitsDG构建专用基准测试集,评估现有去噪与OSDG方法的局限性。为此,我们提出HyProMeta框架:结合双曲类别原型实现噪声感知的元学习,并引入可学习的新类别无关提示,增强对未见类别的泛化能力。大量实验表明,相较于现有先进方法,HyProMeta在新建立的基准上表现更优。代码已开源。
原文摘要 · Abstract (English)
Open-Set Domain Generalization (OSDG) is a challenging task requiring models to accurately predict familiar categories while minimizing confidence for unknown categories to effectively reject them in unseen domains. While the OSDG field has seen considerable advancements, the impact of label noise--a common issue in real-world datasets--has been largely overlooked. Label noise can mislead model optimization, thereby exacerbating the challenges of open-set recognition in novel domains. In this study, we take the first step towards addressing Open-Set Domain Generalization under Noisy Labels (OSDG-NL) by constructing dedicated benchmarks derived from widely used OSDG datasets, including PACS and DigitsDG. We evaluate baseline approaches by integrating techniques from both label denoising and OSDG methodologies, highlighting the limitations of existing strategies in handling label noise effectively. To address these limitations, we propose HyProMeta, a novel framework that integrates hyperbolic category prototypes for label noise-aware meta-learning alongside a learnable new-category agnostic prompt designed to enhance generalization to unseen classes. Our extensive experiments demonstrate the superior performance of HyProMeta compared to state-of-the-art methods across the newly established benchmarks. The source code of this work is released at https://github.com/KPeng9510/HyProMeta.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。