← Search

Ouail Kitouni

4 accepted papers

2024

From Neurons to Neutrons: A Case Study in Interpretability

ICML 2024poster

Mechanistic Interpretability (MI) proposes a path toward fully understanding how neural networks make their predictions. Prior work demonstrates that even when trained to perform simple arithmetic, models can implement a variety of algorithms (sometimes concurrently) depending on initialization and…

2024

The Factorization Curse: Which Tokens You Predict Underlie the Reversal Curse and More

NeurIPS 2024poster

Today's best language models still struggle with "hallucinations", factually incorrect generations, which impede their ability to reliably retrieve information seen during training. The *reversal curse*, where models cannot recall information when probed in a different order than was encountered dur…

Cited by 9SourcePDFScholar
2022

Towards Understanding Grokking: An Effective Theory of Representation Learning

NeurIPS 2022accept

We aim to understand grokking, a phenomenon where models generalize long after overfitting their training set. We present both a microscopic analysis anchored by an effective theory and a macroscopic analysis of phase diagrams describing learning performance across hyperparameters. We find that gene…