← Search

Nikola Momchev

7 accepted papers

2025

BOND: Aligning LLMs with Best-of-N Distillation

ICLR 2025poster

Reinforcement learning from human feedback (RLHF) is a key driver of quality and safety in state-of-the-art large language models. Yet, a surprisingly simple and strong inference-time strategy is Best-of-N sampling that selects the best generation among N candidates. In this paper, we propose Best-o…

Cited by 26SourcePDFScholar
2024

Imitating Language via Scalable Inverse Reinforcement Learning

NeurIPS 2024poster

The majority of language model training builds on imitation learning. It covers pretraining, supervised fine-tuning, and affects the starting conditions for reinforcement learning from human feedback (RLHF). The simplicity and scalability of maximum likelihood estimation (MLE) for next token predict…

Cited by 8SourcePDFScholar
2024

Nash Learning from Human Feedback

ICML 2024spotlight

Reinforcement learning from human feedback (RLHF) has emerged as the main paradigm for aligning large language models (LLMs) with human preferences. Traditionally, RLHF involves the initial step of learning a reward model from pairwise human feedback, i.e., expressed as preferences between pairs of…

Cited by 129SourcePDFScholar
2023

Factually Consistent Summarization via Reinforcement Learning with Textual Entailment Feedback

ACL 2023long

Despite the seeming success of contemporary grounded text generation systems, they often tend to generate factually inconsistent text with respect to their input. This phenomenon is emphasized in tasks like summarization, in which the generated summaries should be corroborated by their source articl…

Cited by 82SourcePDFScholar
2021

*-CFQ: Analyzing the Scalability of Machine Learning on a Compositional Task

AAAI 2021technical

We present *-CFQ ("star-CFQ"): a suite of large-scale datasets of varying scope based on the CFQ semantic parsing benchmark, designed for principled investigation of the scalability of machine learning systems in a realistic compositional task setting. Using this suite, we conduct a series of experi…

2021

Hyperparameter Selection for Imitation Learning

ICML 2021oral

We address the issue of tuning hyperparameters (HPs) for imitation learning algorithms in the context of continuous-control, when the underlying reward function of the demonstrating expert cannot be observed at any time. The vast literature in imitation learning mostly considers this reward function…

2020

Measuring Compositional Generalization: A Comprehensive Method on Realistic Data

ICLR 2020poster

State-of-the-art machine learning methods exhibit limited compositional generalization. At the same time, there is a lack of realistic benchmarks that comprehensively measure this ability, which makes it challenging to find and evaluate improvements. We introduce a novel method to systematically con…

Cited by 406SourcecodeScholar