Command Palette
Search for a command to run...
이론적 점수 분석을 통한 동적 시스템의 인과 구조 학습
이론적 점수 분석을 통한 동적 시스템의 인과 구조 학습
Nicholas Tagliapietra Katharina Ensinger Christoph Zimmer Osman Mian
초록
실제 세계의 시스템은 근본적인 인과 관계에 따라 연속 시간으로 진화하지만, 그 역학은 종종 알려져 있지 않습니다. 이러한 역학을 학습하는 기존 접근법은 일반적으로 시간을 이산화하거나(불규칙하게 샘플링된 데이터에서 성능 저하를 초래) 근본적인 인과성을 무시합니다. 우리는 이러한 두 가지 문제를 모두 해결하는 동적 시스템의 인과 발견을 위한 새로운 방법인 CADYT를 제안합니다. 문제를 이산 시간 동적 베이지안 네트워크로 모델링하는 최신 인과 발견 방법과 달리, 우리의 공식은 시스템의 연속적인 특성을 모델링하기 위한 더 완화된 가정을 허용하는 차분 기반 인과 모델에 기반을 둡니다. CADYT는 기본 동적 과정과 더 잘 일치하는 연속 시간 역학을 모델링하기 위해 정확한 가우시안 프로세스 추론을 활용합니다. 우리는 알고리즘적 마르코프 조건과 최소 설명 길이 원칙에 의해 안내되는 탐욕 검색을 통해 인과 구조를 식별하는 실용적인 구현을 제안합니다. 우리의 실험은 CADYT가 정규 및 불규칙하게 샘플링된 데이터 모두에서 최신 방법보다 우수하며, 실제 기본 역학에 더 가까운 인과 네트워크를 발견함을 보여줍니다.
One-sentence Summary
Researchers from Bosch Center for Artificial Intelligence, TU Darmstadt, Baden-Wuerttemberg Cooperative State University Mannheim, and IKIM propose CADYT, a causal discovery method for dynamical systems that leverages difference-based causal models and exact Gaussian Process inference, outperforming discrete-time Dynamic Bayesian network approaches on both regularly and irregularly sampled data by identifying causal structures closer to the true underlying dynamics via greedy search guided by the Algorithmic Markov Condition and Minimum Description Length principle.
Key Contributions
- CADYT is a causal discovery method for continuous-time dynamical systems that models the problem using Difference-based causal models instead of discrete-time Dynamic Bayesian networks, enabling milder assumptions about the continuous nature of the system.
- CADYT combines exact Gaussian Process inference for continuous-time dynamics with a greedy structural search guided by the Algorithmic Markov Condition and the Minimum Description Length principle; its score is proven to be a valid regularized log-likelihood score with an asymptotic upper bound similar to the BIC.
- Experiments show that CADYT outperforms state-of-the-art causal discovery methods on both regularly and irregularly sampled trajectory data, recovering causal networks closer to the true underlying dynamics.
Introduction
Real-world physical systems evolve through continuous-time dynamics with intrinsic causal mechanisms, yet learning these dynamics from data poses a significant challenge. In systems like a mass-spring network, the position of each mass causally influences the velocities of connected masses, highlighting the need to model directionality in causal relationships for physically plausible predictions. While differential equations are the standard mathematical tool for such systems, inferring them from observed data remains difficult, and data-driven approaches such as neural networks or Gaussian processes offer no guarantees that learned dynamics will respect true causality, often absorbing spurious correlations instead.
Prior work falls short in two complementary ways. State-of-the-art causal discovery methods for time series typically recover only discrete-time causal structures, assume regular sampling of trajectories, and are unable to handle irregularly sampled data. Conversely, continuous-time dynamical system models can adapt to irregular sampling but ignore causality entirely, offering no guarantees of causal generalization. These two challenges, causal structure identification and continuous-time modeling, have only ever been tackled in isolation.
The authors address both jointly. They propose CADYT, a causal discovery framework for continuous-time dynamical systems that relaxes the regular sampling assumption while simultaneously recovering the true causal structure. Their method builds on Dynamic Structural Causal Models and leverages a Gaussian Process-based inference scheme for exact continuous-time modeling, even under irregular sampling. To identify causal parents, they introduce a novel score based on the Algorithmic Markov Condition, which selects causes as those variables providing the simplest description in terms of Kolmogorov complexity. Since Kolmogorov complexity is uncomputable, the authors upper bound it via the Minimum Description Length principle and minimize the resulting score through structure search, yielding an end-to-end algorithm that learns both the dynamics and the underlying causal graph.
Method
The authors propose a score-based causal discovery method for multivariate continuous-valued dynamical systems, relying on the framework of Dynamic Structural Causal Models (DSCMs). Under the assumption of dynamical stability, the system of ordinary differential equations can be decomposed into per-component structural equations where each effect variable is a function of its direct causes. The goal is to identify the underlying directed graph and learn the dynamics from trajectory data. To this end, they design a Minimum Description Length (MDL) score that trades off model complexity against data fit, and they instantiate it using additive noise models where each observed component is the sum of a deterministic dynamics term and independent Gaussian noise.
The total description length of a candidate model M is defined as
L(T,M)=L(M)+i=1∑DL(Xi[t1:tN]∣Pai,Fi),where L(M) encodes the model structure and parameters, and the second sum measures the compression of each component given its causal parents and the learned structural equation. Under the additive noise assumption, this reduces to encoding the residual noises νi, since the deterministic part is captured by the model.
The model cost L(M) is split into a global part and a sum of local per-variable costs. The global cost Lglobal(M) stores the initial s samples of the trajectory and the integrator step size s, requiring logN bits for s and a fixed per-sample cost. For each variable Xi, the local model Mi stores its parents Pai and the parameters of the structural equation Fi. The cost of storing the set of parents is
L(Mi)=LN(∥Pai∥)+∥Pai∥logD+LF(Fi),where LN is the MDL-optimal encoding for nonnegative integers, and LF(Fi) is the cost of describing the dynamics function.
To encode the function Fi, the authors use a Gaussian process regression scheme based on multi-step integrators. This choice is motivated by two factors: GPs provide a natural continuous model for ODEs and handle irregularly sampled trajectories well, and their nonparametric nature avoids imposing parametric assumptions. For each variable, a GP is conditioned on the trajectory of its parents, yielding a posterior distribution over the dynamics function. The kernel used in the GP is derived from the multi-step integrator coefficients, allowing exact inference. Formally, the estimated dynamics function follows
Fi(Xit⋆∣Pai[t1:t⋆))∼N(μpost(Xit⋆),Σpost(Xit⋆)),from which the trajectory can be predicted by numerical integration. The cost LF(Fi) is determined by the complexity of the kernel matrix and its parameters. The kernel matrix is stored via singular value decomposition, and the rotation angles are quantized to a predefined precision. The length scales and variances are encoded using the parameter encoding scheme of Marx and Vreeken, which accounts for the sign, magnitude, and precision of each parameter.
The residual noise after model fitting is encoded as zero-mean Gaussian noise. Given an empirical variance estimate σ^i2, the code length for each noise sequence is
L(νi)=2N(ln21+log2πσ^i2),which follows the optimal encoding for Gaussian residuals. This term rewards models that explain most of the variance in the data, since models with smaller residuals yield shorter descriptions.
The authors provide theoretical guarantees for their score. Under two additional assumptions, namely that the kernel is finite-dimensional and that hyperparameters and precision values are bounded, they show that the total score is asymptotically upper bounded by a regularized log-likelihood score akin to the Bayesian Information Criterion. The proof relies on showing that the eigenvalue encoding cost scales sublinearly with the number of trajectory points. This result establishes that minimizing the MDL score is equivalent to performing model selection via a valid penalized likelihood criterion.
Because an exhaustive search over all possible graphs is intractable, the authors incorporate their score into a greedy three-step search procedure. The first step, edge ranking, computes the gain Γij for each possible directed edge Xi→Xj as the difference in description length before and after adding that edge. A priority queue of edges is built, ordered by decreasing gain. The forward search then iteratively adds the highest-scoring edge, updating the scores of all incoming edges to the affected node. Each edge is tested for statistical significance using the no-hypercompression inequality, and the process continues until no further improvement is obtained. A subsequent backward phase removes any edge whose deletion improves the score. The final output is the learned causal graph. The overall computational complexity is O(N3D3logD), dominated by the Gaussian process regression, but the algorithm is designed to be parallelizable, keeping runtime practical for typical dynamical systems.
Experiment
The evaluation compares CADYT against constraint, score, and noise based baselines on synthetic dynamical systems with varying sizes, cyclicities, and sampling regularity, using structural distance and edge confidence metrics. Across all settings, CADYT consistently outperforms baselines by maintaining high precision and avoiding spurious edges, while higher-order integrators improve robustness, especially on irregular timelines and with cyclic dynamics. Experiments on oscillators and chaotic systems confirm that CADYT recovers causal structures more accurately and with higher confidence than competing methods.
CADYT consistently achieves the highest AUPRC across all three dynamical systems, outperforming DYNOTEARS, PCMCI+, and VARLINGAM. The baselines show relatively low performance, with DYNOTEARS and PCMCI+ exhibiting high false positive rates. CADYT's high AUPRC indicates that its high-confidence causal edges are mostly correct. CADYT leads by a clear margin on all simulated and real dynamical systems. Baselines like DYNOTEARS and PCMCI+ produce many spurious edges, lowering their AUPRC. CADYT maintains high precision on causal edges it is most confident about.
In experiments across three dynamical systems, CADYT consistently achieves the highest AUPRC, clearly outperforming baselines like DYNOTEARS, PCMCI+, and VARLINGAM. The baselines suffer from high false positive rates, producing many spurious edges, whereas CADYT's high AUPRC indicates that its most confident causal edges are predominantly correct.