← Search

Jonathan D. Cohen

9 accepted papers

2026

Bound by semanticity: universal laws governing the generalization-identification tradeoff

ICLR 2026poster

Intelligent systems must form internal representations that support both broad generalization and precise identification. Here, we show that these two goals are fundamentally in tension with one another. We derive closed-form expressions proving that any model whose representations have a finite s…

Cited by 0SourceScholar
2026

Learning is Forgetting; LLM Training As Lossy Compression

ICLR 2026poster

Despite the increasing prevalence of large language models (LLMs), we still have a limited understanding of how their representational spaces are structured. This limits our ability to interpret how and what they learn or relate them to learning in humans. We argue LLMs are best seen as an instance…

Cited by 0SourcecodeScholar
2025

Causal Head Gating: A Framework for Interpreting Roles of Attention Heads in Transformers

NeurIPS 2025poster

We present causal head gating (CHG), a scalable method for interpreting the functional roles of attention heads in transformer models. CHG learns soft gates over heads and assigns them a causal taxonomy—facilitating, interfering, or irrelevant—based on their impact on task performance. Unlike prior…

Cited by 0SourceScholar
2025

Emergent Symbolic Mechanisms Support Abstract Reasoning in Large Language Models

ICML 2025poster

Many recent studies have found evidence for emergent reasoning capabilities in large language models (LLMs), but debate persists concerning the robustness of these capabilities, and the extent to which they depend on structured reasoning mechanisms. To shed light on these issues, we study the intern…

2024

Abstractors and relational cross-attention: An inductive bias for explicit relational reasoning in Transformers

ICLR 2024poster

An extension of Transformers is proposed that enables explicit relational reasoning through a novel module called the *Abstractor*. At the core of the Abstractor is a variant of attention called *relational cross-attention*. The approach is motivated by an architectural inductive bias for relational…

2024

Slot Abstractors: Toward Scalable Abstract Visual Reasoning

ICML 2024poster

Abstract visual reasoning is a characteristically human ability, allowing the identification of relational patterns that are abstracted away from object features, and the systematic generalization of those patterns to unseen problems. Recent work has demonstrated strong systematic generalization in…

Cited by 8SourcePDFScholar
2024

Understanding the Limits of Vision Language Models Through the Lens of the Binding Problem

NeurIPS 2024poster

Recent work has documented striking heterogeneity in the performance of state-of-the-art vision language models (VLMs), including both multimodal language models and text-to-image models. These models are able to describe and generate a diverse array of complex, naturalistic images, yet they exhibit…

Cited by 7SourcePDFScholar
2017

A graph-theoretic approach to multitasking

NeurIPS 2017oral

A key feature of neural network architectures is their ability to support the simultaneous interaction among large numbers of units in the learning and processing of representations. However, how the richness of such interactions trades off against the ability of a network to simultaneously carry ou…

Cited by 18SourcePDFScholar
2015

A Theory of Decision Making Under Dynamic Context

NeurIPS 2015poster

The dynamics of simple decisions are well understood and modeled as a class of random walk models (e.g. Laming, 1968; Ratcliff, 1978; Busemeyer and Townsend, 1993; Usher and McClelland, 2001; Bogacz et al., 2006). However, most real-life decisions include a rich and dynamically-changing influence of…