Search papers, labs, and topics across Lattice.
This paper introduces an adversarial generator-discriminator framework that enhances reinforcement learning with verifiable rewards (RLVR) by incorporating feedback from human demonstrations to address the shortcomings of current RLVR methods. By training a generator to optimize both task accuracy and an adversarial reward signal derived from a discriminator, the authors successfully improve non-verifiable aspects of language model outputs, such as style and diversity, while maintaining high performance on verifiable tasks. The results demonstrate significant advancements in bug fixing and story generation, reducing edit distances and increasing the human-likeness of generated content, thus bridging the gap between RL and supervised fine-tuning (SFT).
Adversarial training with human demonstrations can significantly enhance the quality and diversity of language model outputs while preserving accuracy.
RL with verifiable rewards (RLVR) has emerged as a powerful paradigm for training LMs on tasks with well-defined success metrics, such as code generation and mathematical reasoning. However, current RLVR methods optimize only what can be objectively scored, often neglecting subjective, non-verifiable aspects of human-like outputs, such as style and structure. This limitation leads to well-documented failure modes such as diversity collapse, unnatural-sounding responses, and reward hacking. We propose an adversarial generator-discriminator framework that augments verifiable rewards with a learned signal from human demonstrations. A generator model is trained using RL to maximize both task accuracy and an adversarial reward derived from a discriminator. The discriminator, trained alongside the generator policy, learns to distinguish human-written outputs from model-generated ones. The discriminator serves as a learned proxy for the human output distribution, providing feedback on aspects of generation that are difficult to formalize as scalar rewards. Across diverse domains, including bug fixing and open-ended generation, our approach consistently improves non-verifiable properties while preserving the accuracy gains of RLVR. In bug fixing, our method produces solutions with significantly lower edit distance compared to RLVR baselines while matching end performance. In story generation, our method significantly improves win rate while producing stories that are diverse and more human-like. And in a simple reward hacking benchmark, our method nearly eliminates model misbehavior while maintaining high benchmark scores. Together, these results show that our approach bridges RL and SFT, offering a scalable path toward jointly optimizing the verifiable and non-verifiable properties of a task.
MIT CSAIL