← Search

Raanan Yehezkel Rohekar

5 accepted papers

2025

A Causal World Model Underlying Next Token Prediction: Exploring GPT in a Controlled Environment

ICML 2025poster

Are generative pre-trained transformer (GPT) models, trained only to predict the next token, implicitly learning a world model from which sequences are generated one token at a time? We address this question by deriving a causal interpretation of the attention mechanism in GPT and presenting a causa…

2023

Causal Interpretation of Self-Attention in Pre-Trained Transformers

NeurIPS 2023poster

We propose a causal interpretation of self-attention in the Transformer neural network architecture. We interpret self-attention as a mechanism that estimates a structural equation model for a given input sequence of symbols (tokens). The structural equation model can be interpreted, in turn, as a c…

2023

From Temporal to Contemporaneous Iterative Causal Discovery in the Presence of Latent Confounders

ICML 2023poster

We present a constraint-based algorithm for learning causal structures from observational time-series data, in the presence of latent confounders. We assume a discrete-time, stationary structural vector autoregressive process, with both temporal and contemporaneous causal relations. One may ask if t…

2021

Iterative Causal Discovery in the Possible Presence of Latent Confounders and Selection Bias

NeurIPS 2021poster

We present a sound and complete algorithm, called iterative causal discovery (ICD), for recovering causal graphs in the presence of latent confounders and selection bias. ICD relies on the causal Markov and faithfulness assumptions and recovers the equivalence class of the underlying causal graph. I…

2019

Modeling Uncertainty by Learning a Hierarchy of Deep Neural Connections

NeurIPS 2019poster

Modeling uncertainty in deep neural networks, despite recent important advances, is still an open problem. Bayesian neural networks are a powerful solution, where the prior over network weights is a design choice, often a normal distribution or other distribution encouraging sparsity. However, this…

Cited by 18SourcePDFScholar