← Search

Rohin Shah

11 accepted papers

2025

MONA: Myopic Optimization with Non-myopic Approval Can Mitigate Multi-step Reward Hacking

ICML 2025poster

Future advanced AI systems may learn sophisticated strategies through reinforcement learning (RL) that humans cannot understand well enough to safely evaluate. We propose a training method which avoids agents learning undesired multi-step plans that receive high reward (multi-step "reward hacks") ev…

Cited by 1SourcePDFScholar
2024

Improving Sparse Decomposition of Language Model Activations with Gated Sparse Autoencoders

NeurIPS 2024poster

Recent work has found that sparse autoencoders (SAEs) are an effective technique for unsupervised discovery of interpretable features in language models' (LMs) activations, by finding sparse, linear reconstructions of those activations. We introduce the Gated Sparse Autoencoder (Gated SAE), which ac…

Cited by 6SourcePDFScholar
2024

On scalable oversight with weak LLMs judging strong LLMs

NeurIPS 2024poster

Scalable oversight protocols aim to enable humans to accurately supervise superhuman AI. In this paper we study debate, where two AI's compete to convince a judge; consultancy, where a single AI tries to convince a judge that asks questions; and compare to a baseline of direct question-answering,…

Cited by 31SourcePDFScholar
2023

BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Tasks

NeurIPS 2023oral

The MineRL BASALT competition has served to catalyze advances in learning from human feedback through four hard-to-specify tasks in Minecraft, such as create and photograph a waterfall. Given the completion of two years of BASALT competitions, we offer to the community a formalized benchmark through…

2021

An Empirical Investigation of Representation Learning for Imitation

NeurIPS 2021poster

Imitation learning often needs a large demonstration set in order to handle the full range of situations that an agent might find itself in during deployment. However, collecting expert demonstrations can be expensive. Recent work in vision, reinforcement learning, and NLP has shown that auxiliary r…

Cited by 33SourceScholar
2021

Optimal Policies Tend To Seek Power

NeurIPS 2021spotlight

Some researchers speculate that intelligent reinforcement learning (RL) agents would be incentivized to seek resources and power in pursuit of the objectives we specify for them. Other researchers point out that RL agents need not have human-like power-seeking instincts. To clarify this discussion,…

2019

On the Feasibility of Learning, Rather than Assuming, Human Biases for Reward Inference

ICML 2019oral

Our goal is for agents to optimize the right reward function, despite how difficult it is for us to specify what that is. Inverse Reinforcement Learning (IRL) enables us to infer reward functions from demonstrations, but it usually assumes that the expert is noisily optimal. Real people, on the othe…

Cited by 81SourcePDFScholar
2019

On the Utility of Learning about Humans for Human-AI Coordination

NeurIPS 2019poster

While we would like agents that can coordinate with humans, current algorithms such as self-play and population-based training create agents that can coordinate with themselves. Agents that assume their partner to be optimal or similar to them can converge to coordination protocols that fail to unde…

2019

Preferences Implicit in the State of the World

ICLR 2019poster

Reinforcement learning (RL) agents optimize only the features specified in a reward function and are indifferent to anything left out inadvertently. This means that we must not only specify what to do, but also the much larger space of what not to do. It is easy to forget these preferences, since th…