← Search

Jonathan Lee

19 accepted papers

2026

Quadrotor Navigation Using Reinforcement Learning with Privileged Information

ICRA 2026poster

This paper presents a reinforcement learning-based quadrotor navigation method that leverages efficient differentiable simulation, novel loss functions, and privileged information to navigate around large obstacles. Prior learning-based methods perform well in scenes that exhibit narrow obstacles, b…

2025

EVOLvE: Evaluating and Optimizing LLMs For In-Context Exploration

ICML 2025poster

Despite their success in many domains, large language models (LLMs) remain under-studied in scenarios requiring optimal decision-making under uncertainty. This is crucial as many real-world applications, ranging from personalized recommendations to healthcare interventions, demand that LLMs not only…

Cited by 5SourcePDFScholar
2025

Towards Robust Mathematical Reasoning

EMNLP 2025

Finding the right north-star metrics is highly critical for advancing mathematical reasoning capabilities of foundation models, especially given that existing evaluations are either too easy or only focusing on getting correct short answers. To address these issues, we present IMO-Bench, a suite of

2023

Learning in POMDPs is Sample-Efficient with Hindsight Observability

ICML 2023poster

POMDPs capture a broad class of decision making problems, but hardness results suggest that learning is intractable even in simple settings due to the inherent partial observability. However, in many realistic problems, more information is either revealed or can be computed during some point of the…

Cited by 27SourcePDFScholar
2023

Supervised Pretraining Can Learn In-Context Reinforcement Learning

NeurIPS 2023spotlight

Large transformer models trained on diverse datasets have shown a remarkable ability to learn in-context, achieving high few-shot performance on tasks they were not explicitly trained to solve. In this paper, we study the in-context learning capabilities of transformers in decision-making problems,…

Cited by 80SourcePDFScholar
2022

Oracle Inequalities for Model Selection in Offline Reinforcement Learning

NeurIPS 2022accept

In offline reinforcement learning (RL), a learner leverages prior logged data to learn a good policy without interacting with the environment. A major challenge in applying such methods in practice is the lack of both theoretically principled and practical tools for model selection and evaluation. T…

Cited by 15SourcePDFScholar
2021

Design of Experiments for Stochastic Contextual Linear Bandits

NeurIPS 2021poster

In the stochastic linear contextual bandit setting there exist several minimax procedures for exploration with policies that are reactive to the data being acquired. In practice, there can be a significant engineering overhead to deploy these algorithms, especially when the dataset is collected in a…

Cited by 33SourcePDFScholar
2021

Online Model Selection for Reinforcement Learning with Function Approximation

AISTATS 2021poster

Deep reinforcement learning has achieved impressive successes yet often requires a very large amount of interaction data. This result is perhaps unsurprising, as using complicated function approximation often requires more data to fit, and early theoretical results on linear Markov decision processe…

Cited by 46SourcePDFScholar
2020

Accelerated Message Passing for Entropy-Regularized MAP Inference

ICML 2020poster

Maximum a posteriori (MAP) inference in discrete-valued Markov random fields is a fundamental problem in machine learning that involves identifying the most likely configuration of random variables given a distribution. Due to the difficulty of this combinatorial problem, linear programming (LP) rel…

Cited by 0SourcePDFScholar
2020

Convergence Rates of Smooth Message Passing with Rounding in Entropy-Regularized MAP Inference

AISTATS 2020poster

Maximum a posteriori (MAP) inference is a fundamental computational paradigm for statistical inference. In the setting of graphical models, MAP inference entails solving a combinatorial optimization problem to find the most likely configuration of the discrete-valued model. Linear programming (LP) r…

Cited by 3SourcePDFScholar
2020

Online Learning with Continuous Variations: Dynamic Regret and Reductions

AISTATS 2020poster

Online learning is a powerful tool for analyzing iterative algorithms. However, the classic adversarial setup fails to capture regularity that can exist in practice. Motivated by this observation, we establish a new setup, called Continuous Online Learning (COL), where the gradient of online loss f…

Cited by 21SourcePDFScholar
2019

On-Policy Robot Imitation Learning from a Converging Supervisor

CoRL 2019

Existing on-policy imitation learning algorithms, such as DAgger, assume access to a fixed supervisor. However, there are many settings where the supervisor may evolve during policy learning, such as a human performing a novel task or an improving algorithmic controller. We formalize imitation learn

2017

Comparing human-centric and robot-centric sampling for robot deep learning from demonstrations

ICRA 2017poster

Motivated by recent advances in Deep Learning for robot control, this paper considers two learning algorithms in terms of how they acquire demonstrations from fallible human supervisors. Human-Centric (HC) sampling is a standard supervised learning algorithm, where a human supervisor demonstrates th…

Cited by 89SourceScholar
2017

DART: Noise Injection for Robust Imitation Learning

CoRL 2017

One approach to Imitation Learning is Behavior Cloning, in which a robot observes a supervisor and infers a control policy. A known problem with this “off-policy" approach is that the robot’s errors compound when drifting away from the supervisor’s demonstrations. On-policy, techniques alleviate thi