arXiv:2601.05474cs.LGcs.AI2026-01

用稀疏低秩分解构建可靠超结构,加速因果发现优化

Efficient Differentiable Causal Discovery via Reliable Super-Structure Learning

  • 通过稀疏低秩分解学习数据精度矩阵,提取关键因果成分
  • 生成的超结构是真实因果图的超集,显著缩小搜索空间
  • 适用于高维、有隐混杂变量场景,提升效率与准确率

近期可微分因果发现方法虽提升了准确性与效率,但在高维数据或存在隐混杂变量时,仍面临搜索空间过大、目标函数复杂及图论约束难处理等问题。为此,本文提出ALVGL,一种通用增强框架。该方法通过稀疏低秩分解学习数据精度矩阵,并设计ADMM算法识别对因果结构最相关的成分,据此构建一个可证明为真实因果图超集的超结构。该超结构用于初始化标准可微分因果发现方法,大幅缩小搜索空间,提升优化效率与准确率。我们在多种结构因果模型(包括高斯与非高斯设置,含与不含未测量混杂因子)上验证了ALVGL的泛化能力。在合成与真实数据集上的实验表明,ALVGL不仅达到当前最优准确率,还显著提升优化效率,成为可微分因果发现中可靠有效的解决方案。

原文摘要 · Abstract (English)

Recently, differentiable causal discovery has emerged as a promising approach to improve the accuracy and efficiency of existing methods. However, when applied to high-dimensional data or data with latent confounders, these methods, often based on off-the-shelf continuous optimization algorithms, struggle with the vast search space, the complexity of the objective function, and the nontrivial nature of graph-theoretical constraints. As a result, there has been a surge of interest in leveraging super-structures to guide the optimization process. Nonetheless, learning an appropriate super-structure at the right level of granularity, and doing so efficiently across various settings, presents significant challenges. In this paper, we propose ALVGL, a novel and general enhancement to the differentiable causal discovery pipeline. ALVGL employs a sparse and low-rank decomposition to learn the precision matrix of the data. We design an ADMM procedure to optimize this decomposition, identifying components in the precision matrix that are most relevant to the underlying causal structure. These components are then combined to construct a super-structure that is provably a superset of the true causal graph. This super-structure is used to initialize a standard differentiable causal discovery method with a more focused search space, thereby improving both optimization efficiency and accuracy. We demonstrate the versatility of ALVGL by instantiating it across a range of structural causal models, including both Gaussian and non-Gaussian settings, with and without unmeasured confounders. Extensive experiments on synthetic and real-world datasets show that ALVGL not only achieves state-of-the-art accuracy but also significantly improves optimization efficiency, making it a reliable and effective solution for differentiable causal discovery.

因果发现可微分超结构优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。