← Search

Daniel D. Johnson

7 accepted papers

2025

Eliciting Language Model Behaviors with Investigator Agents

ICML 2025poster

Language models exhibit complex, diverse behaviors when prompted with free-form text, making it hard to characterize the space of possible outputs. We study the problem of behavioral elicitation, where the goal is to search for prompts that induce specific target behaviors (e.g., hallucinations, har…

Cited by 3SourcePDFScholar
2024

Experts Don't Cheat: Learning What You Don't Know By Predicting Pairs

ICML 2024poster

Identifying how much a model $\hat{p}\_{Y|X}^{\theta}$ knows about the stochastic real-world process $p\_{Y|X}$ it was trained on is important to ensure it avoids producing incorrect or "hallucinated" answers or taking unsafe actions. But this is difficult for generative models because probabilistic…

Cited by 9SourcePDFScholar
2023

Contrastive Learning Can Find An Optimal Basis For Approximately View-Invariant Functions

ICLR 2023poster

Contrastive learning is a powerful framework for learning self-supervised representations that generalize well to downstream supervised tasks. We show that multiple existing contrastive learning methods can be reinterpeted as learning kernel functions that approximate a fixed *positive-pair kernel*.…

Cited by 28SourcePDFScholar
2023

R-U-SURE? Uncertainty-Aware Code Suggestions By Maximizing Utility Across Random User Intents

ICML 2023poster

Large language models show impressive results at predicting structured text such as code, but also commonly introduce errors and hallucinations in their output. When used to assist software developers, these models may make mistakes that users must go back and fix, or worse, introduce subtle bugs th…

2021

Learning Generalized Gumbel-max Causal Mechanisms

NeurIPS 2021spotlight

To perform counterfactual reasoning in Structural Causal Models (SCMs), one needs to know the causal mechanisms, which provide factorizations of conditional distributions into noise sources and deterministic functions mapping realizations of noise to samples. Unfortunately, the causal mechanism is n…

2021

Structured Denoising Diffusion Models in Discrete State-Spaces

NeurIPS 2021poster

Denoising diffusion probabilistic models (DDPMs) [Ho et al. 2021] have shown impressive results on image and waveform generation in continuous state spaces. Here, we introduce Discrete Denoising Diffusion Probabilistic Models (D3PMs), diffusion-like generative models for discrete data that generaliz…

Cited by 1001SourcePDFScholar
2017

Learning Graphical State Transitions

ICLR 2017oral

Graph-structured data is important in modeling relationships between multiple entities, and can be used to represent states of the world as well as many data structures. Li et al. (2016) describe a model known as a Gated Graph Sequence Neural Network (GGS-NN) that produces sequences from graph-struc…

Cited by 131SourceScholar