arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.03608cs.LGcs.CVstat.ME

Mixstyle-Entropy:基于因果干预与扰动的领域泛化

Mixstyle-Entropy: Domain Generalization with Causal Intervention and Perturbation

Xiamen University(厦门大学);The University of Hong Kong(香港大学)

Luyao Tang, Yuxuan Yuan, Chaoqi Chen, Xinghao Ding, Yue Huang

AI总结:

提出基于因果的框架InPer,通过训练时基于熵的因果干预优化因果变量选择,测试时利用因果扰动计算稳态分数构建原型分类器,有效提升了模型在跨域任务中的泛化能力。

摘要:

尽管深度神经网络取得了巨大进展,但当测试环境偏离训练环境时,其性能往往会下降。领域泛化(DG)通过学习独立于领域相关信息的表示来解决这一问题,从而促进向未见环境的泛化。现有方法通常侧重于制定量身定制的训练目标,从源数据中提取共享特征。然而,训练和测试过程的脱节可能会损害鲁棒性,尤其是在部署期间面临不可预见的变化时。在本文中,我们提出了一个基于因果关系的全新整体框架,名为InPer,旨在通过在训练期间结合因果干预和在测试期间结合因果扰动来增强模型泛化能力。具体而言,在训练阶段,我们采用基于熵的因果干预(EnIn)来优化因果变量的选择。为了从目标域中识别具有抗干扰因果变量的样本,我们通过因果扰动(HoPer)提出了一种新指标——稳态分数,以在测试时构建原型分类器。跨多个跨域任务的实验结果证实了InPer的有效性。

英文摘要:

Despite the considerable advancements achieved by deep neural networks, their performance tends to degenerate when the test environment diverges from the training ones. Domain generalization (DG) solves this issue by learning representations independent of domain-related information, thus facilitating extrapolation to unseen environments. Existing approaches typically focus on formulating tailored training objectives to extract shared features from the source data. However, the disjointed training and testing procedures may compromise robustness, particularly in the face of unforeseen variations during deployment. In this paper, we propose a novel and holistic framework based on causality, named InPer, designed to enhance model generalization by incorporating causal intervention during training and causal perturbation during testing. Specifically, during the training phase, we employ entropy-based causal intervention (EnIn) to refine the selection of causal variables. To identify samples with anti-interference causal variables from the target domain, we propose a novel metric, homeostatic score, through causal perturbation (HoPer) to construct a prototype classifier in test time. Experimental results across multiple cross-domain tasks confirm the efficacy of InPer.

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

↑