English

Test-Time Learning of Causal Structure from Interventional Data

Machine Learning 2026-02-24 v1 Artificial Intelligence

Abstract

Supervised causal learning has shown promise in causal discovery, yet it often struggles with generalization across diverse interventional settings, particularly when intervention targets are unknown. To address this, we propose TICL (Test-time Interventional Causal Learning), a novel method that synergizes Test-Time Training with Joint Causal Inference. Specifically, we design a self-augmentation strategy to generate instance-specific training data at test time, effectively avoiding distribution shifts. Furthermore, by integrating joint causal inference, we developed a PC-inspired two-phase supervised learning scheme, which effectively leverages self-augmented training data while ensuring theoretical identifiability. Extensive experiments on bnlearn benchmarks demonstrate TICL's superiority in multiple aspects of causal discovery and intervention target detection.

Keywords

Cite

@article{arxiv.2602.19131,
  title  = {Test-Time Learning of Causal Structure from Interventional Data},
  author = {Wei Chen and Rui Ding and Bojun Huang and Yang Zhang and Qiang Fu and Yuxuan Liang and Han Shi and Dongmei Zhang},
  journal= {arXiv preprint arXiv:2602.19131},
  year   = {2026}
}
R2 v1 2026-07-01T10:46:11.375Z