← Search

John D. Co-Reyes

8 accepted papers

2025

Training Language Models to Self-Correct via Reinforcement Learning

ICLR 2025oral

Self-correction is a highly desirable capability of large language models (LLMs), yet it has consistently been found to be largely ineffective in modern LLMs. Current methods for training self-correction typically depend on either multiple models, a more advanced model, or additional forms of super…

Cited by 113SourcePDFScholar
2024

Many-Shot In-Context Learning

NeurIPS 2024spotlight

Large language models (LLMs) excel at few-shot in-context learning (ICL) -- learning from a few examples provided in context at inference, without any weight updates. Newly expanded context windows allow us to investigate ICL with hundreds or thousands of examples – the many-shot regime. Going from…

Cited by 115SourcePDFScholar
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

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
2021

Evolving Reinforcement Learning Algorithms

ICLR 2021oral

We propose a method for meta-learning reinforcement learning algorithms by searching over the space of computational graphs which compute the loss function for a value-based model-free RL agent to optimize. The learned algorithms are domain-agnostic and can generalize to new environments not seen du…

2021

Information is Power: Intrinsic Control via Information Capture

NeurIPS 2021poster

Humans and animals explore their environment and acquire useful skills even in the absence of clear goals, exhibiting intrinsic motivation. The study of intrinsic motivation in artificial agents is concerned with the following question: what is a good general-purpose objective for an agent? We study…

Cited by 11SourcePDFScholar
2019

Entity Abstraction in Visual Model-Based Reinforcement Learning

CoRL 2019

We present OP3, a framework for model-based reinforcement learning that acquires object representations from raw visual observations without supervision and uses them to predict and plan. To ground these abstract representations of entities to actual objects in the world, we formulate an interactive

2019

Guiding Policies with Language via Meta-Learning

ICLR 2019poster

Behavioral skills or policies for autonomous agents are conventionally learned from reward functions, via reinforcement learning, or from demonstrations, via imitation learning. However, both modes of task specification have their disadvantages: reward functions require manual engineering, while dem…

Cited by 75SourcePDFScholar