English

Score-based Greedy Search for Structure Identification of Partially Observed Linear Causal Models

Machine Learning 2026-05-04 v2

Abstract

Identifying the structure of a partially observed causal system is essential to various scientific fields. Recent advances have focused on constraint-based causal discovery to solve this problem, and yet in practice these methods often face challenges related to multiple testing and error propagation. These issues could be mitigated by a score-based method and thus it has raised great attention whether there exists a score-based greedy search method that can handle the partially observed scenario. In this work, we propose the first score-based greedy search method for the identification of structure involving latent variables with identifiability guarantees. Specifically, we propose Generalized N Factor Model and establish the global consistency: the true structure including latent variables can be identified up to the Markov equivalence class by using score. We then design Latent variable Greedy Equivalence Search (LGES), a greedy search algorithm for this class of model with well-defined operators, which search very efficiently over the graph space to find the optimal structure. Our experiments on both synthetic and real-life data validate the effectiveness of our method (code will be publicly available).

Keywords

Cite

@article{arxiv.2510.04378,
  title  = {Score-based Greedy Search for Structure Identification of Partially Observed Linear Causal Models},
  author = {Xinshuai Dong and Ignavier Ng and Haoyue Dai and Jiaqi Sun and Xiangchen Song and Peter Spirtes and Kun Zhang},
  journal= {arXiv preprint arXiv:2510.04378},
  year   = {2026}
}
R2 v1 2026-07-01T06:18:17.105Z