All articles Research

Foundation Models for Causal Inference (CausalFM)

CausalFM is a transformer-based foundation model trained on causal data so it can reason about cause and effect rather than mere correlation.

CausalFM is a new type of AI model designed to learn cause–effect relationships, not just statistical patterns. Where most models answer “what usually goes together?”, CausalFM aims to answer “what causes what?” — a key step toward AI that reasons like a scientist.

CausalFM = a foundation model that can answer “what causes what?” and “what happens if we change something?”

The problem it solves

Traditional AI — including large language models and most deep-learning systems — learns correlations: it notices that things occur together. For example, it learns that smoking and cancer appear together in data, but it doesn't truly understand whether smoking causes cancer.

CausalFM instead learns cause → effect relationships, so it can distinguish "these happen together" from "this one produces that one."

How it works

CausalFM builds on the transformer architecture but adds a few key ideas:

  • Trained on simulated worlds. It uses synthetic data generated from causal models, learning scenarios of the form “if A changes, what happens to B?”
  • Prior-Data Fitted Networks (PFNs). The model is trained on defined cause-effect rules (priors) and then learns to generalize beyond them.
  • In-context causal reasoning. Given an input, it can answer “what caused this?” and “what if we change X?” — a form of counterfactual reasoning.

A worked example

Input: a patient took drug A and recovered.

Unlike a standard LLM, CausalFM can reason about questions such as:

  • Did the drug actually cause the recovery?
  • What would have happened if the patient hadn't taken it?
  • What if we changed the dosage?

Why it's a breakthrough

  • Moves AI beyond correlation — it can reason about why things happen.
  • Works in critical domains — medicine, economics, and policy decisions, where acting on a spurious correlation is costly.
  • A general framework — it can address back-door and front-door causal problems and instrumental-variable analysis within one model.

How it compares

Model typeWhat it primarily learns
GPT / LLMPatterns (correlation)
Reasoning modelsStep-by-step problem solving
CausalFMCause-and-effect (causality)

One-line intuition: CausalFM tries to make AI think like a scientist, not just a pattern recognizer — a transformer-based foundation model trained on causal data so it can perform cause-effect reasoning and answer “what happens if we change something?”

← Back to all articles