LLM-initialized Differentiable Causal Discovery

TL;DR

This paper introduces LLM-DCD, a novel causal discovery method that leverages large language models for initialization, improving interpretability and accuracy in uncovering causal relationships from observational data.

Contribution

The paper proposes a new approach that integrates LLMs with differentiable causal discovery by using LLMs for initialization, enhancing interpretability and performance.

Findings

Higher accuracy on benchmark datasets compared to state-of-the-art methods.

Initialization quality significantly affects final causal discovery results.

Explicit adjacency matrix optimization improves interpretability.

Abstract

The discovery of causal relationships between random variables is an important yet challenging problem that has applications across many scientific domains. Differentiable causal discovery (DCD) methods are effective in uncovering causal relationships from observational data; however, these approaches often suffer from limited interpretability and face challenges in incorporating domain-specific prior knowledge. In contrast, Large Language Models (LLMs)-based causal discovery approaches have recently been shown capable of providing useful priors for causal discovery but struggle with formal causal reasoning. In this paper, we propose LLM-DCD, which uses an LLM to initialize the optimization of the maximum likelihood objective function of DCD approaches, thereby incorporating strong priors into the discovery method. To achieve this initialization, we design our objective function to depend on an explicitly defined adjacency matrix of the causal graph as its only variational parameter. Directly optimizing the explicitly defined adjacency matrix provides a more interpretable approach to causal discovery. Additionally, we demonstrate higher accuracy on key benchmarking datasets of our approach compared to state-of-the-art alternatives, and provide empirical evidence that the quality of the initialization directly impacts the quality of the final output of our DCD approach. LLM-DCD opens up new opportunities for traditional causal discovery methods like DCD to benefit from future improvements in the causal reasoning capabilities of LLMs.