cs.CLApr 17, 2026

On the Rejection Criterion for Proxy-based Test-time Alignment

Authors: Ayoub HammalPierre ZweigenbaumCaio Corro

Organizations: Université Paris-Saclay, CNRS, LISN · INSA Rennes, IRISA, CNRS, Université de Rennes

Abstract

Recent works proposed test-time alignment methods that rely on a small aligned model as a proxy that guides the generation of a larger base (unaligned) model. The implicit reward approach skews the large model distribution, whereas the nudging approach defers the generation of the next token to the small aligned model when the large base one is unconfident about its outcome. In this work, we first show that both approaches can be reduced to sampling from similar graphical models, where they differ only in the definition of a rejection criterion (or distribution). Moreover, we argue that the confidence criterion is ill-motivated due to linguistic phenomena like ambiguous phrasing. We propose a novel rejection criterion based on a conservative confidence bet. Experimentally, our novel approach outperforms previous work on several datasets.

Explore similar work

CardsList
  1. Stress-testing Alignment Midtraining

    Sep 17, 2026Sid Baines, Jonathan Bostock, Maria Angelica Martinez +3LLM Post-Training MethodsFrontier Models

  2. Theoretical Limits of Language Model Alignment

    May 8, 2026Lucas Monteiro Paes, Natalie Mackraz, Barry-John Theobald +1Large Language Model AlignmentBregman Divergences