activity
20232026
most citedFuture Lens: Anticipating Subsequent Tokens from a Single Hidden State

9 citations · 17 across the 10 of their papers we have counts for

collaborators

10 papers

cs.LG2026

Mechanistic Interpretability of Structure-Aware Numerical Reasoning in LLaMA 3.1 8B

Rahul Chowdhury, Timothy A Rupprecht, Senhao Cao +5

Recent work has shown that large language models (LLMs) exhibit strong numerical sequence modeling capabilities and show promise in time-series prediction. While LLMs display in-co…

cs.CL2026

Shared Lexical Task Representations Explain Behavioral Variability In LLMs

Zhuonan Yang, Jacob Xiaochen Li, Francisco Piedrahita Velez +5

One of the most common complaints about large language models (LLMs) is their prompt sensitivity -- that is, the fact that their ability to perform a task or provide a correct answ…

cs.CL2026

Do explanations generalize across large reasoning models?

Koyena Pal, David Bau, Chandan Singh

Large reasoning models (LRMs) produce a textual chain of thought (CoT) in the process of solving a problem, which serves as a potentially powerful tool to understand the problem by…

cs.AI2025

LLMs Process Lists With General Filter Heads

Arnab Sen Sharma, Giordano Rogers, Natalie Shapira +1

We investigate the mechanisms underlying a range of list-processing tasks in LLMs, and we find that LLMs have learned to encode a compact, causal representation of a general filter…

cs.CL2025

The Dual-Route Model of Induction

Sheridan Feucht, Eric Todd, Byron Wallace +1

Prior work on in-context copying has shown the existence of induction heads, which attend to and promote individual tokens during copying. In this work we discover a new type of in…

cs.CL2025

Elucidating Mechanisms of Demographic Bias in LLMs for Healthcare

Hiba Ahsan, Arnab Sen Sharma, Silvio Amir +2

We know from prior work that LLMs encode social biases, and that this manifests in clinical tasks. In this work we adopt tools from mechanistic interpretability to unveil sociodemo…

cs.LG2024★ 4 cited

NNsight and NDIF: Democratizing Access to Open-Weight Foundation Model Internals

Jaden Fiotto-Kaufman, Alexander R. Loftus, Eric Todd +17

We introduce NNsight and NDIF, technologies that work in tandem to enable scientific study of the representations and computations learned by very large neural networks. NNsight is…

cs.DB2024

Model Lakes

Koyena Pal, David Bau, Renée J. Miller

Given a set of deep learning models, it can be hard to find models appropriate to a task, understand the models, and characterize how models are different one from another. Current…

cs.CL2023★ 9 cited

Future Lens: Anticipating Subsequent Tokens from a Single Hidden State

Koyena Pal, Jiuding Sun, Andrew Yuan +2

We conjecture that hidden state vectors corresponding to individual input tokens encode information sufficient to accurately predict several tokens ahead. More concretely, in this…

cs.CL2023★ 4 cited

Function Vectors in Large Language Models

Eric Todd, Millicent L. Li, Arnab Sen Sharma +3

We report the presence of a simple neural mechanism that represents an input-output function as a vector within autoregressive transformer language models (LMs). Using causal media…