用可微分方法优化因果图中的独立性约束,提升小样本下的因果发现效果。
Differentiable Constraint-Based Causal Discovery
- 基于软逻辑和渗流理论构建可微分d-分离得分
- 在小样本下优于传统约束与评分方法
- 适合需要梯度优化的因果建模场景
从观测数据中进行因果发现是人工智能中的基础任务,对决策、预测和干预具有深远影响。尽管已有显著进展,现有方法主要分为基于约束和基于评分两类。基于约束的方法虽严谨但易受小样本影响,基于评分的方法虽灵活却通常放弃显式的条件独立性检验。本文探索第三种路径:通过软逻辑结合渗流理论,构建可微分的d-分离得分,从而实现基于梯度优化的条件独立性约束。实证评估表明,该方法在低样本条件下表现稳健,在真实世界数据集上超越了传统约束与评分基线。代码与数据已公开于 https://github.com/PurdueMINDS/DAGPA。
原文摘要 · Abstract (English)
Causal discovery from observational data is a fundamental task in artificial intelligence, with far-reaching implications for decision-making, predictions, and interventions. Despite significant advances, existing methods can be broadly categorized as constraint-based or score-based approaches. Constraint-based methods offer rigorous causal discovery but are often hindered by small sample sizes, while score-based methods provide flexible optimization but typically forgo explicit conditional independence testing. This work explores a third avenue: developing differentiable $d$-separation scores, obtained through a percolation theory using soft logic. This enables the implementation of a new type of causal discovery method: gradient-based optimization of conditional independence constraints. Empirical evaluations demonstrate the robust performance of our approach in low-sample regimes, surpassing traditional constraint-based and score-based baselines on a real-world dataset. Code and data of the proposed method are publicly available at https://github$.$com/PurdueMINDS/DAGPA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。