Consolidating Rewarded Perturbations for LLM Post-Training
Abstract
Post-training of language models is commonly framed as a sample-score-update loop implemented by gradient descent. A recent line of work, exemplified by RandOpt, relocates this loop to weight space, sampling Gaussian perturbations around a pretrained model and ensembling the top- rewarded specialists at inference. While competitive with PPO and GRPO under matched training compute, this prediction-level ensemble incurs forward passes per test example and does not extend cleanly to free-form generation. We ask whether the rewarded population can instead be folded into a single deployable model, replacing the inference-time ensemble with one consolidated update. A split-half analysis over 25 model-task pairs reveals reproducible low-rank structure in every case. We turn this geometry into CoRP (Consolidating Rewarded Perturbations), a gradient-free operator that combines reward-weighted aggregation, compatibility-aware reweighting, and a held-out validation gate, with no gradient flowing through the language model. Across five language models from B to B and five tasks covering math, code, and creative writing, CoRP improves the base model by points on average. Using one tenth of RandOpt's perturbation budget, CoRP exceeds single-inference RandOpt by points and recovers more than half of the gain of the 50-pass majority-vote ensemble, at one forward pass per test example.
Related papers
See more related submissions when the market opens.