HyperAIHyperAI

Command Palette

Search for a command to run...

Outstanding Paper AwardAAAI 2026

딥러닝

이론적 점수 분석을 통한 동적 시스템의 인과 구조 학습

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 MMM is defined as

L(T,M)=L(M)+∑i=1DL(Xi[t1:tN]∣Pai,Fi),L(\mathcal{T}, M) = L(M) + \sum_{i=1}^D L(X_i^{[t_1:t_N]} \mid Pa_i, F_i),L(T,M)=L(M)+i=1∑D​L(Xi[t1​:tN​]​∣Pai​,Fi​),

where L(M)L(M)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\nu_iνi​, since the deterministic part is captured by the model.

The model cost L(M)L(M)L(M) is split into a global part and a sum of local per-variable costs. The global cost Lglobal(M)L_{\text{global}}(M)Lglobal​(M) stores the initial sss samples of the trajectory and the integrator step size sss, requiring log⁡N\log NlogN bits for sss and a fixed per-sample cost. For each variable XiX_iXi​, the local model MiM_iMi​ stores its parents PaiPa_iPai​ and the parameters of the structural equation FiF_iFi​. The cost of storing the set of parents is

L(Mi)=LN(∥Pai∥)+∥Pai∥log⁡D+LF(Fi),L(M_i) = L_{\mathbb{N}}(\lVert Pa_i \rVert) + \lVert Pa_i \rVert \log D + L_F(F_i),L(Mi​)=LN​(∥Pai​∥)+∥Pai​∥logD+LF​(Fi​),

where LNL_{\mathbb{N}}LN​ is the MDL-optimal encoding for nonnegative integers, and LF(Fi)L_F(F_i)LF​(Fi​) is the cost of describing the dynamics function.

To encode the function FiF_iFi​, 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⋆)),F_i(X_i^{t_\star} \mid Pa_i^{[t_1:t_\star)}) \sim \mathcal{N}\big(\mu_{\text{post}}(X_i^{t_\star}), \Sigma_{\text{post}}(X_i^{t_\star})\big),Fi​(Xit⋆​​∣Pai[t1​:t⋆​)​)∼N(μpost​(Xit⋆​​),Σpost​(Xit⋆​​)),

from which the trajectory can be predicted by numerical integration. The cost LF(Fi)L_F(F_i)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\hat{\sigma}_i^2σ^i2​, the code length for each noise sequence is

L(νi)=N2(1ln⁡2+log⁡2πσ^i2),L(\nu_i) = \frac{N}{2}\left(\frac{1}{\ln 2} + \log 2\pi \hat{\sigma}_i^2\right),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\Gamma_{ij}Γij​ for each possible directed edge Xi→XjX_i \to X_jXi​→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(N3D3log⁡D)\mathcal{O}(N^3 D^3 \log D)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.


AI로 AI 구축

아이디어에서 출시까지 — 무료 AI 코코딩, 즉시 사용 가능한 환경, 최적의 GPU 가격으로 AI 개발을 가속화하세요.

AI 협업 코딩
바로 사용 가능한 GPU
최적의 가격

HyperAI Newsletters

최신 정보 구독하기
한국 시간 매주 월요일 오전 9시 에 이번 주의 최신 업데이트를 메일로 발송합니다
이메일 서비스 제공: MailChimp