← Search

Andrew Wagenmaker

16 accepted papers

2026

Posterior Behavioral Cloning: Pretraining BC Policies for Efficient RL Finetuning

ICML 2026spotlight

Standard practice across domains from robotics to language is to first pretrain a policy on a large-scale demonstration dataset, and then finetune this policy, typically with reinforcement learning (RL), in order to improve performance on deployment domains. This finetuning step has proved critical …

Cited by 0SourceScholar
2026

Robust Fine-tuning of Vision-Language-Action Robot Policies via Parameter Merging

ICLR 2026poster

Generalist robot policies, trained on large and diverse datasets, have demonstrated the ability to generalize across a wide spectrum of behaviors, enabling a single policy to act in varied real-world environments. However, they still fall short on new tasks not covered in the training data. When fin…

Cited by 0SourceScholar
2025

Behavioral Exploration: Learning to Explore via In-Context Adaptation

ICML 2025poster

Developing autonomous agents that quickly explore an environment and adapt their behavior online is a canonical challenge in robotics and machine learning. While humans are able to achieve such fast online exploration and adaptation, often acquiring new information and skills in only a handful of in…

Cited by 0SourcePDFScholar
2025

Steering Your Diffusion Policy with Latent Space Reinforcement Learning

CoRL 2025oral

Robotic control policies learned from human demonstrations have achieved impressive results in many real-world applications. However, in scenarios where initial performance is not satisfactory, as is often the case in novel open-world settings, such behavioral cloning (BC)-learned policies typically…

Cited by 0SourceScholar
2024

ASID: Active Exploration for System Identification in Robotic Manipulation

ICLR 2024oral

Model-free control strategies such as reinforcement learning have shown the ability to learn control strategies without requiring an accurate model or simulator of the world. While this is appealing due to the lack of modeling requirements, such methods can be sample inefficient, making them impract…

Cited by 14SourcePDFScholar
2024

Active learning of neural population dynamics using two-photon holographic optogenetics

NeurIPS 2024poster

Recent advances in techniques for monitoring and perturbing neural populations have greatly enhanced our ability to study circuits in the brain. In particular, two-photon holographic optogenetics now enables precise photostimulation of experimenter-specified groups of individual neurons, while simu…

Cited by 0SourcePDFScholar
2024

Corruption-Robust Linear Bandits: Minimax Optimality and Gap-Dependent Misspecification

NeurIPS 2024poster

In linear bandits, how can a learner effectively learn when facing corrupted rewards? While significant work has explored this question, a holistic understanding across different adversarial models and corruption measures is lacking, as is a full characterization of the minimax regret bounds. In thi…

Cited by 1SourcePDFScholar
2024

Fair Active Learning in Low-Data Regimes

UAI 2024poster

In critical machine learning applications, ensuring fairness is essential to avoid perpetuating social inequities. In this work, we address the challenges of reducing bias and improving accuracy in data-scarce environments, where the cost of collecting labeled data prohibits the use of large, labele…

Cited by 4SourcePDFScholar
2024

Humor in AI: Massive Scale Crowd-Sourced Preferences and Benchmarks for Cartoon Captioning

NeurIPS 2024spotlight

We present a novel multimodal preference dataset for creative tasks, consisting of over 250 million human votes on more than 2.2 million captions, collected through crowdsourcing rating data for The New Yorker's weekly cartoon caption contest over the past eight years. This unique dataset supports t…

2024

Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RL

NeurIPS 2024poster

In order to mitigate the sample complexity of real-world reinforcement learning, common practice is to first train a policy in a simulator where samples are cheap, and then deploy this policy in the real world, with the hope that it generalizes effectively. Such \emph{direct sim2real} transfer is no…

Cited by 1SourcePDFScholar
2024

Sample Complexity Reduction via Policy Difference Estimation in Tabular Reinforcement Learning

NeurIPS 2024spotlight

In this paper, we study the non-asymptotic sample complexity for the pure exploration problem in contextual bandits and tabular reinforcement learning (RL): identifying an $\epsilon$-optimal policy from a set of policies $\Pi$ with high probability. Existing work in bandits has shown that it is poss…

Cited by 0SourcePDFScholar
2023

Optimal Exploration for Model-Based RL in Nonlinear Systems

NeurIPS 2023spotlight

Learning to control unknown nonlinear dynamical systems is a fundamental problem in reinforcement learning and control theory. A commonly applied approach is to first explore the environment (exploration), learn an accurate model of it (system identification), and then compute an optimal controller…

2022

Active Learning with Safety Constraints

NeurIPS 2022accept

Active learning methods have shown great promise in reducing the number of samples necessary for learning. As automated learning systems are adopted into real-time, real-world decision-making pipelines, it is increasingly important that such algorithms are designed with safety in mind. In this work…

Cited by 21SourcePDFScholar
2022

Instance-Dependent Near-Optimal Policy Identification in Linear MDPs via Online Experiment Design

NeurIPS 2022accept

While much progress has been made in understanding the minimax sample complexity of reinforcement learning (RL)---the complexity of learning on the ``worst-case'' instance---such measures of complexity often do not capture the true difficulty of learning. In practice, on an ``easy'' instance, we mig…

Cited by 36SourcePDFScholar
2021

Experimental Design for Regret Minimization in Linear Bandits

AISTATS 2021poster

In this paper we propose a novel experimental design-based algorithm to minimize regret in online stochastic linear and combinatorial bandits. While existing literature tends to focus on optimism-based algorithms–which have been shown to be suboptimal in many cases–our approach carefully plans which…

Cited by 18SourcePDFScholar