Proceedings

TCCD: Tree-guided Continuous Causal Discovery via Collaborative MCTS-Parameter Optimization

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. AI4Tech: AI Enabling Technologies. Pages 9268-9276.

https://doi.org/10.24963/ijcai.2025/1030

Abstract

Learning causal relationships in directed acyclic graphs (DAGs) from multi-type event sequences is a challenging task, especially in large-scale telecommunication networks. Existing methods struggle with the exponentially growing search space and lack global exploration. Gradient-based approaches are limited by their reliance on local information and often fail to generalize. To address these issues, we propose TCCD, a framework that combines Monte Carlo Tree Search (MCTS) with continuous gradient optimization. TCCD balances global exploration and local optimization, overcoming the shortcomings of purely gradient-based methods and enhancing generalization. By unifying various causal structure learning approaches, TCCD offers a scalable and efficient solution for causal inference in complex networks. Extensive experiments validate its superior performance on both synthetic and real-world datasets. Code and Appendix are available at https://github.com/jzephyrl/TCCD.

  1. [Achab et al., 2018] Massil Achab, Emmanuel Bacry, Stéphane Gaïffas, Iacopo Mastromatteo, and Jean- Franc ¸ois Muzy. Uncovering causality from multivariate hawkes integrated cumulants. Journal of Machine Learning Research, 18(192):1-28, 2018.
  2. [Bhattacharjya et al., 2022] Debarun Bhattacharjya, Karthikeyan Shanmugam, Tian Gao, and Dharmashankar Subramanian. Process independence testing in proxi- mal graphical event models. In Conference on Causal Learning and Reasoning, pages 144-161. PMLR, 2022.
  3. [Cai et al., 2022] Ruichu Cai, Siyu Wu, Jie Qiao, Zhifeng Hao, Keli Zhang, and Xi Zhang. Thps: Topological hawkes processes for learning causal structure on event sequences. IEEE Transactions on Neural Networks and Learning Systems, 2022.
  4. [Cheng et al., 2023] Yuxiao Cheng, Runzhao Yang, Tingx- iong Xiao, Zongren Li, Jinli Suo, Kunlun He, and Qiong- hai Dai. Cuts: Neural causal discovery from irregular time-series data. arXiv preprint arXiv:2302.07458, 2023.
  5. [Chickering, 1996] David Maxwell Chickering. Learning bayesian networks is np-complete. Learning from data: Artificial intelligence and statistics V, pages 121-130, 1996.
  6. [Chickering, 2002] David Maxwell Chickering. Optimal structure identification with greedy search. Journal of ma- chine learning research, 3(Nov):507-554, 2002.
  7. [Idé et al., 2021] Tsuyoshi Idé, Georgios Kollias, Dzung Phan, and Naoki Abe. Cardinality-regularized hawkes- granger model. Advances in Neural Information Process- ing Systems, 34:2682-2694, 2021.
  8. [Itô, 1971] Kiyosi Itô. Poisson point processes attached to markov processes. In Proc. 6th Berk. Symp. Math. Stat. Prob, volume 3, pages 225-240, 1971.
  9. [Jin et al., 2022] Kebing Jin, Hankz Hankui Zhuo, Zhanhao Xiao, Hai Wan, and Subbarao Kambhampati. Gradient- based mixed planning with symbolic and numeric action parameters. Artif. Intell., 313:103789, 2022.
  10. [Kocsis and Szepesvári, 2006] Levente Kocsis and Csaba Szepesvári. Bandit based monte-carlo planning. In Eu- ropean conference on machine learning, pages 282-293. Springer, 2006.
  11. [Lachapelle et al., 2019] Sébastien Lachapelle, Philippe Brouillard, Tristan Deleu, and Simon Lacoste-Julien. Gradient-based neural dag learning. arXiv preprint arXiv:1906.02226, 2019.
  12. [Li et al., 2022] Wenqian Li, Yinchuan Li, Shengyu Zhu, Yunfeng Shao, Jianye Hao, and Yan Pang. Gflowcausal: Generative flow networks for causal discovery. arXiv preprint arXiv:2210.08185, 2022.
  13. [Li et al., 2024] Mingjia Li, Shuo Liu, Hong Qian, and Aimin Zhou. A simple yet scalable granger causal struc- tural learning approach for topological event sequences. In The Thirty-eighth Annual Conference on Neural Informa- tion Processing Systems, 2024.
  14. [Liniger, 2009] Thomas Liniger. Multivariate hawkes pro- cesses. PhD thesis, ETH Zurich, 2009.
  15. [Liu et al., 2024] Yuequn Liu, Ruichu Cai, Wei Chen, Jie Qiao, Yuguang Yan, Zijian Li, Keli Zhang, and Zhifeng Hao. Tnpar: Topological neural poisson auto-regressive model for learning granger causal structure from event se- quences. In AAAI, pages 20491-20499, 2024.
  16. [Qiao et al., 2023] Jie Qiao, Ruichu Cai, Siyu Wu, Yu Xiang, Keli Zhang, and Zhifeng Hao. Structural hawkes processes for learning causal structure from discrete-time event se- quences. arXiv preprint arXiv:2305.05986, 2023. IJCAI
  17. [Runge et al., 2019] Jakob Runge, Peer Nowack, Marlene Kretschmer, Seth Flaxman, and Dino Sejdinovic. Detect- ing and quantifying causal associations in large nonlinear time series datasets. Science advances, 5(11):eaau4996, 2019.
  18. [Shojaie and Fox, 2022] Ali Shojaie and Emily B Fox. Granger causality: A review and recent advances. An- nual Review of Statistics and Its Application, 9(1):289- 319, 2022.
  19. [Spirtes and Glymour, 1991] Peter Spirtes and Clark Gly- mour. An algorithm for fast recovery of sparse causal graphs. Social science computer review, 9(1):62-72, 1991.
  20. [Spirtes et al., 2000] Pater Spirtes, Clark Glymour, Richard Scheines, Stuart Kauffman, Valerio Aimale, and Frank Wimberly. Constructing bayesian network models of gene expression networks from microarray data. 2000.
  21. [Spirtes et al., 2001] Peter Spirtes, Clark Glymour, and Richard Scheines. Causation, prediction, and search. 2001.
  22. [Tsamardinos et al., 2006] Ioannis Tsamardinos, Laura E Brown, and Constantin F Aliferis. The max-min hill- climbing bayesian network structure learning algorithm. Machine learning, 65:31-78, 2006.
  23. [Wang et al., 2021] Xiaoqiang Wang, Yali Du, Shengyu Zhu, Liangjun Ke, Zhitang Chen, Jianye Hao, and Jun Wang. Ordering-based causal discovery with reinforce- ment learning. arXiv preprint arXiv:2105.06631, 2021. IJCAI
  24. [Xu et al., 2016] Hongteng Xu, Mehrdad Farajtabar, and Hongyuan Zha. Learning granger causality for hawkes processes. In International conference on machine learn- ing, pages 1717-1726. PMLR, 2016.
  25. [Zhang et al., 2020] Wei Zhang, Thomas Panum, Somesh Jha, Prasad Chalasani, and David Page. Cause: Learning granger causality from event sequences using attribution methods. In International Conference on Machine Learn- ing, pages 11235-11245. PMLR, 2020.
  26. [Zhang et al., 2021] Keli Zhang, Shengyu Zhu, Marcus Ka- lander, Ignavier Ng, Junjian Ye, Zhitang Chen, and Lujia Pan. gcastle: A python toolbox for causal discovery. arXiv preprint arXiv:2111.15155, 2021.
  27. [Zheng et al., 2018] Xun Zheng, Bryon Aragam, Pradeep K Ravikumar, and Eric P Xing. Dags with no tears: Con- tinuous optimization for structure learning. In NeurIPS, 2018.
  28. [Zhou et al., 2013] Ke Zhou, Hongyuan Zha, and Le Song. Learning social infectivity in sparse low-rank networks us- ing multi-dimensional hawkes processes. In Artificial in- telligence and statistics, pages 641-649, 2013.
  29. [Zhu et al., 2019] Shengyu Zhu, Ignavier Ng, and Zhitang Chen. Causal discovery with reinforcement learning. arXiv preprint arXiv:1906.04477, 2019.
  30. [Zhuo et al., 2011] Hankz Hankui Zhuo, Qiang Yang, Rong Pan, and Lei Li. Cross-domain action-model acquisition for planning via web search. In ICAPS, 2011.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards