← Search

Lei M Zhang

4 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
2023

Human-Timescale Adaptation in an Open-Ended Task Space

ICML 2023oral

Foundation models have shown impressive adaptation and scalability in supervised and self-supervised learning problems, but so far these successes have not fully translated to reinforcement learning (RL). In this work, we demonstrate that training an RL agent at scale leads to a general in-context l…

Cited by 111SourcePDFScholar
2022

Proving Theorems using Incremental Learning and Hindsight Experience Replay

ICML 2022spotlight

Traditional automated theorem proving systems for first-order logic depend on speed-optimized search and many handcrafted heuristics designed to work over a wide range of domains. Machine learning approaches in the literature either depend on these traditional provers to bootstrap themselves, by lev…

Cited by 23SourcePDFScholar