← Search

Disha Shrivastava

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

Repository-Level Prompt Generation for Large Language Models of Code

ICML 2023poster

With the success of large language models (LLMs) of code and their use as code assistants (e.g. Codex used in GitHub Copilot), techniques for introducing domain-specific knowledge in the prompt design process become important. In this work, we propose a framework called Repo-Level Prompt Generator t…

2021

Learning to Combine Per-Example Solutions for Neural Program Synthesis

NeurIPS 2021poster

The goal of program synthesis from examples is to find a computer program that is consistent with a given set of input-output examples. Most learning-based approaches try to find a program that satisfies all examples at once. Our work, by contrast, considers an approach that breaks the problem into…