arXiv:2503.16852cs.CVcs.AI2025-03被引 1

通过消除风格干扰,让模型学出更可靠的因果关系。

Casual Inference via Style Bias Deconfounding for Domain Generalization

  • 构建因果模型并用反向门调整策略去除风格干扰
  • 在多领域和单领域任务中均显著提升泛化性能
  • 适合需要可靠跨域推理的医疗图像等关键场景

深度神经网络在分布外数据上表现不佳,限制了其在真实场景中的可靠性。现有领域泛化方法常忽略训练集中风格频率的影响,导致模型捕捉到由风格混淆引起的虚假视觉关联,而非真正的因果表示,从而降低推断可靠性。本文提出风格去混淆因果学习(SDCL),基于领域泛化问题构建结构因果模型(SCM),采用反向门调整策略消除风格影响。在此基础上,设计风格引导专家模块(SGEM)自适应聚类风格分布,捕获全局混淆风格;同时引入反向门因果学习模块(BDCL),在特征提取中进行因果干预,确保全局混淆风格公平融入样本预测,有效降低风格偏差。该框架可无缝集成主流数据增强技术。在多种自然与医学图像识别任务上的大量实验验证其有效性,在多域及更具挑战性的单域泛化场景中均表现优异。

原文摘要 · Abstract (English)

Deep neural networks (DNNs) often struggle with out-of-distribution data, limiting their reliability in diverse realworld applications. To address this issue, domain generalization methods have been developed to learn domain-invariant features from single or multiple training domains, enabling generalization to unseen testing domains. However, existing approaches usually overlook the impact of style frequency within the training set. This oversight predisposes models to capture spurious visual correlations caused by style confounding factors, rather than learning truly causal representations, thereby undermining inference reliability. In this work, we introduce Style Deconfounding Causal Learning (SDCL), a novel causal inference-based framework designed to explicitly address style as a confounding factor. Our approaches begins with constructing a structural causal model (SCM) tailored to the domain generalization problem and applies a backdoor adjustment strategy to account for style influence. Building on this foundation, we design a style-guided expert module (SGEM) to adaptively clusters style distributions during training, capturing the global confounding style. Additionally, a back-door causal learning module (BDCL) performs causal interventions during feature extraction, ensuring fair integration of global confounding styles into sample predictions, effectively reducing style bias. The SDCL framework is highly versatile and can be seamlessly integrated with state-of-the-art data augmentation techniques. Extensive experiments across diverse natural and medical image recognition tasks validate its efficacy, demonstrating superior performance in both multi-domain and the more challenging single-domain generalization scenarios.

领域泛化因果推断风格偏差医学图像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。