Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Rohin Shah

dblp:145/1009 · DBLP profile ↗
← Back
12ranked-venue papers
2as first author
7since 2021 · last 2025
0000-0002-0656-2800ORCID · corroborated

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 11 · 2 first-author · 7 since 2021Software engineering, systems software and programming languages · 1Human-computer interaction and ubiquitous computing · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
11 papers
Reinforcement learning · 50% Trustworthy machine learning · 25% Language models and text generation · 13%
Human-computer interaction and pervasive computing
3 papers
Human-robot interaction · 75% Human-AI interaction · 25%

Topics — the 24 heaviest of 34, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Trustworthy machine learning
AI safety
1.322024
On scalable oversight with weak LLMs judging strong LLMs · NeurIPS 2024
Optimal Policies Tend To Seek Power · NeurIPS 2021
Machine learning › Reinforcement learning
reward design
1.222025
MONA: Myopic Optimization with Non-myopic Approval Can Mitigate Multi-step Reward Hacking · ICML 2025
Preferences Implicit in the State of the World · ICLR (Poster) 2019
Machine learning › Reinforcement learning
imitation learning
1.122023
BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Tasks · NeurIPS 2023
The MAGICAL Benchmark for Robust Imitation · NeurIPS 2020
Natural language and speech › Language models and text generation › alignment
reward hacking
0.912025
MONA: Myopic Optimization with Non-myopic Approval Can Mitigate Multi-step Reward Hacking · ICML 2025
Machine learning › Trustworthy machine learning
interpretability
0.812024
Improving Sparse Decomposition of Language Model Activations with Gated Sparse Autoencoders · NeurIPS 2024
Machine learning › Reinforcement learning › imitation learning
inverse reinforcement learning
0.822019
On the Feasibility of Learning, Rather than Assuming, Human Biases for Reward Inference · ICML 2019
Preferences Implicit in the State of the World · ICLR (Poster) 2019
Machine learning › Trustworthy machine learning › interpretability
mechanistic interpretability
0.812024
Improving Sparse Decomposition of Language Model Activations with Gated Sparse Autoencoders · NeurIPS 2024
Natural language and speech › Language models and text generation › alignment
scalable oversight
0.812024
On scalable oversight with weak LLMs judging strong LLMs · NeurIPS 2024
Machine learning › Trustworthy machine learning › interpretability › mechanistic interpretability
sparse autoencoder
0.812024
Improving Sparse Decomposition of Language Model Activations with Gated Sparse Autoencoders · NeurIPS 2024
Machine learning › Representation and self-supervised learning › representation learning › unsupervised representation learning › sparse coding
sparse feature learning
0.812024
Improving Sparse Decomposition of Language Model Activations with Gated Sparse Autoencoders · NeurIPS 2024
Machine learning › Reinforcement learning
demonstration dataset
0.712023
BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Tasks · NeurIPS 2023
Machine learning › Reinforcement learning › reinforcement learning from human feedback
learning from human feedback
0.712023
BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Tasks · NeurIPS 2023
Machine learning › Reinforcement learning
reward learning
0.712023
SIRL: Similarity-based Implicit Representation Learning · HRI 2023
Machine learning › Reinforcement learning
model-based reinforcement learning
0.512021
Learning What To Do by Simulating the Past · ICLR 2021
Machine learning › Reinforcement learning › imitation learning
robust imitation learning
0.412020
The MAGICAL Benchmark for Robust Imitation · NeurIPS 2020
Machine learning › Trustworthy machine learning › robustness › distribution shift
robustness to distribution shift
0.412020
The MAGICAL Benchmark for Robust Imitation · NeurIPS 2020
Machine learning › Reinforcement learning › multi-agent reinforcement learning
human-AI collaboration
0.412019
On the Utility of Learning about Humans for Human-AI Coordination · NeurIPS 2019
Machine learning › Reinforcement learning › imitation learning › inverse reinforcement learning
inverse optimal control
0.412019
On the Feasibility of Learning, Rather than Assuming, Human Biases for Reward Inference · ICML 2019
Machine learning › Optimization for machine learning › hyperparameter optimization
population-based training
0.412019
On the Utility of Learning about Humans for Human-AI Coordination · NeurIPS 2019
Knowledge, reasoning and agents › Knowledge representation and reasoning › nonmonotonic reasoning › preference handling › preference reasoning
preference inference
0.412019
Preferences Implicit in the State of the World · ICLR (Poster) 2019
Machine learning › Reinforcement learning › multi-agent reinforcement learning
self-play
0.412019
On the Utility of Learning about Humans for Human-AI Coordination · NeurIPS 2019
Natural language and speech › Language models and text generation
large language model
0.212024
On scalable oversight with weak LLMs judging strong LLMs · NeurIPS 2024
Machine learning › Reinforcement learning
agent evaluation
0.212023
BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Tasks · NeurIPS 2023
Machine learning › Representation and self-supervised learning
contrastive learning
0.212023
SIRL: Similarity-based Implicit Representation Learning · HRI 2023

Methods — techniques the papers use, named apart from their topics

user study · 2.1myopic optimization with non-myopic approval · 1.7contrastive learning · 1.3weak-to-strong supervision · 0.8l1 penalty · 0.8gated sparse autoencoder · 0.8debate · 0.8consultancy · 0.8human evaluation · 0.7dynamics model · 0.5self-play · 0.4population-based training · 0.4planning algorithm · 0.4sketch-based synthesis · 0.4program partitioning · 0.4
YearPublicationVenuePosition
2025 MONA: Myopic Optimization with Non-myopic Approval Can Mitigate Multi-step Reward Hacking
abstract
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") even if humans are not able to detect that the behavior is undesired. The method, Myopic Optimization with Non-myopic Approval (MONA), works by combining short-sighted optimization with far-sighted reward. We demonstrate that MONA can prevent multi-step reward hacking that ordinary RL causes, even without being able to detect the reward hacking and without any extra information that ordinary RL does not get access to. We study MONA empirically in three settings which model different misalignment failure modes including 2-step environments with LLMs representing delegated oversight and encoded reasoning and longer-horizon gridworld environments representing sensor tampering.
Sebastian Farquhar, Vikrant Varma, David Lindner, David Elson, Caleb Biddulph, Ian Goodfellow, Rohin Shah
ICML7
2024 On scalable oversight with weak LLMs judging strong LLMs
abstract
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, where the judge just answers outright without the AI. We use large language models (LLMs) as both AI agents and as stand-ins for human judges, taking the judge models to be weaker than agent models. We benchmark on a diverse range of asymmetries between judges and agents, extending previous work on a single extractive QA task with information asymmetry, to also include mathematics, coding, logic and multimodal reasoning asymmetries. We find that debate outperforms consultancy across all tasks when the consultant is randomly assigned to argue for the correct/incorrect answer. Comparing debate to direct question answering, the results depend on the type of task: in extractive QA tasks with information asymmetry debate outperforms direct question answering, but in other tasks without information asymmetry the results are mixed. Previous work assigned debaters/consultants an answer to argue for. When we allow them to instead choose which answer to argue for, we find judges are less frequently convinced by the wrong answer in debate than in consultancy. Further, we find that stronger debater models increase judge accuracy, though more modestly than in previous studies.
Zachary Kenton, Noah Y. Siegel, János Kramár, Jonah Brown-Cohen, Samuel Albanie, Jannis Bulian, Rishabh Agarwal, David Lindner, Yunhao Tang, Noah D. Goodman, Rohin Shah
NeurIPS11
2024 Improving Sparse Decomposition of Language Model Activations with Gated Sparse Autoencoders
abstract
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 achieves a Pareto improvement over training with prevailing methods. In SAEs, the L1 penalty used to encourage sparsity introduces many undesirable biases, such as shrinkage -- systematic underestimation of feature activations. The key insight of Gated SAEs is to separate the functionality of (a) determining which directions to use and (b) estimating the magnitudes of those directions: this enables us to apply the L1 penalty only to the former, limiting the scope of undesirable side effects. Through training SAEs on LMs of up to 7B parameters we find that, in typical hyper-parameter ranges, Gated SAEs solve shrinkage, are similarly interpretable, and require half as many firing features to achieve comparable reconstruction fidelity.
Senthooran Rajamanoharan, Arthur Conmy, Lewis Smith, Tom Lieberum, Vikrant Varma, János Kramár, Rohin Shah, Neel Nanda
NeurIPS7
2023 SIRL: Similarity-based Implicit Representation Learning
abstract
When robots learn reward functions using high capacity models that take raw state directly as input, they need to both learn a representation for what matters in the task --- the task "features" --- as well as how to combine these features into a single objective. If they try to do both at once from input designed to teach the full reward function, it is easy to end up with a representation that contains spurious correlations in the data, which fails to generalize to new settings. Instead, our ultimate goal is to enable robots to identify and isolate the causal features that people actually care about and use when they represent states and behavior. Our idea is that we can tune into this representation by asking users what behaviors they consider similar: behaviors will be similar if the features that matter are similar, even if low-level behavior is different; conversely, behaviors will be different if even one of the features that matter differs. This, in turn, is what enables the robot to disambiguate between what needs to go into the representation versus what is spurious, as well as what aspects of behavior can be compressed together versus not. The notion of learning representations based on similarity has a nice parallel in contrastive learning, a self-supervised representation learning technique that maps visually similar data points to similar embeddings, where similarity is defined by a designer through data augmentation heuristics. By contrast, in order to learn the representations that people use, so we can learn their preferences and objectives, we use their definition of similarity. In simulation as well as in a user study, we show that learning through such similarity queries leads to representations that, while far from perfect, are indeed more generalizable than self-supervised and task-input alternatives.
Andreea Bobu, Rohin Shah, Daniel S. Brown, Anca D. Dragan
HRI3
2023 BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Tasks
abstract
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 the BASALT Evaluation and Demonstrations Dataset (BEDD), which serves as a resource for algorithm development and performance assessment. BEDD consists of a collection of 26 million image-action pairs from nearly 14,000 videos of human players completing the BASALT tasks in Minecraft. It also includes over 3,000 dense pairwise human evaluations of human and algorithmic agents. These comparisons serve as a fixed, preliminary leaderboard for evaluating newly-developed algorithms. To enable this comparison, we present a streamlined codebase for benchmarking new algorithms against the leaderboard. In addition to presenting these datasets, we conduct a detailed analysis of the data from both datasets to guide algorithm development and evaluation. The released code and data are available at https://github.com/minerllabs/basalt-benchmark.
Stephanie Milani, Anssi Kanervisto, Karolis Ramanauskas, Sander Schulhoff, Brandon Houghton, Rohin Shah
NeurIPS6
2021 Learning What To Do by Simulating the Past
David Lindner, Rohin Shah, Pieter Abbeel, Anca D. Dragan
ICLR2
2021 Optimal Policies Tend To Seek Power
abstract
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, we develop the first formal theory of the statistical tendencies of optimal policies. In the context of Markov decision processes, we prove that certain environmental symmetries are sufficient for optimal policies to tend to seek power over the environment. These symmetries exist in many environments in which the agent can be shut down or destroyed. We prove that in these environments, most reward functions make it optimal to seek power by keeping a range of options available and, when maximizing average reward, by navigating towards larger sets of potential terminal states.
Alexander Matt Turner, Logan Smith, Rohin Shah, Andrew Critch, Prasad Tadepalli
NeurIPS3
2020 The MAGICAL Benchmark for Robust Imitation
abstract
Imitation Learning (IL) algorithms are typically evaluated in the same environment that was used to create demonstrations. This rewards precise reproduction of demonstrations in one particular environment, but provides little information about how robustly an algorithm can generalise the demonstrator's intent to substantially different deployment settings. This paper presents the MAGICAL benchmark suite, which permits systematic evaluation of generalisation by quantifying robustness to different kinds of distribution shift that an IL algorithm is likely to encounter in practice. Using the MAGICAL suite, we confirm that existing IL algorithms overfit significantly to the context in which demonstrations are provided. We also show that standard methods for reducing overfitting are effective at creating narrow perceptual invariances, but are not sufficient to enable transfer to contexts that require substantially different behaviour, which suggests that new approaches will be needed in order to robustly generalise demonstrator intent. Code and data for the MAGICAL suite is available at https://github.com/qxcv/magical/
Sam Toyer, Rohin Shah, Andrew Critch, Stuart Russell 0001
NeurIPS2
2019 Preferences Implicit in the State of the World
Rohin Shah, Dmitrii Krasheninnikov, Jordan Alexander, Pieter Abbeel, Anca D. Dragan
ICLR (Poster)1
2019 On the Feasibility of Learning, Rather than Assuming, Human Biases for Reward Inference
abstract
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 other hand, often have systematic biases: risk-aversion, myopia, etc. One option is to try to characterize these biases and account for them explicitly during learning. But in the era of deep learning, a natural suggestion researchers make is to avoid mathematical models of human behavior that are fraught with specific assumptions, and instead use a purely data-driven approach. We decided to put this to the test – rather than relying on assumptions about which specific bias the demonstrator has when planning, we instead learn the demonstrator’s planning algorithm that they use to generate demonstrations, as a differentiable planner. Our exploration yielded mixed findings: on the one hand, learning the planner can lead to better reward inference than relying on the wrong assumption; on the other hand, this benefit is dwarfed by the loss we incur by going from an exact to a differentiable planner. This suggest that at least for the foreseeable future, agents need a middle ground between the flexibility of data-driven methods and the useful bias of known human biases. Code is available at https://tinyurl.com/learningbiases.
Rohin Shah, Noah Gundotra, Pieter Abbeel, Anca D. Dragan
ICML1
2019 On the Utility of Learning about Humans for Human-AI Coordination
abstract
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 understand and be understood by humans. To demonstrate this, we introduce a simple environment that requires challenging coordination, based on the popular game Overcooked, and learn a simple model that mimics human play. We evaluate the performance of agents trained via self-play and population-based training. These agents perform very well when paired with themselves, but when paired with our human model, they are significantly worse than agents designed to play with the human model. An experiment with a planning algorithm yields the same conclusion, though only when the human-aware planner is given the exact human model that it is playing with. A user study with real humans shows this pattern as well, though less strongly. Qualitatively, we find that the gains come from having the agent adapt to the human's gameplay. Given this result, we suggest several approaches for designing agents that learn about humans in order to better coordinate with them. Code is available at https://github.com/HumanCompatibleAI/overcooked_ai.
Micah Carroll, Rohin Shah, Mark K. Ho, Thomas L. Griffiths 0001, Sanjit A. Seshia, Pieter Abbeel, Anca D. Dragan
NeurIPS2
2014 Chlorophyll: synthesis-aided compiler for low-power spatial architectures
abstract
We developed Chlorophyll, a synthesis-aided programming model and compiler for the GreenArrays GA144, an extremely minimalist low-power spatial architecture that requires partitioning the program into fragments of no more than 256 instructions and 64 words of data. This processor is 100-times more energy efficient than its competitors, but currently can only be programmed using a low-level stack-based language.
Phitchaya Mangpo Phothilimthana, Tikhon Jelvis, Rohin Shah, Nishant Totla, Sarah E. Chasins, Rastislav Bodík
PLDI3