通过扩展数据域和扩大类别边界,提升未知场景下的模型泛化能力。
Domain Expansion and Boundary Growth for Open-Set Single-Source Domain Generalization
- 用背景抑制与风格增强生成新样本,扩充稀缺源域数据。
- 利用边缘图作为额外模态,扩大类间边界,提升未知类别识别能力。
- 适合处理数据少、目标域未知的跨域分类任务。
开放集单源域泛化旨在仅使用单一源域数据,训练出能泛化到存在域偏移和标签偏移的未知目标域的鲁棒模型。源域数据稀缺以及目标域数据分布未知,给域不变特征学习和未知类别识别带来巨大挑战。本文提出一种基于域扩展与边界增长的新方法:通过在源数据上采用背景抑制与风格增强合成新样本,实现域扩展;同时强制模型从合成样本中提取一致知识,以学习域不变信息。此外,训练多二分类器时引入边缘图作为额外模态,实现类间边界的扩大,从而间接拓宽已知与未知类之间的边界。实验表明,该方法在多个跨域图像分类数据集上显著提升性能,达到当前最优水平。
原文摘要 · Abstract (English)
Open-set single-source domain generalization aims to use a single-source domain to learn a robust model that can be generalized to unknown target domains with both domain shifts and label shifts. The scarcity of the source domain and the unknown data distribution of the target domain pose a great challenge for domain-invariant feature learning and unknown class recognition. In this paper, we propose a novel learning approach based on domain expansion and boundary growth to expand the scarce source samples and enlarge the boundaries across the known classes that indirectly broaden the boundary between the known and unknown classes. Specifically, we achieve domain expansion by employing both background suppression and style augmentation on the source data to synthesize new samples. Then we force the model to distill consistent knowledge from the synthesized samples so that the model can learn domain-invariant information. Furthermore, we realize boundary growth across classes by using edge maps as an additional modality of samples when training multi-binary classifiers. In this way, it enlarges the boundary between the inliers and outliers, and consequently improves the unknown class recognition during open-set generalization. Extensive experiments show that our approach can achieve significant improvements and reach state-of-the-art performance on several cross-domain image classification datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。