用对抗记忆库增强单一领域数据,提升模型泛化能力
Single Domain Generalization with Adversarial Memory
- 构建对抗性记忆库生成多样化特征,扩充训练数据
- 在标准基准上达到当前最优性能,显著优于基线方法
- 适合数据稀缺场景,尤其适用于单一领域训练任务
领域泛化(DG)旨在通过多个训练领域数据,使模型能泛化到未见测试领域。但传统DG方法依赖多领域数据,难以应用于数据受限场景。单领域泛化(SDG)则在仅有一个训练领域分布的更现实挑战设置下提出。其主要难点在于训练数据多样性不足,且无法获取未见测试数据分布。为此,本文提出一种基于对抗记忆库的单领域泛化方法,通过记忆库增强训练特征。该网络将训练与测试特征映射至由多样化记忆特征构成的不变子空间,隐式对齐训练与测试域。为保持记忆库多样性和代表性,引入对抗特征生成机制,生成超出训练域分布的特征。实验表明,该方法在标准单领域泛化基准上达到当前最优性能。
原文摘要 · Abstract (English)
Domain Generalization (DG) aims to train models that can generalize to unseen testing domains by leveraging data from multiple training domains. However, traditional DG methods rely on the availability of multiple diverse training domains, limiting their applicability in data-constrained scenarios. Single Domain Generalization (SDG) addresses the more realistic and challenging setting by restricting the training data to a single domain distribution. The main challenges in SDG stem from the limited diversity of training data and the inaccessibility of unseen testing data distributions. To tackle these challenges, we propose a single domain generalization method that leverages an adversarial memory bank to augment training features. Our memory-based feature augmentation network maps both training and testing features into an invariant subspace spanned by diverse memory features, implicitly aligning the training and testing domains in the projected space. To maintain a diverse and representative feature memory bank, we introduce an adversarial feature generation method that creates features extending beyond the training domain distribution. Experimental results demonstrate that our approach achieves state-of-the-art performance on standard single domain generalization benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。