REVIEW 2 major objections 1 minor 38 references
CauTion: Knowing When to Trust LLMs for Ensemble Causal Discovery
T0 review · 2 major / 1 minor · reviewed 2026-06-28 · grok-4.3
Pith's one-line read CauTion resolves most causal edges by algorithm consensus and applies LLM input only where statistical evidence is weak, after estimating relative reliability without annotations.
desk verdict The consensus stage looks practical, but the annotation-free trust calibration lacks separate ground-truth validation, so the robustness claims rest on an untested proxy. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The trust-calibrated arbitration mechanism, which estimates relative reliability of LLM versus statistical algorithms without ground-truth annotations and uses that estimate to decide arbitration on disputed edges.
What would settle it
A controlled test on held-out data in which the trust calibration is deliberately replaced by random or inverted decisions on disputed edges, producing lower final accuracy than the calibrated version.
Extended reading notes
Core claim
CauTion runs three stages: an algorithm ensemble uses consensus voting to resolve up to 96 percent of edges with near-perfect accuracy; a trust-calibrated arbitration estimates the relative reliability of the LLM versus the algorithms through an annotation-free procedure and restricts LLM input to edges where algorithmic evidence is unreliable; a cycle repair step guarantees the output graph is acyclic. On six datasets the method beats both data-centric and prior LLM-augmented baselines, with larger gains on bigger graphs and robustness to LLM mistakes.
Load-bearing premise
The annotation-free procedure can correctly judge when the LLM is more reliable than the algorithms on a given edge.
Editorial extensions
If this is right
- Consensus among multiple statistical algorithms already settles the large majority of edges at high accuracy.
- LLM knowledge is invoked only on a small fraction of edges, limiting both error propagation and token cost.
- Performance advantage increases with graph size, suggesting the method scales where pure statistical methods weaken.
- The framework remains accurate even when the LLM makes occasional mistakes because arbitration is gated by the reliability estimate.
Reading between the lines
- The same calibration idea could be tested on other knowledge sources beyond LLMs, such as expert rules or simulation outputs.
- If the reliability estimate generalizes across domains, the method could reduce the need for expensive labeled causal benchmarks.
- The approach implicitly treats causal discovery as a per-edge reliability problem rather than a global model selection problem.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The manuscript introduces CauTion, a three-stage framework for causal discovery from observational data. Stage 1 applies consensus voting across an ensemble of statistical algorithms to resolve up to 96% of edges at near-perfect accuracy. Stage 2 uses an annotation-free trust calibration procedure to estimate the relative reliability of the LLM versus the algorithms and restricts LLM arbitration to edges with unreliable algorithmic evidence via trust-weighted voting. Stage 3 applies cycle repair to ensure the output graph is acyclic. Experiments on six datasets are reported to show consistent outperformance over both data-centric and LLM-augmented baselines, with larger gains on larger graphs and robustness to LLM errors. Code is provided at a public GitHub repository.
Significance. If the central claims hold, the work offers a practical route to safely incorporating LLM domain knowledge into causal discovery while mitigating error propagation and token costs through selective arbitration and ensemble consensus. The reproducible code release is a clear strength. The approach could be significant for applications where purely statistical methods leave many edges unresolved within equivalence classes.
major comments (2)
- [Stage 2 (trust-calibrated arbitration)] The annotation-free trust calibration procedure (described in the second stage of the pipeline) is load-bearing for the robustness-to-LLM-errors claim. Because the procedure is explicitly annotation-free, it must rely on an internal proxy (e.g., output consistency or self-reported confidence) to decide when algorithmic evidence is unreliable. The manuscript provides no separate validation—such as a correlation between the calibration scores and actual edge-wise correctness on held-out ground-truth edges—to confirm that the proxy selects the more reliable source. Without this check, the restriction of LLM arbitration cannot be verified and the reported outperformance may not generalize.
- [Experiments section] The experimental claims (outperformance on six datasets, 96% consensus edges with near-perfect accuracy, larger gains on larger graphs) are central to the paper's contribution. However, the manuscript description supplies no quantitative metrics, baseline implementations, error bars, dataset characteristics (node/edge counts, sample sizes), or statistical significance tests. This absence prevents assessment of whether the gains are substantive or whether the consensus accuracy claim holds under the reported conditions.
minor comments (1)
- [Abstract] The abstract would be strengthened by including at least one key quantitative result (e.g., average improvement or consensus accuracy) to support the high-level claims.
Simulated Author's Rebuttal
We thank the referee for the constructive feedback on our manuscript. We address each major comment below and outline revisions to improve clarity and address the concerns.
read point-by-point responses
-
Referee: [Stage 2 (trust-calibrated arbitration)] The annotation-free trust calibration procedure (described in the second stage of the pipeline) is load-bearing for the robustness-to-LLM-errors claim. Because the procedure is explicitly annotation-free, it must rely on an internal proxy (e.g., output consistency or self-reported confidence) to decide when algorithmic evidence is unreliable. The manuscript provides no separate validation—such as a correlation between the calibration scores and actual edge-wise correctness on held-out ground-truth edges—to confirm that the proxy selects the more reliable source. Without this check, the restriction of LLM arbitration cannot be verified and the reported outperformance may not generalize.
Authors: We agree that an explicit check correlating the trust calibration scores with ground-truth edge accuracy would strengthen the robustness claim. The current procedure derives trust scores from output consistency on high-consensus edges without using any ground-truth annotations during calibration. To address the concern, we will add a dedicated validation subsection reporting Pearson correlation between the calibration-derived trust weights and actual per-edge correctness on held-out ground truth for the available datasets. This analysis will be presented after the main results and will not alter the annotation-free nature of the calibration step itself. revision: yes
-
Referee: [Experiments section] The experimental claims (outperformance on six datasets, 96% consensus edges with near-perfect accuracy, larger gains on larger graphs) are central to the paper's contribution. However, the manuscript description supplies no quantitative metrics, baseline implementations, error bars, dataset characteristics (node/edge counts, sample sizes), or statistical significance tests. This absence prevents assessment of whether the gains are substantive or whether the consensus accuracy claim holds under the reported conditions.
Authors: The manuscript contains these elements in Section 4, including Table 1 (dataset node/edge counts and sample sizes), Table 2 (performance metrics with means and standard deviations over five random seeds), and statistical significance via paired t-tests. The 96% consensus figure is accompanied by the corresponding accuracy. We will revise the experimental narrative to explicitly summarize these quantitative results, reference the tables and figures in the text, and add a brief description of baseline implementations for improved readability. revision: partial
Circularity Check
No circularity: procedural pipeline with empirical validation on external datasets
full rationale
The paper presents CauTion as a three-stage procedural framework (consensus voting resolving up to 96% of edges, annotation-free trust calibration for arbitration on remaining edges, and cycle repair). All performance claims, including outperformance on six datasets and robustness to LLM errors, are grounded in external experimental results rather than any derivation, equation, or prediction that reduces by construction to fitted parameters or self-citations inside the method. No load-bearing step equates outputs to inputs via definition, renaming, or internal proxy without independent falsifiability. This is the common case of a self-contained empirical method.
Assumptions & free parameters
assumptions (2)
- domain assumption Statistical causal discovery algorithms produce near-perfect accuracy on edges where they reach consensus via voting
- domain assumption An annotation-free procedure can estimate relative reliability between LLM and algorithms
Cite this review
Pith. "Pith review of CauTion: Knowing When to Trust LLMs for Ensemble Causal Discovery." pith.science (2026). https://pith.science/paper/GRILJRWO
@misc{pith2026260603602,
author = {Pith},
title = {Pith review of: CauTion: Knowing When to Trust LLMs for Ensemble Causal Discovery},
year = {2026},
howpublished = {\url{https://pith.science/paper/GRILJRWO}},
note = {Machine review of arXiv:2606.03602}
}
read the original 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 and incur high token costs. Moreover, reliance on a single data-centric algorithm can make results sensitive to algorithm-specific biases. To address these limitations, we propose CauTion, a framework that reliably integrates LLM domain knowledge into an ensemble of statistical causal discovery algorithms through consensus filtering and LLM reliability estimation. 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, with larger gains on larger graphs and strong robustness to LLM errors. Code is available at https://github.com/OpenCausaLab/CauTion.
Reference graph
Works this paper leans on
-
[1]
The MIT Press, Cambridge, MA, 2nd edition, 2000
Peter Spirtes, Clark Glymour, and Richard Scheines.Causation, Prediction, and Search. The MIT Press, Cambridge, MA, 2nd edition, 2000
2000
-
[2]
Cambridge University Press, 2009
Judea Pearl.Causality. Cambridge University Press, 2009
2009
-
[3]
Basic Books, 2018
Judea Pearl.The book of why: The new science of cause and effect. Basic Books, 2018
2018
-
[4]
A survey on causal inference.ACM Transactions on Knowledge Discovery from Data (TKDD), 15(5):1–46, 2021
Liuyi Yao, Zhixuan Chu, Sheng Li, Yaliang Li, Jing Gao, and Aidong Zhang. A survey on causal inference.ACM Transactions on Knowledge Discovery from Data (TKDD), 15(5):1–46, 2021
2021
-
[5]
Causal inference in the presence of latent variables and selection bias
Peter Spirtes, Christopher Meek, and Thomas Richardson. Causal inference in the presence of latent variables and selection bias. InProceedings of the Eleventh Conference on Uncertainty in Artificial Intelligence, page 499–506, 1995
1995
-
[6]
Optimal structure identification with greedy search.Journal of machine learning research, 3:507–554, 2002
David Maxwell Chickering. Optimal structure identification with greedy search.Journal of machine learning research, 3:507–554, 2002. 11 CauTion: Knowing When to Trust LLMs for Ensemble Causal Discovery
2002
-
[7]
Fast scalable and accurate discovery of dags using the best order score search and grow shrink trees
Bryan Andrews, Joseph Ramsey, Ruben Sanchez Romero, Jazmin Camchong, and Erich Kum- merfeld. Fast scalable and accurate discovery of dags using the best order score search and grow shrink trees. InAdvances in Neural Information Processing Systems, volume 36, pages 63945–63956, 2023
2023
-
[8]
Greedyrelaxationsofthesparsestpermutation algorithm
Wai-YatLam,BryanAndrews,andJosephRamsey. Greedyrelaxationsofthesparsestpermutation algorithm. InProceedings of the 38th Conference on Uncertainty in Artificial Intelligence, pages 1052–1062. PMLR, 2022
2022
Show all 38 references
-
[9]
Xun Zheng, Bryon Aragam, Pradeep Ravikumar, and Eric P. Xing. DAGs with NO TEARS: Continuous Optimization for Structure Learning. InAdvances in Neural Information Processing Systems, 2018
2018
-
[10]
DAGMA: Learning DAGs via M-matrices and a Log-Determinant Acyclicity Characterization
Kevin Bello, Bryon Aragam, and Pradeep Ravikumar. DAGMA: Learning DAGs via M-matrices and a Log-Determinant Acyclicity Characterization. InAdvances in Neural Information Processing Systems, volume 35, pages 8226–8239, 2022
2022
-
[11]
Equivalence and synthesis of causal models
Thomas Verma and Judea Pearl. Equivalence and synthesis of causal models. InProceedings of the Sixth Annual Conference on Uncertainty in Artificial Intelligence, pages 255–270, 1990
1990
-
[12]
Efficient causal graph discovery using large language models
Thomas Jiralerspong, Xiaoyin Chen, Yash More, Vedant Shah, and Yoshua Bengio. Efficient causal graph discovery using large language models. InICLR 2024 Workshop: How Far Are We From AGI, 2024. URLhttps://openreview.net/forum?id=5RBUTx75yr
2024
-
[13]
Causal-llm: A unified one-shot framework for prompt-and data-driven causal graph discovery
Amartya Roy, N Devharish, Shreya Ganguly, and Kripabandhu Ghosh. Causal-llm: A unified one-shot framework for prompt-and data-driven causal graph discovery. InFindings of the Association for Computational Linguistics: EMNLP 2025, pages 8259–8279, 2025
2025
-
[14]
Integrating large language model for improved causal discovery.IEEE Transactions on Artificial Intelligence, 6(11):3030–3042, 2025
Taiyu Ban, Lyuzhou Chen, Derui Lyu, Xiangyu Wang, Qinrui Zhu, Qiang Tu, and Huanhuan Chen. Integrating large language model for improved causal discovery.IEEE Transactions on Artificial Intelligence, 6(11):3030–3042, 2025. doi: 10.1109/TAI.2025.3560927
2025 doi
-
[15]
Causal modelling agents: Causal graph discovery through synergising metadata-and data-driven reasoning
AhmedAbdulaal, NinaMontana-Brown, TiantianHe, AyodejiIjishakin, IvanaDrobnjak, DanielC Castro, Daniel C Alexander, et al. Causal modelling agents: Causal graph discovery through synergising metadata-and data-driven reasoning. InThe Twelfth International Conference on Learning ...
2024
-
[16]
Causcientist: Teaching llms to respect data for causal discovery.arXiv preprint arXiv:2601.13614, 2026
Bo Peng, Sirui Chen, Lei Xu, and Chaochao Lu. Causcientist: Teaching llms to respect data for causal discovery.arXiv preprint arXiv:2601.13614, 2026
2026
-
[17]
Balasubramanian, and Amit Sharma
Aniket Vashishtha, Abbavaram Gowtham Reddy, Abhinav Kumar, Saketh Bachu, Vineeth N. Balasubramanian, and Amit Sharma. Causal order: The key to leveraging imperfect experts in causal inference. InThe Thirteenth International Conference on Learning Representations, 2025. URLhttp...
2025
-
[18]
Integrating large language models in causal discovery: A statistical causal approach.Transactions on Machine Learning Research, 2025
Masayuki Takayama, Tadahisa OKUDA, Thong Pham, Tatsuyoshi Ikenoue, Shingo Fukuma, Shohei Shimizu, and Akiyoshi Sannai. Integrating large language models in causal discovery: A statistical causal approach.Transactions on Machine Learning Research, 2025. ISSN 2835-8856. URLhttps...
2025
-
[19]
Causal discovery with language models as imperfect experts
Stephanie Long, Alexandre Piché, Valentina Zantedeschi, Tibor Schuster, and Alexandre Drouin. Causal discovery with language models as imperfect experts. InICML 2023 Workshop on Structured Probabilistic Inference{∖&}Generative Modeling, 2023. 12 CauTion: Knowing When to Trust ...
2023
-
[20]
Causal discovery via mml
Chris S Wallace, Kevin B Korb, and Honghua Dai. Causal discovery via mml. InProceedings of the Thirteenth International Conference on Machine Learning, pages 516–524, 1996
1996
-
[21]
Can LLMs express their uncertainty? an empirical evaluation of confidence elicitation in LLMs
Miao Xiong, Zhiyuan Hu, Xinyang Lu, YIFEI LI, Jie Fu, Junxian He, and Bryan Hooi. Can LLMs express their uncertainty? an empirical evaluation of confidence elicitation in LLMs. InThe Twelfth International Conference on Learning Representations, 2024. URLhttps:// openreview.net...
2024
-
[22]
Deep structural causal models for tractable counterfactual inference.Advances in neural information processing systems, 33: 857–869, 2020
Nick Pawlowski, Daniel Coelho de Castro, and Ben Glocker. Deep structural causal models for tractable counterfactual inference.Advances in neural information processing systems, 33: 857–869, 2020
2020
-
[23]
Learning bayesian networks with the bnlearn r package.Journal of statistical software, 35:1–22, 2010
Marco Scutari. Learning bayesian networks with the bnlearn r package.Journal of statistical software, 35:1–22, 2010
2010
-
[24]
CRC Press, 2nd edition, 2010
Kevin B Korb and Ann E Nicholson.Bayesian Artificial Intelligence. CRC Press, 2nd edition, 2010
2010
-
[25]
Adaptive probabilistic networks with hidden variables.Machine Learning, 29(2):213–244, 1997
John Binder, Daphne Koller, Stuart Russell, and Keiji Kanazawa. Adaptive probabilistic networks with hidden variables.Machine Learning, 29(2):213–244, 1997
1997
-
[26]
An expert system for control of waste water treatment—a pilot project
FV Jensen, U Kjærulff, KG Olesen, and J Pedersen. An expert system for control of waste water treatment—a pilot project. Technical report, Technical report, Judex Datasystemer A/S, Aalborg,
-
[27]
The alarm monitoring system: A case study with two probabilistic inference techniques for belief networks
Ingo A Beinlich, Henri Jacques Suermondt, R Martin Chavez, and Gregory F Cooper. The alarm monitoring system: A case study with two probabilistic inference techniques for belief networks. InAIME 89: Second European Conference on Artificial Intelligence in Medicine, London, Aug...
1989
-
[28]
Kristian Kristensen and Ilse A Rasmussen. The use of a bayesian network in the design of a decision support system for growing malting barley without use of pesticides.Computers and Electronics in Agriculture, 33(3):197–217, 2002
2002
-
[29]
Decision-theoretic troubleshooting
David Heckerman, John S Breese, and Koos Rommelse. Decision-theoretic troubleshooting. Communications of the ACM, 38(3):49–57, 1995
1995
-
[30]
Structural intervention distance for evaluating causal graphs
Jonas Peters and Peter Bühlmann. Structural intervention distance for evaluating causal graphs. Neural computation, 27(3):771–799, 2015
2015
-
[31]
Learning sparse nonparametric dags
Xun Zheng, Chen Dan, Bryon Aragam, Pradeep Ravikumar, and Eric Xing. Learning sparse nonparametric dags. InInternational Conference on Artificial Intelligence and Statistics, pages 3414–3425. PMLR, 2020
2020
-
[32]
Sample, estimate, aggregate: A recipe for causal discovery foundation models.Transactions on Machine Learning Research,
Menghua Wu, Yujia Bao, Regina Barzilay, and Tommi Jaakkola. Sample, estimate, aggregate: A recipe for causal discovery foundation models.Transactions on Machine Learning Research,
-
[33]
URLhttps://openreview.net/forum?id=h434zx5SX0
ISSN 2835-8856. URLhttps://openreview.net/forum?id=h434zx5SX0
-
[34]
Causcale: Neural causal discovery at scale
Bo Peng, Sirui Chen, Jiaguo Tian, Yu Qiao, and Chaochao Lu. Causcale: Neural causal discovery at scale. InProceedings of the 43rd International Conference on Machine Learning, 2026
2026
-
[35]
pgmpy: A python toolkit for bayesian networks.Journal of Machine Learning Research, 25(265):1–8, 2024
Ankur Ankan and Johannes Textor. pgmpy: A python toolkit for bayesian networks.Journal of Machine Learning Research, 25(265):1–8, 2024. URLhttp://jmlr.org/papers/v25/ 23-0487.html. 13 CauTion: Knowing When to Trust LLMs for Ensemble Causal Discovery
2024
-
[36]
Pedregosa, G
F. Pedregosa, G. Varoquaux, A. Gramfort, V. Michel, B. Thirion, O. Grisel, M. Blondel, P. Pretten- hofer, R. Weiss, V. Dubourg, J. Vanderplas, A. Passos, D. Cournapeau, M. Brucher, M. Perrot, and E. Duchesnay. Scikit-learn: Machine learning in Python.Journal of Machine Learnin...
2011
-
[37]
Causal-learn: Causal discovery in python.Journal of Machine Learning Research, 25(60):1–8, 2024
Yujia Zheng, Biwei Huang, Wei Chen, Joseph Ramsey, Mingming Gong, Ruichu Cai, Shohei Shimizu, Peter Spirtes, and Kun Zhang. Causal-learn: Causal discovery in python.Journal of Machine Learning Research, 25(60):1–8, 2024
2024
-
[38]
Statistical Causal Prompting,
Keli Zhang, Shengyu Wang, Yian Zheng, Xiaolu Zhang, Jianye Hu, Xuan Hua, et al. gcastle: A python implementation of causal discovery algorithms.arXiv preprint arXiv:2111.15155, 2021. 14 CauTion: Knowing When to Trust LLMs for Ensemble Causal Discovery A. Related Work Details A...
2021
Reviewed June 28, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.