← Search

Dan Klein

59 accepted papers

2026

GEPA: Reflective Prompt Evolution Can Outperform Reinforcement Learning

ICLR 2026oral

Large language models (LLMs) are increasingly adapted to downstream tasks via reinforcement learning (RL) methods like Group Relative Policy Optimization (GRPO), which often require thousands of rollouts to learn new tasks. We argue that the interpretable nature of language often provides a much ric…

Cited by 0SourcecodeScholar
2025

Enough Coin Flips Can Make LLMs Act Bayesian

ACL 2025long

Large language models (LLMs) exhibit the ability to generalize given few-shot examples in their input prompt, an emergent capability known as in-context learning (ICL). We investigate whether LLMs use ICL to perform structured reasoning in ways that are consistent with a Bayesian framework or rely o…

Cited by 0SourcePDFScholar
2025

FactTrack: Time-Aware World State Tracking in Story Outlines

NAACL 2025long

While accurately detecting and correcting factual contradictions in language model outputs has become increasingly important as their capabilities improve, doing so is highly challenging. We propose a novel method, FactTrack, for tracking atomic facts and addressing factual contradictions. Crucially…

2025

LangProBe: a Language Program Benchmark

EMNLP 2025

Composing language models (LMs) into multi-step language programs and automatically optimizing their modular prompts is now a mainstream paradigm for building AI systems, but the tradeoffs in this space have only scarcely been studied before. We introduce LangProBe, the first large-scale benchmark f

Cited by 0SourcePDFScholar
2025

Pose Priors from Language Models

CVPR 2025poster

Language is often used to describe physical interaction, yet most 3D human pose estimation methods overlook this rich source of information. We bridge this gap by leveraging large multimodal models (LMMs) as priors for reconstructing contact poses, offering a scalable alternative to traditional meth…

2025

Why Do Multi-Agent LLM Systems Fail?

NeurIPS 2025spotlight

Despite enthusiasm for Multi-Agent LLM Systems (MAS), their performance gains on popular benchmarks are often minimal. This gap highlights a critical need for a principled understanding of why MAS fail. Addressing this question requires systematic identification and analysis of failure patterns. We…

Cited by 0SourcecodeScholar
2024

American Sign Language Handshapes Reflect Pressures for Communicative Efficiency

ACL 2024long

Communicative efficiency is a key topic in linguistics and cognitive psychology, with many studies demonstrating how the pressure to communicate with minimal effort guides the form of natural language. However, this phenomenon is rarely explored in signed languages. This paper shows how handshapes i…

2024

Explaining Datasets in Words: Statistical Models with Natural Language Parameters

NeurIPS 2024poster

To make sense of massive data, we often first fit simplified models and then interpret the parameters; for example, we cluster the text embeddings and then interpret the mean parameters of each cluster. However, these parameters are often high-dimensional and hard to interpret. To make model paramet…

2024

Ghostbuster: Detecting Text Ghostwritten by Large Language Models

NAACL 2024long

We introduce Ghostbuster, a state-of-the-art system for detecting AI-generated text.Our method works by passing documents through a series of weaker language models, running a structured search over possible combinations of their features, and then training a classifier on the selected features to p…

2024

Inferring Ontological Categories of OWL Classes Using Foundational Rules (Extended Abstract)

IJCAI 2024poster

Several efforts that leverage the tools of formal ontology have demonstrated the fruitfulness of considering key metaproperties of classes in ontology engineering. Despite that, it is still a common practice to apply representation schemes and approaches--such as OWL--that do not benefit from identi…

Cited by 0SourcePDFScholar
2024

Learning to Model the World With Language

ICML 2024oral

To interact with humans and act in the world, agents need to understand the range of language that people use and relate it to the visual world. While current agents can learn to execute simple language instructions, we aim to build agents that leverage diverse language---language like "this button…

Cited by 51SourcePDFScholar
2024

Linguistic Bias in ChatGPT: Language Models Reinforce Dialect Discrimination

EMNLP 2024main

We present a large-scale study of linguistic bias exhibited by ChatGPT covering ten dialects of English (Standard American English, Standard British English, and eight widely spoken non-”standard” varieties from around the world). We prompted GPT-3.5 Turbo and GPT-4 with text by native speakers of e…

Cited by 24SourcePDFScholar
2024

RLCD: Reinforcement Learning from Contrastive Distillation for LM Alignment

ICLR 2024poster

We propose Reinforcement Learning from Contrastive Distillation (RLCD), a method for aligning language models to follow principles expressed in natural language (e.g., to be more harmless) without using human feedback. RLCD creates preference pairs from two contrasting model outputs, one using a pos…

Cited by 23SourcePDFScholar
2024

Re-evaluating the Need for Visual Signals in Unsupervised Grammar Induction

NAACL 2024findings

Are multimodal inputs necessary for grammar induction? Recent work has shown that multimodal training inputs can improve grammar induction. However, these improvements are based on comparisons to weak text-only baselines that were trained on relatively little textual data. To determine whether multi…

2024

The Perspectivist Paradigm Shift: Assumptions and Challenges of Capturing Human Labels

NAACL 2024long

Longstanding data labeling practices in machine learning involve collecting and aggregating labels from multiple annotators. But what should we do when annotators disagree? Though annotator disagreement has long been seen as a problem to minimize, new perspectivist approaches challenge this assumpti…

Cited by 19SourcePDFScholar
2024

Which One? Leveraging Context Between Objects and Multiple Views for Language Grounding

NAACL 2024long

When connecting objects and their language referents in an embodied 3D environment, it is important to note that: (1) an object can be better characterized by leveraging comparative information between itself and other objects, and (2) an object’s appearance can vary with camera position. As such, w…

2023

Are Layout-Infused Language Models Robust to Layout Distribution Shifts? A Case Study with Scientific Documents

ACL 2023findings

Recent work has shown that infusing layout features into language models (LMs) improves processing of visually-rich documents such as scientific papers. Layout-infused LMs are often evaluated on documents with familiar layout features (e.g., papers from the same publisher), but in practice models en…

2023

Can Language Models Learn to Listen?

ICCV 2023poster

We present a framework for generating appropriate facial responses from a listener in dyadic social interactions based on the speaker's words. Given an input transcription of the speaker's words with their timestamps, our approach autoregressively predicts a response of a listener: a sequence of lis…

Cited by 24PDFScholar
2023

Centering the Margins: Outlier-Based Identification of Harmed Populations in Toxicity Detection

EMNLP 2023long main

The impact of AI models on marginalized communities has traditionally been measured by identifying performance differences between specified demographic subgroups. Though this approach aims to center vulnerable groups, it risks obscuring patterns of harm faced by intersectional subgroups or shared a…

Cited by 0SourceScholar
2023

DOC: Improving Long Story Coherence With Detailed Outline Control

ACL 2023long

We propose the Detailed Outline Control (DOC) framework for improving long-range plot coherence when automatically generating several-thousand-word-long stories. DOC consists of two complementary components: a detailed outliner and a detailed controller. The detailed outliner creates a more detailed…

2023

Discovering Latent Knowledge in Language Models Without Supervision

ICLR 2023poster

Existing techniques for training language models can be misaligned with the truth: if we train models with imitation learning, they may reproduce errors that humans make; if we train them to generate text that humans rate highly, they may output errors that human evaluators can't detect. We propose…

2023

Goal Driven Discovery of Distributional Differences via Language Descriptions

NeurIPS 2023poster

Exploring large corpora can generate useful discoveries but is time-consuming for humans. We formulate a new task, D5, that automatically discovers differences between two large corpora in a goal-driven way. The task input is a problem comprising a user-specified research goal (“*comparing…

Cited by 46SourcePDFScholar
2023

Modular Visual Question Answering via Code Generation

ACL 2023short

We present a framework that formulates visual question answering as modular code generation. In contrast to prior work on modular approaches to VQA, our approach requires no additional training and relies on pre-trained language models (LMs), visual models pre-trained on image-caption pairs, and fif…

2023

Non-Programmers Can Label Programs Indirectly via Active Examples: A Case Study with Text-to-SQL

EMNLP 2023long main

Can non-programmers annotate natural language utterances with complex programs that represent their meaning? We introduce APEL, a framework in which non-programmers select among candidate programs generated by a seed semantic parser (e.g., Codex). Since they cannot understand the candidate program…

Cited by 0SourcecodeScholar
2023

The Whole Truth and Nothing But the Truth: Faithful and Controllable Dialogue Response Generation with Dataflow Transduction and Constrained Decoding

ACL 2023findings

In a real-world dialogue system, generated text must be truthful and informative while remaining fluent and adhering to a prescribed style. Satisfying these constraints simultaneously isdifficult for the two predominant paradigms in language generation: neural language modeling and rule-based genera…

2023

When the Majority is Wrong: Modeling Annotator Disagreement for Subjective Tasks

EMNLP 2023long main

Though majority vote among annotators is typically used for ground truth labels in machine learning, annotator disagreement in tasks such as hate speech detection may reflect systematic differences in opinion across groups, not noise. Thus, a crucial problem in hate speech detection is determining i…

Cited by 0SourceScholar
2022

Automated Crossword Solving

ACL 2022long

We present the Berkeley Crossword Solver, a state-of-the-art approach for automatically solving crossword puzzles. Our system works by generating answer candidates for each crossword clue using neural question answering models and then combines loopy belief propagation with local search to find full…

2022

Re3: Generating Longer Stories With Recursive Reprompting and Revision

EMNLP 2022main

We consider the problem of automatically generating longer stories of over two thousand words. Compared to prior work on shorter stories, long-range plot coherence and relevance are more central challenges here. We propose the Recursive Reprompting and Revision framework (Re3) to address these chall…

2021

Adapting Language Models for Zero-shot Learning by Meta-tuning on Dataset and Prompt Collections

EMNLP 2021finding

Large pre-trained language models (LMs) such as GPT-3 have acquired a surprising ability to perform zero-shot learning. For example, to classify sentiment without any training examples, we can “prompt” the LM with the review and the label description “Does the user like this movie?”, and ask whether…

2021

Calibrate Before Use: Improving Few-shot Performance of Language Models

ICML 2021oral

GPT-3 can perform numerous tasks when provided a natural language prompt that contains a few training examples. We show that this type of few-shot learning can be unstable: the choice of prompt format, training examples, and even the order of the examples can cause accuracy to vary from near chance…

2021

Constrained Language Models Yield Few-Shot Semantic Parsers

EMNLP 2021main

We explore the use of large pretrained language models as few-shot semantic parsers. The goal in semantic parsing is to generate a structured meaning representation given a natural language input. However, language models are trained to generate natural language. To bridge the gap, we use language m…

2021

Detoxifying Language Models Risks Marginalizing Minority Voices

NAACL 2021long

Language models (LMs) must be both safe and equitable to be responsibly deployed in practice. With safety in mind, numerous detoxification techniques (e.g., Dathathri et al. 2020; Krause et al. 2020) have been proposed to mitigate toxic LM generations. In this work, we show that these detoxification…

2021

Learning Space Partitions for Path Planning

NeurIPS 2021poster

Path planning, the problem of efficiently discovering high-reward trajectories, often requires optimizing a high-dimensional and multimodal reward function. Popular approaches like CEM and CMA-ES greedily focus on promising regions of the search space and may get trapped in local maxima. DOO and VOO…

2021

Modular Networks for Compositional Instruction Following

NAACL 2021long

Standard architectures used in instruction following often struggle on novel compositions of subgoals (e.g. navigating to landmarks or picking up objects) observed during training. We propose a modular architecture for following natural language instructions that describe sequences of diverse subgoa…

2021

Value-Agnostic Conversational Semantic Parsing

ACL 2021long

Conversational semantic parsers map user utterances to executable programs given dialogue histories composed of previous utterances, programs, and system responses. Existing parsers typically condition on rich representations of history that include the complete set of values and computations previo…

2020

Train Big, Then Compress: Rethinking Model Size for Efficient Training and Inference of Transformers

ICML 2020poster

Since hardware resources are limited, the objective of training deep learning models is typically to maximize accuracy subject to the time and memory constraints of training and inference. We study the impact of model size in this setting, focusing on Transformer models for NLP tasks that are limite…

Cited by 360SourcePDFScholar
2018

Speaker-Follower Models for Vision-and-Language Navigation

NeurIPS 2018poster

Navigation guided by natural language instructions presents a challenging reasoning problem for instruction followers. Natural language instructions typically identify only a few high-level decisions and landmarks rather than complete low-level motor behaviors; much of the missing information must b…

Cited by 589SourcePDFScholar
2015

On the Accuracy of Self-Normalized Log-Linear Models

NeurIPS 2015poster

Calculation of the log-normalizer is a major computational obstacle in applications of log-linear models with large output spaces. The problem of fast normalizer computation has therefore attracted significant attention in the theoretical and applied machine learning literature. In this paper, we an…

Cited by 21SourcePDFScholar