English

PRCD-MAP: Learning How Much to Trust Imperfect Priors in Causal Discovery

Machine Learning 2026-05-08 v2 Machine Learning Methodology

Abstract

External priors of unknown reliability create a brittle trade-off in causal discovery: blind trust amplifies errors, blind rejection wastes signal. Real priors are also heterogeneously reliable -- physical laws are trustworthy, LLM-suggested edges are speculative -- yet existing methods either ignore priors or impose them through globally uniform trust. We propose PRCD-MAP, a soft prior-consumption layer that assigns per-edge trust to an imperfect prior and uses it to modulate a prior-aware 1\ell_1 and prior-weighted 2\ell_2 regularizer in a MAP objective. Trust is calibrated by empirical Bayes on a Laplace-approximated marginal likelihood and propagated along the prior graph by an MLP, so data-confirmed neighborhoods boost trust and contradictions suppress it. PRCD-MAP enjoys a population-level safety guarantee: it is ε\varepsilon-safe in expectation over the prior-generation distribution, with εCacc(1acc)d2/T\varepsilon\leq C\cdot\mathrm{acc}(1{-}\mathrm{acc})\cdot d^2/T at the parametric T1T^{-1} rate and vanishing at the prior-quality endpoints. When the prior is uninformative, learned trust provably collapses to its floor and the method recovers a no-prior baseline. Empirically, on real CausalTime data PRCD-MAP exploits informative LLM priors (LLM-prior gain +0.067/+0.089+0.067/+0.089 AUROC on AQI/Medical over a no-prior PRCD-MAP backbone; combined backbone+prior lead +0.123/+0.043+0.123/+0.043 over PCMCI+), auto-attenuates on the anonymous-variable Traffic stress test, and retains a lead at d=300d{=}300; against BayesDAG, the closest soft-Bayesian baseline, PRCD-MAP wins on every CausalTime dataset under a matched W0W_0-only protocol. A four-way ablation isolates each component: EB calibration and MLP trust propagation jointly carry the plurality of the gain, with positive sign on every dataset. Extensions to nonlinear (NAM) and cross-sectional settings show the calibrated-trust principle is setting-agnostic.

Keywords

Cite

@article{arxiv.2605.01669,
  title  = {PRCD-MAP: Learning How Much to Trust Imperfect Priors in Causal Discovery},
  author = {Xihang Shan and Da Zhou},
  journal= {arXiv preprint arXiv:2605.01669},
  year   = {2026}
}
R2 v1 2026-07-01T12:47:08.229Z