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

领域泛化中领域感知的三元组丢失

Domain-aware Triplet loss in Domain Generalization

University of Queensland(昆士兰大学)

Kaiyu Guo, Brian Lovell

摘要:

尽管随着深度学习的发展,在对象识别领域取得了很大进展,但仍有几个因素对深度学习模型的性能产生负面影响。域偏移是这些因素之一,并且是由测试数据和训练数据的分布的差异引起的。本文研究了领域泛化中的紧凑特征聚类问题,以优化多领域数据的嵌入空间。我们为领域泛化设计了一个领域感知的三元组损失,以帮助模型不仅可以聚类相似的语义特征,而且可以分散来自领域的特征。与以前的方法集中在分布对齐,我们的算法被设计为分散域信息的嵌入空间。其基本思想是基于嵌入特征可以根据域信息进行聚类的假设,这在本文中得到了数学和经验的支持。此外,在我们的探索领域泛化的特征聚类,我们注意到,在领域泛化的度量学习损失的收敛影响因素比预定义的领域更重要。为了解决这个问题,我们利用两种方法来规范化嵌入空间,减少嵌入特征的内部协变量偏移。消融研究证明了我们的算法的有效性。在PACS、VLCS和Bull-Home等基准数据集上的实验表明,该方法优于关注领域差异的相关方法。特别是,我们在RegnetY-16上的结果明显优于基准数据集上的最新方法。我们的代码将在https://github.com/workerbcd/DCT上发布

英文摘要:

Despite much progress being made in the field of object recognition with the advances of deep learning, there are still several factors negatively affecting the performance of deep learning models. Domain shift is one of these factors and is caused by discrepancies in the distributions of the testing and training data. In this paper, we focus on the problem of compact feature clustering in domain generalization to help optimize the embedding space from multi-domain data. We design a domainaware triplet loss for domain generalization to help the model to not only cluster similar semantic features, but also to disperse features arising from the domain. Unlike previous methods focusing on distribution alignment, our algorithm is designed to disperse domain information in the embedding space. The basic idea is motivated based on the assumption that embedding features can be clustered based on domain information, which is mathematically and empirically supported in this paper. In addition, during our exploration of feature clustering in domain generalization, we note that factors affecting the convergence of metric learning loss in domain generalization are more important than the pre-defined domains. To solve this issue, we utilize two methods to normalize the embedding space, reducing the internal covariate shift of the embedding features. The ablation study demonstrates the effectiveness of our algorithm. Moreover, the experiments on the benchmark datasets, including PACS, VLCS and Office-Home, show that our method outperforms related methods focusing on domain discrepancy. In particular, our results on RegnetY-16 are significantly better than state-of-the-art methods on the benchmark datasets. Our code will be released at https://github.com/workerbcd/DCT

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

↑