← Search

Rowan Thomas McAllister

6 accepted papers

2023

Waymax: An Accelerated, Data-Driven Simulator for Large-Scale Autonomous Driving Research

NeurIPS 2023poster

Simulation is an essential tool to develop and benchmark autonomous vehicle planning software in a safe and cost-effective manner. However, realistic simulation requires accurate modeling of multi-agent interactive behaviors to be trustworthy, behaviors which can be highly nuanced and complex. To ad…

Cited by 116SourcePDFScholar
2022

Dynamics-Aware Comparison of Learned Reward Functions

ICLR 2022spotlight

The ability to learn reward functions plays an important role in enabling the deployment of intelligent agents in the real world. However, $\textit{comparing}$ reward functions, for example as a means of evaluating reward learning methods, presents a challenge. Reward functions are typically compare…

Cited by 26SourcePDFScholar
2022

Is Anyone There? Learning a Planner Contingent on Perceptual Uncertainty

CoRL 2022poster

Robots in complex multi-agent environments should reason about the intentions of observed and currently unobserved agents. In this paper, we present a new learning-based method for prediction and planning in complex multi-agent environments where the states of the other agents are partially-observed…

Cited by 14SourceScholar
2022

RAP: Risk-Aware Prediction for Robust Planning

CoRL 2022oral

Robust planning in interactive scenarios requires predicting the uncertain future to make risk-aware decisions. Unfortunately, due to long-tail safety-critical events, the risk is often under-estimated by finite-sampling approximations of probabilistic motion forecasts. This can lead to overconfiden…

Cited by 17SourcecodeScholar
2021

Learning Invariant Representations for Reinforcement Learning without Reconstruction

ICLR 2021oral

We study how representation learning can accelerate reinforcement learning from rich observations, such as images, without relying either on domain knowledge or pixel-reconstruction. Our goal is to learn representations that provide for effective downstream control and invariance to task-irrelevant…

2021

Outcome-Driven Reinforcement Learning via Variational Inference

NeurIPS 2021poster

While reinforcement learning algorithms provide automated acquisition of optimal policies, practical application of such methods requires a number of design decisions, such as manually designing reward functions that not only define the task, but also provide sufficient shaping to accomplish it. In…

Cited by 20SourcePDFScholar