← Search

Kelvin Xu

16 accepted papers

2025

LMRL Gym: Benchmarks for Multi-Turn Reinforcement Learning with Language Models

ICML 2025poster

Large language models (LLMs) provide excellent text-generation capabilities, but standard prompting and generation methods generally do not lead to intentional or goal-directed agents and might necessitate considerable prompt tuning. Even the best current LLMs rarely ask clarifying questions, engage…

2025

Scaling LLM Test-Time Compute Optimally Can be More Effective than Scaling Parameters for Reasoning

ICLR 2025oral

Enabling LLMs to improve their outputs by using more test-time compute is a critical step towards building self-improving agents that can operate on open-ended natural language. In this paper, we scale up inference-time computation in LLMs, with a focus on answering: if an LLM is allowed to use a fi…

Cited by 0SourcePDFScholar
2024

Small-scale proxies for large-scale Transformer training instabilities

ICLR 2024oral

Teams that have trained large Transformer-based models have reported training instabilities at large scale that did not appear when training with the same hyperparameters at smaller scales. Although the causes of such instabilities are of scientific interest, the amount of resources required to repr…

Cited by 79SourcePDFScholar
2023

Dexterous Manipulation from Images: Autonomous Real-World RL via Substep Guidance

ICRA 2023poster

Complex and contact-rich robotic manipulation tasks, particularly those that involve multi-fingered hands and underactuated object manipulation, present a significant challenge to any control method. Methods based on reinforcement learning offer an appealing choice for such settings, as they can ena…

Cited by 24SourceScholar
2022

Autonomous Reinforcement Learning: Formalism and Benchmarking

ICLR 2022poster

Reinforcement learning (RL) provides a naturalistic framing for learning through trial and error, which is appealing both because of its simplicity and effectiveness and because of its resemblance to how humans and animals acquire skills through experience. However, real-world embodied learning, suc…

2021

Reset-Free Reinforcement Learning via Multi-Task Learning: Learning Dexterous Manipulation Behaviors without Human Intervention

ICRA 2021poster

Reinforcement Learning (RL) algorithms can in principle acquire complex robotic skills by learning from large amounts of data in the real world, collected via trial and error. However, most RL algorithms use a carefully engineered setup in order to collect data, requiring human supervision and inter…

Cited by 118SourceScholar
2020

Continual Learning of Control Primitives : Skill Discovery via Reset-Games

NeurIPS 2020poster

Reinforcement learning has the potential to automate the acquisition of behavior in complex settings, but in order for it to be successfully deployed, a number of practical challenges must be addressed. First, in real world settings, when an agent attempts a tasks and fails, the environment must som…

2020

Meta-Dataset: A Dataset of Datasets for Learning to Learn from Few Examples

ICLR 2020poster

Few-shot classification refers to learning a classifier for new classes given only a few examples. While a plethora of models have emerged to tackle it, we find the procedure and datasets that are used to assess their progress lacking. To address this limitation, we propose Meta-Dataset: a new bench…

Cited by 788SourcecodeScholar
2019

Learning a Prior over Intent via Meta-Inverse Reinforcement Learning

ICML 2019oral

A significant challenge for the practical application of reinforcement learning to real world problems is the need to specify an oracle reward function that correctly defines a task. Inverse reinforcement learning (IRL) seeks to avoid this challenge by instead inferring a reward function from expert…

Cited by 88SourcePDFScholar
2018

Trust-PCL: An Off-Policy Trust Region Method for Continuous Control

ICLR 2018poster

Trust region methods, such as TRPO, are often used to stabilize policy optimization algorithms in reinforcement learning (RL). While current trust region strategies are effective for continuous control, they typically require a large amount of on-policy interaction with the environment. To address t…

2017

An Actor-Critic Algorithm for Sequence Prediction

ICLR 2017poster

We present an approach to training neural networks to generate sequences using actor-critic methods from reinforcement learning (RL). Current log-likelihood training methods are limited by the discrepancy between their training and testing modes, as models must generate tokens conditioned on their p…

Cited by 758SourcecodeScholar
2017

Bridging the Gap Between Value and Policy Based Reinforcement Learning

NeurIPS 2017poster

We establish a new connection between value and policy based reinforcement learning (RL) based on a relationship between softmax temporal value consistency and policy optimality under entropy regularization. Specifically, we show that softmax consistent action values correspond to optimal entropy re…

2015

Show, Attend and Tell: Neural Image Caption Generation with Visual Attention

ICML 2015poster

Inspired by recent work in machine translation and object detection, we introduce an attention based model that automatically learns to describe the content of images. We describe how we can train this model in a deterministic manner using standard backpropagation techniques and stochastically by ma…

Cited by 13573SourcePDFScholar