LGMLOct 28, 2023

Temporally Disentangled Representation Learning under Unknown Nonstationarity

SalesforceStanford
arXiv:2310.18615v227 citationsh-index: 64
Originality Incremental advance
AI Analysis

This addresses the challenge of disentangling causal representations in nonstationary sequential data without relying on observed auxiliary variables, which is incremental as it builds on prior stationary work but extends to more realistic scenarios.

The paper tackled the problem of unsupervised causal representation learning for sequential data with time-delayed latent causal influences in nonstationary settings, showing that independent latent components can be recovered up to permutation and transformation without auxiliary variables, and introduced NCTRL, which substantially outperformed baselines in identifying these influences.

In unsupervised causal representation learning for sequential data with time-delayed latent causal influences, strong identifiability results for the disentanglement of causally-related latent variables have been established in stationary settings by leveraging temporal structure. However, in nonstationary setting, existing work only partially addressed the problem by either utilizing observed auxiliary variables (e.g., class labels and/or domain indexes) as side information or assuming simplified latent causal dynamics. Both constrain the method to a limited range of scenarios. In this study, we further explored the Markov Assumption under time-delayed causally related process in nonstationary setting and showed that under mild conditions, the independent latent components can be recovered from their nonlinear mixture up to a permutation and a component-wise transformation, without the observation of auxiliary variables. We then introduce NCTRL, a principled estimation framework, to reconstruct time-delayed latent causal variables and identify their relations from measured sequential data only. Empirical evaluations demonstrated the reliable identification of time-delayed latent causal influences, with our methodology substantially outperforming existing baselines that fail to exploit the nonstationarity adequately and then, consequently, cannot distinguish distribution shifts.

Code Implementations1 repo
Foundations

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

Your Notes