Robust Causal Discovery Under Imperfect Structural Constraints
Zidong Wang, Xi Lin, Chuchao He, Xiaoguang Gao
Abstract
Robust causal discovery from observational data under imperfect prior knowledge remains a significant and largely unresolved challenge. Existing methods typically presuppose perfect priors or can only handle specific, pre-identified error types. And their performance degrades substantially when confronted with flawed constraints of unknown location and type. This decline arises because most of them rely on inflexible and biased thresholding strategies that may conflict with the data distribution. To overcome these limitations, we propose to harmonizes knowledge and data through prior alignment and conflict resolution. First, we assess the credibility of imperfect structural constraints through a surrogate model, which then guides a sparse penalization term measuring the loss between the learned and constrained adjacency matrices. We theoretically prove that, under ideal assumption, the knowledge-driven objective aligns with the data-driven objective. Furthermore, to resolve conflicts when this assumption is violated, we introduce a multi-task learning framework optimized via multi-gradient descent, jointly minimizing both objectives. Our proposed method is robust to both linear and nonlinear settings. Extensive experiments, conducted under diverse noise conditions and structural equation model types, demonstrate the effectiveness and efficiency of our method under imperfect structural constraints.
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 4de0c6da-7d9e-4a37-bb17-90585f88892bBuilds on20
- Gradient-Based Neural DAG LearningSébastien Lachapelle, Philippe Brouillard, Tristan Deleu, Simon Lacoste-JulienICLR 2020 · 337 citations
- On the Role of Sparsity and DAG Constraints for Learning Linear DAGsIgnavier Ng, AmirEmad Ghassami, Kun ZhangNeurIPS 2020 · 306 citations
- Differentiable Causal Discovery from Interventional DataPhilippe Brouillard, Sébastien Lachapelle, Alexandre Lacoste, Simon Lacoste-Julien et al.NeurIPS 2020 · 295 citations
- Causal Discovery with Reinforcement LearningShengyu Zhu, Ignavier Ng, Zhitang ChenICLR 2020 · 285 citations
- DAGMA: Learning DAGs via M-matrices and a Log-Determinant Acyclicity CharacterizationKevin Bello, Bryon Aragam, Pradeep RavikumarNeurIPS 2022 · 222 citations
Related papers
- Structure Learning from Time-Series Data with Lag-Agnostic Structural PriorTaiyu Ban, Changxin Rong, Xiangyu Wang, Lyuzhou Chen et al.ICLR 2026
- Pattern-Guided Adaptive Prior for Structure LearningLyuzhou Chen, Yijia Sun, Yanze Gao, Xiangyu Wang et al.NeurIPS 2025
- GRACE-C: Generalized Rate Agnostic Causal Estimation via ConstraintsMohammadsajad Abavisani, David Danks, Sergey M. PlisICLR 2023
- Hybrid Top-Down Global Causal Discovery with Local Search for Linear and Nonlinear Additive Noise ModelsSujai Hiremath, Jacqueline R. M. A. Maasch, Mengxiao Gao, Promit Ghosal et al.NeurIPS 2024 · 11 citations
- Causal Discovery for Irregularly Time Series with Consistency GuaranteesWeihong Li, Baohong Li, Anpeng Wu, Zhihan Li et al.ICML 2026
