arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2103.16050cs.CV
收藏

用于单领域泛化的渐进式领域扩展网络

Progressive Domain Expansion Network for Single Domain Generalization

University of Chinese Academy of Sciences(中国科学院大学)

Lei Li, Ke Gao, Juan Cao, Ziyao Huang, Yepeng Weng, Xiaoyue Mi, Zhengze Yu, Xiaoya Li, Boyang xia

摘要:

单领域泛化是模型泛化的一个具有挑战性的案例,其中模型在单个领域上进行训练,并在其他未知领域上进行测试。一个有前途的解决方案是通过扩大训练域的覆盖范围来学习跨域不变表示。由于缺乏适当的安全性和有效性约束,这些方法在实际应用中的泛化性能增益有限。在本文中,我们提出了一种新的学习框架称为渐进域扩展网络(PDEN)的单领域泛化。PDEN中的域扩展子网络和表示学习子网络通过联合学习实现了优势互补。对于域扩展子网络,逐步生成多个域,以模拟看不见的域中的各种光度和几何变换。为了保证扩展域的安全性和有效性,提出了一系列的策略。对于域不变表示学习子网络,引入对比学习方法,学习类聚类较好的域不变表示,从而学习到较好的决策边界,提高其泛化能力。大量的分类和分割实验表明,PDEN可以实现高达15.28%的改善相比,最先进的单领域泛化方法。

英文摘要:

Single domain generalization is a challenging case of model generalization, where the models are trained on a single domain and tested on other unseen domains. A promising solution is to learn cross-domain invariant representations by expanding the coverage of the training domain. These methods have limited generalization performance gains in practical applications due to the lack of appropriate safety and effectiveness constraints. In this paper, we propose a novel learning framework called progressive domain expansion network (PDEN) for single domain generalization. The domain expansion subnetwork and representation learning subnetwork in PDEN mutually benefit from each other by joint learning. For the domain expansion subnetwork, multiple domains are progressively generated in order to simulate various photometric and geometric transforms in unseen domains. A series of strategies are introduced to guarantee the safety and effectiveness of the expanded domains. For the domain invariant representation learning subnetwork, contrastive learning is introduced to learn the domain invariant representation in which each class is well clustered so that a better decision boundary can be learned to improve it's generalization. Extensive experiments on classification and segmentation have shown that PDEN can achieve up to 15.28% improvement compared with the state-of-the-art single-domain generalization methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑