Leveraging Large Language Models for Causal Discovery: a Constraint-based, Argumentation-driven Approach
Organizations: Department of Computing, Imperial College London, UK
Abstract
Causal discovery seeks to uncover causal relations from data, typically represented as causal graphs, and is essential for predicting the effects of interventions. While expert knowledge is required to construct principled causal graphs, many statistical methods have been proposed to leverage observational data with varying formal guarantees. Causal Assumption-based Argumentation (ABA) is a framework that uses symbolic reasoning to ensure correspondence between input constraints and output graphs, while offering a principled way to combine data and expertise. We explore the use of large language models (LLMs) as imperfect experts for Causal ABA, eliciting semantic structural priors from variable names and descriptions and integrating them with conditional-independence evidence. Experiments on standard benchmarks and semantically grounded synthetic graphs demonstrate state-of-the-art performance, and we additionally introduce an evaluation protocol to mitigate memorisation bias when assessing LLMs for causal discovery.
Figures & tables
| Method | SHD | Precision | Recall | F1 | |
|---|---|---|---|---|---|
| Random | |||||
| FGS | |||||
| NOTEARS-MLP | |||||
| GRaSP | |||||
| BOSS | |||||
| MPC |
Appendix figures & tables24 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset Name | |||
|---|---|---|---|
| CANCER | 5 | 4 | 0.8 |
| EARTHQUAKE | 5 | 4 | 0.8 |
| SURVEY | 6 | 6 | 1 |
| ASIA | 8 | 8 | 1 |
| SACHS | 11 | 17 | 1.55 |
| CHILD | 20 | 25 | 1.25 |
| Method | ER | SF | LT | Average | ||
|---|---|---|---|---|---|---|
| SHD | Random | |||||
| FGS | ||||||
| NOTEARS-MLP | ||||||
| GRaSP | ||||||
| BOSS | ||||||
| MPC |
| Metric | Method | ||||
|---|---|---|---|---|---|
| SHD | Random | ||||
| FGS | |||||
| NOTEARS-MLP | |||||
| GRaSP | |||||
| BOSS | |||||
| MPC |
| Dataset | Method | SHD | SID | Precision | Recall | F1 |
|---|---|---|---|---|---|---|
| cancer | Random | |||||
| FGS | ||||||
| NOTEARS-MLP | ||||||
| GRaSP | ||||||
| BOSS | ||||||
| MPC | ** |
| Method | |||
|---|---|---|---|
| Random | |||
| FGS | |||
| NOTEARS-MLP | |||
| GRaSP | |||
| BOSS | |||
| MPC |
| bnlearn | CauseNet | |||||||
| Metric | Average | Consensus | Average | Consensus | ||||
| w/o desc | with desc | w/o desc | with desc | w/o desc | with desc | w/o desc | with desc | |
| Forbidden constraints | ||||||||
| # constraints | ||||||||
| Precision | ||||||||
| Recall | ||||||||
| bnlearn | CauseNet | |||||||
| Metric | Average | Consensus | Average | Consensus | ||||
| w/o desc | with desc | w/o desc | with desc | w/o desc | with desc | w/o desc | with desc | |
| Forbidden constraints | ||||||||
| # constraints | ||||||||
| Precision | ||||||||
| Recall | ||||||||
| Method | SHD | Precision | Recall | F1 | |
|---|---|---|---|---|---|
| Random | |||||
| FGS | |||||
| NOTEARS-MLP | |||||
| GRaSP | |||||
| BOSS | |||||
| MPC |
| Group | Metric | Methods | Means Std | t | p-value | |||
| SHD | ABAPC-LLM vs MPC-LLM | vs | ||||||
| Precision | MPC-LLM vs ABAPC-LLM | vs | ||||||
| Recall | ABAPC-LLM vs MPC-LLM | vs | ||||||
| F1 | ABAPC-LLM vs MPC-LLM | vs | ||||||
| SHD | ABAPC-LLM vs ABAPC | vs | ||||||
| Precision | BOSS vs MPC-LLM | vs | ||||||
| Table | Group | Metric | Methods | Means Std | t | p-value | |||
| 1 | Precision | MPC vs MPC-LLM | vs | . | . | ||||
| 3 | SF | F1 | ABAPC-LLM vs MPC-LLM | vs | . | . | |||
| 3 | LT | SHD | MPC-LLM vs ABAPC-LLM | vs | ns | ns | |||
| 3 | ER | F1 | MPC-LLM vs ABAPC-LLM | vs | ns | ns | |||
| 3 | SF | SHD | ABAPC-LLM vs MPC-LLM | vs | . | . | |||
| 3 | LT | F1 | BOSS vs ABAPC-LLM | vs | . | . | |||
| Group | Metric | Methods | Means Std | t | p-value | |||
| ER, | SHD | ABAPC-LLM vs ABAPC | vs | |||||
| ER, | F1 | ABAPC-LLM vs MPC-LLM | vs | |||||
| SF, | SHD | ABAPC-LLM vs MPC-LLM | vs | ** | ** | |||
| SF, | F1 | ABAPC-LLM vs ABAPC | vs | |||||
| SF, | F1 | MPC-LLM vs ABAPC | vs | |||||
| LT, | SHD | MPC-LLM vs ABAPC | vs | |||||
| Group | Metric | Methods | Means Std | t | p-value | |||
| , | SHD | ABAPC-LLM vs MPC-LLM | vs | |||||
| , | F1 | ABAPC-LLM vs MPC-LLM | vs | |||||
| , | SHD | ABAPC-LLM vs ABAPC | vs | |||||
| , | F1 | ABAPC-LLM vs MPC-LLM | vs | |||||
| , | SHD | ABAPC-LLM vs MPC-LLM | vs | ** | ** | |||
| , | F1 | MPC-LLM vs ABAPC-LLM | vs | |||||
| Group | Metric | Methods | Means Std | t | p-value | |||
|---|---|---|---|---|---|---|---|---|
| earthquake | SHD | MPC vs NOTEARS-MLP | vs | |||||
| earthquake | SHD | MPC-LLM vs NOTEARS-MLP | vs | |||||
| earthquake | SHD | ABAPC-LLM vs NOTEARS-MLP | vs | |||||
| earthquake | SHD | ABAPC vs NOTEARS-MLP | vs | |||||
| earthquake | SHD | GRaSP vs NOTEARS-MLP | vs | |||||
| earthquake | SHD | BOSS vs NOTEARS-MLP | vs | |||||
| Group | Metric | Methods | Means Std | t | p-value | |||
| asia | SHD | MPC-LLM vs ABAPC-LLM | vs | |||||
| asia | Precision | ABAPC-LLM vs ABAPC | vs | |||||
| asia | Precision | MPC-LLM vs ABAPC | vs | |||||
| asia | Precision | MPC vs ABAPC | vs | |||||
| asia | Recall | MPC-LLM vs ABAPC-LLM | vs | |||||
| asia | F1 | MPC-LLM vs ABAPC-LLM | vs | |||||
| Group | Metric | Methods | Means Std | t | p-value | |||
|---|---|---|---|---|---|---|---|---|
| child | SHD | ABAPC-LLM vs MPC | vs | |||||
| child | SHD | MPC-LLM vs MPC | vs | |||||
| child | SHD | ABAPC vs MPC | vs | |||||
| child | Precision | BOSS vs ABAPC-LLM | vs | |||||
| child | Recall | ABAPC-LLM vs MPC | vs | |||||
| child | Recall | MPC-LLM vs MPC | vs | |||||
| Group | Metric | Methods | Means Std | t | p-value | |||
| asia | Precision | ABAPC-LLM vs MPC | vs | ns | ns | |||
| asia | Precision | ABAPC-LLM vs MPC-LLM | vs | ns | ns | |||
| cancer | SHD | MPC-LLM vs ABAPC-LLM | vs | ns | ns | |||
| cancer | Precision | ABAPC-LLM vs MPC-LLM | vs | ns | ns | |||
| cancer | Recall | MPC-LLM vs ABAPC-LLM | vs | ns | ns | |||
| cancer | F1 | MPC vs MPC-LLM | vs | ns | ns | |||
| Group | Metric | Methods | Means Std | t | p-value | |||
|---|---|---|---|---|---|---|---|---|
| SHD | ABAPC-LLM (Gemini) vs MPC-LLM (Gemini) | vs | ||||||
| Precision | MPC-LLM (Gemini) vs ABAPC-LLM (Gemini) | vs | ||||||
| Recall | ABAPC-LLM (Gemini) vs MPC-LLM (Gemini) | vs | ||||||
| F1 | ABAPC-LLM (Gemini) vs ABAPC-LLM (GPT) | vs | ||||||
| F1 | MPC-LLM (Gemini) vs ABAPC-LLM (GPT) | vs | ||||||
| SHD | ABAPC-LLM (GPT) vs MPC-LLM (Gemini) | vs | ||||||
Causal Discovery
Causal discovery infers cause-and-effect structures from observational and interventional data, distinguishing causal relationships from correlation.
See all 220 papers