Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Learning Temporal Causal Structure via Smooth Differentiable Optimization

About

Causal discovery with instantaneous effects in multivariate time series is challenging, as the instantaneous structure must be acyclic. Prior methods enforce this by either separating instantaneous and lagged estimation into multi-stage pipelines or imposing algebraic acyclicity constraints via complex augmented Lagrangian optimization, both of which incur high computational cost. In this work, we propose a different approach: we learn a differentiable permutation of variables using the Gumbel--Sinkhorn operator and triangularize the instantaneous coefficient matrix of a Structural Vector Autoregressive (SVAR) model in the learned order. This converts acyclicity from a hard constraint into a parameterization and keeps it valid throughout optimization. In doing so, our method enables unified, continuous optimization with gradient-based learning, leading to improved efficiency in time--series causal discovery. Across three real-world benchmarks, our method achieves the best overall performance compared with 12 baselines in both discovery accuracy and efficiency. On the large-scale benchmark, it further demonstrates strong scalability, achieving more than a 6x speedup over competing methods.

Tong Zhao, Ce Guo, Wayne Luk, Emil Lupu, Ray Dipojjwal• 2026

Related benchmarks

TaskDatasetResultRank
Causal DiscoverySWaT
F1 Score22.02
25
Causal DiscoveryFlood CausalRiver
F1 Score32.26
22
Causal DiscoveryWeb 2
Runtime (s)8.7547
17
Causal DiscoveryFlood
Runtime (s)13.6563
16
Causal DiscoverySWaT
Runtime (s)16.4386
15
Causal DiscoveryIT Monitoring
MoM 1 Score0.4
13
Causal DiscoveryWeb 1
Runtime (s)9.97
13
Causal DiscoveryMoM 1
Runtime (s)7.25
13
Causal DiscoveryMoM 2
Runtime (s)5.62
13
Causal DiscoveryStorm
Runtime (s)6.99
13
Showing 10 of 16 rows

Other info

Follow for update