LGMLJun 21, 2025

Learning Time-Aware Causal Representation for Model Generalization in Evolving Domains

arXiv:2506.17718v23 citationsh-index: 11ICML
Originality Incremental advance
AI Analysis

This work addresses model generalization in evolving domains for real-world deployment, representing an incremental improvement by focusing on causal factors to mitigate spurious correlations.

The paper tackles the problem of evolving domain generalization (EDG) by addressing spurious correlations in dynamic scenarios, proposing a time-aware structural causal model and SYNC method that learns causal representations, achieving superior temporal generalization performance on synthetic and real-world datasets.

Endowing deep models with the ability to generalize in dynamic scenarios is of vital significance for real-world deployment, given the continuous and complex changes in data distribution. Recently, evolving domain generalization (EDG) has emerged to address distribution shifts over time, aiming to capture evolving patterns for improved model generalization. However, existing EDG methods may suffer from spurious correlations by modeling only the dependence between data and targets across domains, creating a shortcut between task-irrelevant factors and the target, which hinders generalization. To this end, we design a time-aware structural causal model (SCM) that incorporates dynamic causal factors and the causal mechanism drifts, and propose \textbf{S}tatic-D\textbf{YN}amic \textbf{C}ausal Representation Learning (\textbf{SYNC}), an approach that effectively learns time-aware causal representations. Specifically, it integrates specially designed information-theoretic objectives into a sequential VAE framework which captures evolving patterns, and produces the desired representations by preserving intra-class compactness of causal factors both across and within domains. Moreover, we theoretically show that our method can yield the optimal causal predictor for each time domain. Results on both synthetic and real-world datasets exhibit that SYNC can achieve superior temporal generalization performance.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes