acceptodds
Under review as a conference paper at ICLR 2027

CAUTION: Knowing When to Trust LLMs for Ensemble Causal Discovery

Abstract

Causal discovery from observational data remains challenging due to the fundamental limitations of purely statistical methods, such as statistical distinguishability within equivalence classes and sensitivity to finite sample sizes. While large language models (LLMs) offer a promising source of domain knowledge to complement statistical inference, existing LLM-augmented methods are vulnerable to LLM errors. A central challenge is therefore deciding when to trust LLM knowledge and when to trust statistical evidence. We propose CauTion, a trust-calibrated ensemble framework that addresses this challenge through annotation-free reliability estimation and selective arbitration. CauTion proceeds in three stages. First, an algorithm ensemble utilizes a consensus voting to resolve up to 96% of edges on which algorithms agree, achieving near-perfect accuracy on the filtered consensus edges. Second, a trust-calibrated arbitration mechanism estimates the relative reliability of the LLM and the algorithms via an annotation-free trust calibration procedure, which is then utilized to govern a trust-weighted voting process that restricts LLM arbitration exclusively to edges with unreliable algorithmic evidence. Third, a cycle repair step is applied to guarantee the final causal graph is validly acyclic. Experiments on six datasets demonstrate that CauTion consistently outperforms both data-centric and LLM-augmented baselines in SHD and F1, with strong robustness to LLM errors.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.