arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.17067cs.CV
收藏

CADG:一种基于交叉注意的领域泛化模型

CADG: A Model Based on Cross Attention for Domain Generalization

Kuaishou Technology(快手科技);Sichuan University(四川大学)

Cheng Dai, Yingqiao Lin, Fan Li, Xiyao Li, Donglin Xie

摘要:

在领域泛化(DG)任务中,模型只使用来自源域的训练数据来训练,以实现对未知目标域的泛化,这将受到分布偏移问题的影响。因此,重要的是要学习一个分类器,专注于共同的表示,可以用来分类多个领域,使这个分类器可以实现高性能的一个看不见的目标领域以及。随着交叉注意在各种跨模态任务中的成功应用,我们发现交叉注意是一种强大的机制,可以将来自不同分布的特征对齐。因此,我们设计了一个称为CADG(交叉注意领域泛化)的模型,其中交叉注意起着重要的作用,以解决分布偏移问题。这样的设计使得分类器可以在多个领域中使用,从而使分类器在未知领域中具有很好的泛化能力。实验结果表明,该方法在各种领域泛化基准测试中的性能均优于其他单一模型,甚至优于一些基于集成的方法.

英文摘要:

In Domain Generalization (DG) tasks, models are trained by using only training data from the source domains to achieve generalization on an unseen target domain, this will suffer from the distribution shift problem. So it's important to learn a classifier to focus on the common representation which can be used to classify on multi-domains, so that this classifier can achieve a high performance on an unseen target domain as well. With the success of cross attention in various cross-modal tasks, we find that cross attention is a powerful mechanism to align the features come from different distributions. So we design a model named CADG (cross attention for domain generalization), wherein cross attention plays a important role, to address distribution shift problem. Such design makes the classifier can be adopted on multi-domains, so the classifier will generalize well on an unseen domain. Experiments show that our proposed method achieves state-of-the-art performance on a variety of domain generalization benchmarks compared with other single model and can even achieve a better performance than some ensemble-based methods.

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

↑