← Search

Andrew Cohen

6 accepted papers

2026

LLM Pretraining with Continuous Concepts

ICLR 2026poster

Next token prediction has been the standard training objective used in large language model pretraining. Representations are learned as a result of optimizing for token-level perplexity. We propose Continuous Concept Mixing (CoCoMix), a novel pretraining framework that combines discrete next token p…

Cited by 0SourcecodeScholar
2026

Reuse your FLOPs: Scaling RL on Hard Problems by Conditioning on Very Off-Policy Prefixes

ICML 2026poster

Typical reinforcement learning (RL) methods for LLM reasoning waste compute on hard problems, where correct on-policy traces are rare and policy gradients vanish. To bootstrap more efficient RL, we consider reusing old sampling FLOPs (from prior inference or RL training) in the form of off-policy tr…

Cited by 0SourceScholar
2025

reWordBench: Benchmarking and Improving the Robustness of Reward Models with Transformed Inputs

EMNLP 2025

Reward models have become a staple in modern NLP, serving as not only a scalable text evaluator, but also an indispensable component in many alignment recipes and inference-time algorithms. However, while recent reward models increase performance on standard benchmarks, this may partly be due to ove

2024

Learning Personalized Alignment for Evaluating Open-ended Text Generation

EMNLP 2024main

Recent research has increasingly focused on evaluating large language models’ (LLMs) alignment with diverse human values and preferences, particularly for open-ended tasks like story generation. Traditional evaluation metrics rely heavily on lexical similarity with human-written references, often sh…

2024

The ART of LLM Refinement: Ask, Refine, and Trust

NAACL 2024long

Large Language Models (LLMs) have demonstrated remarkable generative abilities, but can they judge the quality of their own generations and self-improve?A popular concept, referred to as *self-refinement*, postulates that LLMs can detect and correct the errors in their generations when asked to do s…

2024

To the Globe (TTG): Towards Language-Driven Guaranteed Travel Planning

EMNLP 2024system demonstrations

Travel planning is a challenging and time-consuming task that aims to find an itinerary which satisfies multiple, interdependent constraints regarding flights, accommodations, attractions, and other travel arrangements. In this paper, we propose To the Globe (TTG), a real-time demo system that takes…