Learning Discrete Latent Variable Structures with Tensor Rank Conditions
Zhengming Chen, Ruichu Cai, Feng Xie, Jie Qiao, Anpeng Wu, Zijian Li, Zhifeng Hao, Kun Zhang
Abstract
Unobserved discrete data are ubiquitous in many scientific disciplines, and how to learn the causal structure of these latent variables is crucial for uncovering data patterns. Most studies focus on the linear latent variable model or impose strict constraints on latent structures, which fail to address cases in discrete data involving non-linear relationships or complex latent structures. To achieve this, we explore a tensor rank condition on contingency tables for an observed variable set , showing that the rank is determined by the minimum support of a specific conditional set (not necessary in ) that d-separates all variables in . By this, one can locate the latent variable through probing the rank on different observed variables set, and further identify the latent causal structure under some structure assumptions. We present the corresponding identification algorithm and conduct simulated experiments to verify the effectiveness of our method. In general, our results elegantly extend the identification boundary for causal discovery with discrete latent variables and expand the application scope of causal discovery with latent variables.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 64bfb186-be82-471f-b4b3-ef96f091a971Cited by top-tier papers4
- Distributional Equivalence in Linear Non-Gaussian Latent-Variable Cyclic Causal Models: Characterization and LearningHaoyue Dai, Immanuel Albrecht, Peter Spirtes, Kun ZhangICLR 2026 · 4 citations
- On the Identifiability of Poisson Branching Structural Causal Model Under Latent ConfoundingJie Qiao, Zihuai Zeng, Ruichu Cai, Zhengming Chen et al.ICML 2026
- Conditional Independent Component Analysis for Estimating Causal Structure with Latent VariablesYewei Xia, Zhengming Chen, Haoyue Dai, Fuhong Wang et al.ICLR 2026
- Identification of Latent Confounders via Investigating the Tensor Ranks of the Nonlinear ObservationsZhengming Chen, Yewei Xia, Feng Xie, Jie Qiao et al.ICML 2025
Builds on6
- Generalized Independent Noise Condition for Estimating Latent Variable Causal GraphsFeng Xie, Ruichu Cai, Biwei Huang, Clark Glymour et al.NeurIPS 2020 · 119 citations
- Latent Hierarchical Causal Structure Discovery with Rank ConstraintsBiwei Huang, Charles Jia Han Low, Feng Xie, Clark Glymour et al.NeurIPS 2022 · 78 citations
- Identification of Linear Non-Gaussian Latent Hierarchical StructureFeng Xie, Biwei Huang, Zhengming Chen, Yangbo He et al.ICML 2022 · 65 citations
- Identification of Partially Observed Linear Causal Models: Graphical Conditions for the Non-Gaussian and Heterogeneous CasesJeffrey Adams, Niels Richard Hansen, Kun ZhangNeurIPS 2021 · 61 citations
- Identification of Linear Latent Variable Model with Arbitrary DistributionZhengming Chen, Feng Xie, Jie Qiao, Zhifeng Hao et al.AAAI 2022 · 24 citations
Related papers
- A Versatile Causal Discovery Framework to Allow Causally-Related Hidden VariablesXinshuai Dong, Biwei Huang, Ignavier Ng, Xiangchen Song et al.ICLR 2024 · 29 citations
- Latent Variable Causal Discovery under Selection BiasHaoyue Dai, Yiwen Qiu, Ignavier Ng, Xinshuai Dong et al.ICML 2025
- Identifying Partially Observed Causal Models from Heterogeneous/Nonstationary DataXinshuai Dong, Haoyue Dai, Ignavier Ng, Peter Spirtes et al.ICML 2026
- Horizontal and Vertical Federated Causal Structure Learning via Higher-order CumulantsWei Chen, Wanyang Gu, Linjun Peng, Ting Yan et al.AAAI 2026
- Causal Structure Recovery with Latent Variables under Milder Distributional and Graphical AssumptionsXiu-Chuan Li, Kun Zhang, Tongliang LiuICLR 2024 · 5 citations
