Predict a causal graph from a table of observational data without designing a custom model
Fill in missing values in a dataset using the model's built-in imputation feature
Benchmark a new causal discovery method against a state-of-the-art pretrained baseline
Reproduce or build on the CDFM results reported in the associated arXiv paper
| dmirlab-group/cdfm | fudancvl/sam-mt | kizuna-intelligence/irodori-tts-lite | |
|---|---|---|---|
| Stars | 53 | 53 | 53 |
| Language | Python | Python | Python |
| Setup difficulty | easy | hard | hard |
| Complexity | 3/5 | 4/5 | 4/5 |
| Audience | researcher | researcher | researcher |
Figures from each repo's GitHub metadata at analysis time.
Requires PyTorch 2.0 or newer and downloads a pretrained checkpoint from Hugging Face.
CDFM, short for Causal Discovery Foundation Model, is a pretrained machine learning model that tries to figure out cause and effect relationships hidden in a dataset. Given a table of numbers with no labels telling you what causes what, the model looks at the raw data and predicts a causal graph, meaning a diagram of which variables in the data appear to influence which other variables, in a single pass rather than through a lengthy statistical search. The README explains that CDFM was trained on a large and varied collection of synthetic datasets built from artificial cause and effect structures, and that this training lets it recognize the kinds of statistical patterns that show up when one variable causes another. Because of that training, it works in what the paper calls a zero-shot setting: the same pretrained model can be applied to a new dataset, of a different size and number of variables, without retraining it first. The README states it outperforms other methods it was compared against across 15 different types of underlying data generating processes and on real-world benchmark datasets. Using the model is meant to be simple. It is installed with a single pip command, and Python code loads it from Hugging Face Hub with one line and produces a prediction with another, calling model.predict on your data to get back a causal graph as a grid of zeros and ones. The library also includes a separate feature for filling in missing values in a dataset, which the README shows performing noticeably better than a simple average based fill-in method on an example dataset. The project is research code released alongside an academic paper on arXiv, and it targets people doing causal inference or structural learning research in Python, such as academics or data scientists who need to estimate causal structure from observational data without hand designing a model for each new dataset. It is released under the Apache 2.0 license.
A pretrained Python model that predicts cause-and-effect graphs from raw observational data in a single pass, without retraining per dataset.
Mainly Python. The stack also includes Python, PyTorch, Hugging Face Hub.
Apache 2.0: free to use, modify, and distribute, including commercially, as long as you keep the license and copyright notices.
Setup difficulty is rated easy, with roughly 5min to a first successful run.
Mainly researcher.
This repo across BitVibe Labs
Don't trust strangers blindly. Verify against the repo.