Casual Inference via Style Bias Deconfounding for Domain Generalization
By: Jiaxi Li , Di Lin , Hao Chen and more
Potential Business Impact:
Teaches computers to ignore style, see true patterns.
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.
Similar Papers
ConstStyle: Robust Domain Generalization with Unified Style Transformation
CV and Pattern Recognition
Helps AI learn from different kinds of pictures.
Decentralized Domain Generalization with Style Sharing: Formal Model and Convergence Analysis
Machine Learning (CS)
Helps phones learn from different users' styles.
Advancing Cross-Organ Domain Generalization with Test-Time Style Transfer and Diversity Enhancement
CV and Pattern Recognition
Helps AI see diseases in different medical images.