alphaFeedback

WorksComputer Science

ECHO: Emergent Common-Mode Hidden Offsets Drive Vocabulary Collapse in Preference Optimization

AI contributions
  • Claude Code
  • Gemini
View details

Abstract

Preference optimization can improve every ranking metric while building a hidden, shared representation shift that the relative loss barely sees, one that can eventually destroy generation. We trace this failure to the common-mode offset, a shared directional shift in the model's hidden states that appears across token positions, prompts, and both chosen and rejected responses. Because it moves chosen and rejected representations together, the relative ranking loss barely reacts to its accumulation. As this offset grows, it projects onto the output layer to form a vocabulary well, elevating a single token's probability until it dominates decoding and causes generation collapse. The isolated offset alone reconstructs the vocabulary well, injecting the learned offset into an untreated base model recreates the collapse where norm-matched random directions do not, and masking this common component from the loss prevents the well in our flagship setting. Under the per-token ranking loss we study, the same collapse appears in 14 of 17 base-model lineages and under full-parameter fine-tuning. Moreover, the offset is not merely an artifact of collapse, since it is already present in textbook DPO and SimPO runs that form no well. Textbook DPO keeps it weakly aligned with any single vocabulary direction even with three times our base budget. Trained from base models for that budget, however, full-parameter SimPO drives it into collapse on both Llama-2 and Llama-3.1, and the same kind of offset grows, without a well, from a public SFT checkpoint under the official SimPO recipe. Across 17 UltraFeedback runs, the better a run ranks held-out responses, the less often two public reward models prefer its samples to the base model's. These findings expose a blind spot in standard ranking metrics and introduce a representation-level signal that can warn of generation collapse before it occurs.

Introduction

Preference optimization has become the standard paradigm for adapting pretrained language models toward human-preferred responses ((Ouyang et al., 2022)), ranging from pairwise Direct Preference Optimization (DPO) ((Rafailov et al., 2023)) to listwise objectives based on Plackett–Luce ranking ([24, 22] (Plackett 1975)). Recent pipelines push this ranking signal further, with repeated rounds of preference optimization on continually refreshed preference data ((Yuan et al., 2024; Graffatiori et al., 2024)). However, recent work reveals a growing disconnect between successful ranking optimization and healthy text generation ((Rafailov et al., 2024)). Razin et al. (2025) identify likelihood displacement, where the absolute likelihood of the chosen response can plummet even as its relative preference margin improves. Ren & Sutherland (2025) show that suppressing rejected tokens can undesirably squeeze probability mass toward tokens the base model already favored. Yuan et al. (2025) mathematically explain why chosen and rejected likelihoods move together via gradient entanglement. While these studies illuminate how probabilities shift during ranking, the terminal state of this dynamic remains largely unexplained. How does this movement manifest inside the model, why does it eventually fixate on a seemingly irrelevant token, and why do standard ranking metrics look healthy while generation breaks down?

We find that preference training induces a surprisingly simple internal anomaly. The final hidden states begin to drift by a shared vector across token positions, prompts, and both chosen and rejected responses. We term this shared movement the common-mode offset. Because this offset moves the scores of chosen and rejected responses by nearly the same amount, a ranking loss that optimizes their relative difference has almost no incentive to penalize it. Consequently, this shared shift can keep growing in the background, masked by improving preference margins.

This hidden shift becomes destructive because the output layer projects these hidden states into vocabulary logits. When the offset pushes a single token’s logit far higher than the tokens the base model would predict, it creates a vocabulary well. Once this well becomes deep enough to dictate decoding, the model enters generation collapse, uncontrollably repeating that single token. Importantly, the well need not be a rare token with an unusually large output-vector norm, since the same mechanism produces a well on the ordinary token Conf on UltraFeedback ([5]).

Causal interventions confirm that this offset causes the well. Keeping only its common component reconstructs the well, injecting it into an untreated base model recreates the collapse where norm-matched random directions do not, and centering the rejected-side loss to hide it prevents the well in the flagship setting (§ 5). The offset moves both chosen and rejected scores by tens of nats but their margin by less than 3 nat in an average list, which is why held-out ranking improves while generation collapses (Figure 1). Across 17 UltraFeedback runs, the better a run ranks held-out responses, the less often two public reward models prefer its samples to the base model’s (§ 8).

Three-panel illustration of ranking accuracy, common-mode offset, and vocabulary well

Figure 1. (a) Two Llama-2 runs on the same driving lists rank held-out responses equally well, but only the one without offset control collapses onto one token. (b) The common-mode offset bˉ\bar{b} moves chosen and rejected hidden states together, so the margin the loss sees barely changes. (c) The offset lifts a rare token above the base model’s top token once its gain gtg_t exceeds the base gap mtm_t.

This vulnerability is not an artifact of one unusual hyperparameter setting. Our main objective is the ranking form of PRO (Song et al. (2024)), and we observe the collapse signature in 14 of 17 base-model lineages and under full-parameter fine-tuning (§ 4). Moreover, the common-mode offset grows under every objective we test, including textbook DPO and SimPO, and the objectives differ in how hard they push it. We measure this push by the loss’s demand, the per-token margin beyond which it stops pushing. Textbook DPO at its usual setting has a low demand, so its offset stays weakly aligned with any vocabulary direction and generation stays intact. This protection does not come from the reference model, since adding a reference model to the per-token objective does not prevent the well (§ 6). SimPO lies between the two. Trained from base models for three times our base budget, full-parameter SimPO drives the offset into collapse, and the offset also grows, without forming a well, from a public SFT checkpoint under the official SimPO recipe.

Therefore, generation collapse is not an inevitability of preference optimization, but the extreme terminal state of a hidden dynamic that is active long before generation visibly fails. Hiding the offset from the loss and penalizing it prevents collapse up to three times the base budget in the four base models we tested.

Prior work showed that preference optimization can move probability mass where it should not go and couples chosen and rejected updates. We link these probabilistic symptoms to one translation vector in the final hidden state that the loss barely sees, show that it causes the well, and give a necessary condition for its onset. This exposes a blind spot in ranking metrics.

Related Work

Likelihood displacement, squeezing, and gradient entanglement. Razin et al. (2025) demonstrate that DPO training can decrease the chosen response’s absolute likelihood and predict when a dataset risks it from the similarity of the responses’ hidden embeddings. Yuan et al. (2025) prove that margin-based losses entangle the gradients of chosen and rejected log-probabilities, explaining why the two tend to move in tandem. While these works map the symptoms in probability space, we isolate in representation space the common-mode offset of the final hidden state, which produces the well. Furthermore, Ren & Sutherland (2025) show that the negative gradient on rejected tokens “squeezes” probability mass toward tokens the base model is already confident about. The vocabulary wells we discover are different, since the base model initially ranks our well tokens far below its preferred tokens, typically by 17 to 39 nat, until the offset inflates their logits to dominate decoding.

Regularizing absolute likelihood. DPO-Positive (Pal et al., 2024), an explicit NLL term (Pang et al., 2024), and N-DPO, a length-normalized regularization toward the reference (Im et al., [16]), constrain scalar probabilities rather than the shared direction in the hidden state. In our experiments, an NLL term on the chosen response does not prevent the well (§7).

Unembedding geometry and representation anisotropy. Pretrained embeddings and hidden states share a common mean, a few dominant directions, and outlier dimensions ([7, 2, 9, 25] (Mu & Viswanath 2018; Puccetti et al. 2022; Sun et al. 2024; Stolfo et al. 2024)), whereas the common-mode offset is built during preference training. Functionally, it resembles activation steering (Rimsky et al., 2024), and DPO is known to change behavior by shifting the residual stream (Lee et al., [20]), but here preference optimization grows the steering vector on its own until generation collapses. Many of our severe wells emerge from under-trained tokens (Land & Bartolo, [19]), which merely provide a path of least resistance for the offset, and decoding-time corrections (Ding et al., [6]) leave the training-time cause in place. Appendix § 11 gives more detail.

Setup

Listwise and pairwise preference objectives. Our main runs train the model to rank a list of responses. Given a prompt xx and KK responses ranked from most to least preferred, y(1)≻y(2)≻⋯≻y(K)y^{(1)} \succ y^{(2)} \succ\cdots\succ y^{(K)}, the sequential Plackett–Luce objective (Plackett, 1975; Luce, [24]) asks the model to prefer the top remaining response over all responses below it, one stage at a time, by minimizing

LPL=−∑k=1K−1log⁡exp⁡(βsθ(x,y(k)))∑j=kKexp⁡(βsθ(x,y(j))),sθ(x,y)=1∣y∣∑t=1∣y∣log⁡πθ(yt∣x,y<t).\mathcal{L}_{\mathrm{PL}}=-\sum_{k=1}^{K-1}\log\frac{\exp\left(\beta s_{\theta}(x,y^{(k)})\right)}{\sum_{j=k}^{K}\exp\left(\beta s_{\theta}(x,y^{(j)})\right)}, \qquad s_{\theta}(x,y)=\frac{1}{\lvert y\rvert}\sum_{t=1}^{\lvert y\rvert}\log\pi_{\theta}(y_t\mid x,y_{<t}).

Each response is scored by the average per-token log-probability of the policy πθ\pi_{\theta}, without a reference model, and β>0\beta>0 sets the scale of the score differences. Up to this scale, the objective recovers the ranking term of PRO (Song et al., 2024), which also adds a supervised term on the top response. For K=2K=2 it becomes the length-normalized, reference-free pairwise objective of SimPO (Meng et al., 2024) without its target margin γ\gamma, which we call pairwise ranking.

We train on two preference datasets. The first comprises 5,383 driving-preference lists adapted from Li et al. [21], featuring four responses ranked by the severity of their physical consequences. The second is UltraFeedback (Cui et al., [5]), mostly used as pairs. As a boundary case we also train textbook DPO (Rafailov et al., 2023), which scores a whole response by its summed log-probability ratio against a frozen base model, sθ(x,y)=log⁡πθ(y∣x)−log⁡πref(y∣x)s_\theta(x,y)=\log\pi_\theta(y\mid x)-\log\pi_{\mathrm{ref}}(y\mid x). Of the many preference objectives, including IPO [11], ORPO [14], and KTO [8], we study per-token ranking, SimPO, and textbook DPO, whose demands span two orders of magnitude (§ 6). We primarily train LoRA adapters [15], optionally adding an NLL term λNLL⋅NLL(y(1))\lambda_{\mathrm{NLL}}\cdot\mathrm{NLL}(y^{(1)}) on the chosen response (Pang et al., 2024), and some runs on Llama-2, Llama-3.1, and the Llama-3-8B SFT checkpoint use full-parameter fine-tuning.

A nearly unconstrained translation. Every ranking objective above depends only on differences between scores. Adding the same amount to the scores of all responses of a prompt therefore leaves the loss unchanged. This invariance makes rewards under the Bradley–Terry [3] and Plackett–Luce models identifiable only up to a prompt-dependent constant (Rafailov et al., 2023). A shift of the hidden states shared by every position does not add exactly the same amount to every score, since its effect on each token depends on the token and its context. To first order, however, the objective sees such a shift only through its component along one direction, a weighted difference of the responses’ mean score gradients (Appendix § 12.17). Whether a learned shift lies mostly outside that direction is an empirical question, which § 5 answers by measurement.

The common-mode offset. Let hth_t and htbaseh_t^{\mathrm{base}} denote the final hidden states that the unembedding reads at position tt of a rejected response (evaluated under teacher forcing) for the trained and base models, respectively. We define the displacement as Δht=ht−htbase\Delta h_t=h_t-h_t^{\mathrm{base}}. The common-mode offset bˉ\bar{b} is the mean displacement over the response’s positions, bˉ=1T∑t=1TΔht\bar{b}=\frac{1}{T}\sum_{t=1}^{T}\Delta h_t, averaged again over a probe set of rejected responses. The residual rt=Δht−bˉr_t=\Delta h_t-\bar{b} represents the position-specific part, such that Δht=bˉ+rt\Delta h_t=\bar{b}+r_t holds exactly. This decomposition requires only two checkpoints and no access to training internals. We track the norm ∥bˉ∥\|\bar{b}\|, the cosine between the offsets of chosen and rejected responses, and the loudness κlog⁡∑vexp⁡((Wbˉ)v/κ)\kappa\log\sum_v\exp((W\bar{b})_v/\kappa), a smooth maximum of its projection onto the vocabulary, where WW is the unembedding matrix and κ\kappa a temperature.

Three stages of failure. Likelihood displacement is the routine erosion of absolute log-likelihoods for both chosen and rejected responses and, in isolation, is not pathological. Vocabulary well formation is the intermediate state, in which the probability mass released by the displacement concentrates on one or a few tokens or token families; it can be measured before any text is sampled. Generation collapse is the terminal state, in which these tokens dominate decoding. We count a run as collapsed once it reaches either state, that is, once one token takes most of the released mass or most of its samples degenerate, by repeating text, ending nearly empty, or cycling through the forms of one word. Released mass can also spread thinly across the vocabulary without forming a well. We call the resulting loss of sample quality diffuse degradation, measure it by the base model’s per-token negative log-likelihood of the samples, and do not count it as collapse.

Measurement and configuration. Mass flow is computed under teacher forcing on rejected responses. At every position we track the probability mass surrendered relative to the base model, the released mass, the share of it absorbed by the top-1 token, the capture, and the share leaking beyond the 10,000th-ranked vocabulary item, the tail; a well has nucleated once it captures the majority of the released mass. Degeneration scores sampled generations for nn-gram repetition [13] (Welleck et al. 2020), together with a word-stem ring rate and the base model’s per-token negative log-likelihood of the samples, which detects non-repeating degeneration that standard repetition counts overlook (Appendix § 12.14). Held-out ranking is evaluated on 600 unseen lists, or on 300 of them for a few runs, by top-1 accuracy, pairwise accuracy, and the margin between the chosen response’s mean log-probability and that of the highest-ranked rejected response (Appendix § 12.§ 2). Unless stated otherwise, we fine-tune a rank-32 LoRA adapter with a learning rate of 10−510^{-5} and an effective batch size of 32; full-parameter runs use a learning rate of 10−610^{-6}. A base budget is three passes over the training data, which is 504 steps on the driving lists and about three quarters of one epoch of the official SimPO training data on UltraFeedback. Extended runs operate at fixed multiples of this budget under a schedule of the same form, and a base model’s onset budget is the smallest tested multiple at which its untreated run collapses. Offsets and mass flow use 80 rejected training responses unless stated otherwise.

BaseBudgetΔ\Deltachosen (nat)Capture / tailDegen. / ring∥bˉ∥\lVert\bar{b}\rVertWell
Llama-2-7B1×\times−10.97-10.9789.4% / 89.5%96.0% / –92.4textt
Mistral-7B-v0.31×\times−17.51-17.5195.1% / 95.1%100% / 0%300.4*/******
Qwen2.5-7B3×\times−21.67-21.6793.9% / 86.7%99.3% / –328.7CJK fragment
Qwen2.5-14B3×\times−3.85-3.8527.0% / 7.7%80.3% / 78.7%198.6collision-fam.
Gemma-2-9B1×\times+0.62+0.6212.5% / 37.5%0% / 11 of 12†163.7crash-fam.
Llama-3.1-8B1×\times−14.27-14.2787.6% / 88.5%18.3% / 22.0%115.2.scalabtypied
OLMo-2-7B6×\times−2.90-2.9089.6% / 89.6%21.0% / 8.0%87.8contador
OLMo-3-7B3×\times−6.83-6.8393.8% / 93.8%96.0% / 0%79.7(quote+4×\timesnewline)
Llama-3.2-3B6×\times+1.17+1.1743.7% / 8.3%10.7% / 7.7%66.7none (ordinary tokens)

Table 1. Every base here but Llama-3.2-3B collapses, onto one token or the forms of one word. Untreated runs at their onset budget, or at 6×\times for Llama-3.2-3B. Ring counts samples dominated by one word stem, on the 12 saved samples where marked † and unmeasured where marked –.

Ranking Improves While Generation Collapses

We first train Llama-2-7B (Touvron et al., 2023b) with a LoRA adapter under sequential K=4K = 4 Plackett–Luce ranking on the driving lists. The lists were built for the Llama-2-based driving agent of Li et al. [21]. On 600 unseen lists, top-1 accuracy reaches 0.995, and pairwise accuracy stays above 0.94. Generation, however, fails. Within 504 steps, 89% of the probability mass released relative to the base model concentrates on a single token, textt, which the base model ranked deep in its vocabulary tail, and almost every sampled generation degenerates into repeating it. By every ranking metric the objective has succeeded, yet the output distribution has collapsed.

We replicate this training across 17 base-model lineages from five tokenizer families, spanning 0.5B to 14B parameters. Fourteen of the 17 collapse within six times the base budget (Table 1; Appendix Table 4), including TinyLlama-1.1B and the Qwen2.5 models from 0.5B to 3B. Twelve form a single-token well, and in Qwen2.5-14B and Gemma-2-9B most samples cycle through the forms of one word, collision or crash. Phi-3-mini and Gemma-2-2B, tested only up to three times the budget, stay below this criterion, although 39% of Phi-3-mini’s samples repeat speed. The clear exception is Llama-3.2-3B. Its rejected responses fall by over 30 nat per token, yet its released mass stays on ordinary tokens and its samples stay nearly as fluent as the base model’s (§6).

The collapse does not depend on adapters either. Full-parameter fine-tuning of Llama-2-7B at a learning rate of 10−610^{-6} collapses onto textt by step 250 of the same 504-step budget. Under full-parameter fine-tuning, onset tracks the learning rate summed over training, and from two earlier runs we wrote down before this run that it would collapse between steps 200 and 300 (Appendix § 12.10). The failure is therefore tied neither to one model lineage nor to adapters.

The Common-Mode Offset Causes the Well

The offset is shared across positions, prompts, and both responses. We computed the offset separately for each of 900 rejected responses and compared each with their global mean. Across prompts, the per-response offsets align with their mean with median cosines above 0.97 on each of the six runs we measured. Across the two sides of a preference pair, the chosen and rejected offsets exhibit cosines of 0.83 to 0.996. The shift is therefore a single geometric translation that moves both sides of every pair together (Figure 1b). Furthermore, this shared vector points at the well, since for single-token wells the well token has the largest entry of the projected vector Wˉb\bar{W}b (Figure 2b).

Decomposition of hidden-state shifts and well-ranking results

Figure 2. (a) At every position of a rejected response, the shift of the final hidden state splits into the shared offset bˉ\bar{b} and a position-specific residual rtr_t. (b) Through the unembedding WW, only bˉ\bar{b} singles out one vocabulary row. (c) Adding only bˉ\bar{b} to the base hidden state puts the well on top at nearly every rejected position, while adding only rtr_t rarely does; UF denotes UltraFeedback.

The loss barely sees the offset. We added the trained offset to the base model’s hidden states and recomputed the scores of chosen and rejected responses on held-out lists. In the four collapsing runs we evaluated, the offset lowers both scores by 15 to 65 nat per token but moves the margin between them by only 0.2 to 2.7 nat in an average list (Appendix Table 8). Its effect on both scores is 10 to 270 times its effect on their difference, and its cosine with the gradient of that difference in the base model is at most 0.16. The objective therefore observes a healthy margin, while the log-probabilities underneath fall by tens of nats. A zz-loss ([4] (Team OLMo et al. 2024)) that compresses the log-partition function 2.5-fold still leaves 65 to 66% of the released mass in the well (Appendix § 12.§ 8). Little in the loss therefore holds the offset back, and fine-tuning alone already builds one. A supervised control without the ranking term grows an offset of norm 56 shared by chosen and rejected responses (cosine 0.99), yet it lifts no token by more than 3.4 (Appendix § 12.11).

Keeping, hiding, adding, or removing the offset controls the well. We test the offset’s causal role in four ways. The first adds bˉ\bar{b} alone or rtr_t alone to the base hidden state. On untreated runs, bˉ\bar{b} alone makes the well the top token at nearly every rejected position, more often than the full trained state, while rtr_t alone rarely does (Figure 2c). The residual therefore partly works against the well. The same holds under full-parameter fine-tuning, where the fine-tuned unembedding rows alone almost never lift the well (Appendix § 12.10).

The second test hides the offset from the loss. Centering computes the rejected-side ranking loss on ht−stopgrad⁡(bˉ)h_t-\operatorname{stopgrad}(\bar{b}). The rejected scores then no longer change when the offset grows, so pointing bˉ\bar{b} at one vocabulary row no longer lowers them. The gradient can still move the offset, since it treats the subtracted mean as a constant. On the Llama-2 flagship, centering keeps most of the offset’s norm but turns its projection away from the vocabulary, and no well forms.

The third test transplants the trained offset into an untreated base model. This injection adds bˉ\bar{b} to the final hidden state at every position, causing the well token to dominate generation on every held-out prompt. Stripping only the well’s row component from bˉ\bar{b} removes that well, but generation instead collapses onto a different single token, whereas random or orthogonal directions of the same norm produce no well. The well therefore follows the offset rather than any particular token. The fourth test removes the offset from a model that has already collapsed. Subtracting half of its logit bias WbˉW\bar{b} from the untreated flagship model at decoding time lowers the share of degenerate greedy samples on 60 held-out prompts from 100% to 0%, and the chosen response regains 8.7 nat per token (Appendix § 12.§ 7). Keeping only the offset reconstructs the well, hiding it prevents the well, adding it creates one, and removing it stops the collapse, making the offset a controllable cause of collapse.

When the Offset Becomes a Well

A well forms only after the offset’s gain passes the base gap. The formation of a well depends on a logit ratio. Let mtm_t be the base gap, which measures how much the base model prefers its own top token over the candidate well token at position tt, and let gtg_t be the logit advantage that bˉ\bar{b} adds to that well token. In the offset-only state, the well overtakes the base model’s top token where the dose Dt=gt/mtD_t=g_t/m_t exceeds one (Figure 3a). Scaling the injected offset confirms this threshold, since a quarter-strength bˉ\bar{b} fails to trigger the well while a half-strength one pushes DtD_t above one and puts the well on top at nearly all rejected positions. We also tracked the dose along natural training runs (Figure 3b; Appendix Figure 9 shows each lineage). Across all 510 snapshots, the well never tops a majority of rejected positions while the median dose stays below one, so in our untreated runs a well needs a dose above one, but a dose above one does not guarantee a well. In Llama-3.2-3B the median dose reaches 2.75 on the ordinary subword. Yet the trained model puts one on top at only 29% of rejected positions, against 80 and 83% for the wells of Llama-3.1 and Llama-2, and its samples stay nearly as fluent as the base model’s.

Schematic of the offset threshold, well prevalence across collapsing lineages, and median doses under different budgets

Figure 3. (a) The offset overrides the base model at a position once its gain gtg_t on the would-be well exceeds the base gap mtm_t, that is, once the dose DtD_t exceeds one. (b) Across 510 snapshots of ten collapsing lineages, colored by training progress, and three summed-score runs that stay healthy, in gray, the well never tops a majority of rejected positions while the median dose stays below one. (c) Median doses of Table 2 at the usual and the three-fold budget.

Wells tend to land on anomalous vocabulary rows. Because the same bˉ\bar{b} projects more strongly onto rows with large norms, rare tokens with large unembedding rows gain the most. In the adapter runs on the driving data, most single-token wells sit in the top 1% of their base model’s row norms and never occur in WikiText-103 (Merity et al., 2017), whereas in Llama-3.2-3B the highest-norm rows belong to common subwords. High row norms are not required, however. On UltraFeedback Llama-2 collapses onto the ordinary token Conf, which the offset alone still puts on top at nearly every position (Figure 2c). Row geometry makes some tokens more exposed, but in untreated runs the offset is what lifts them.

Different losses push the offset by very different amounts. We trained Llama-2-7B on the UltraFeedback pairs with our per-token ranking loss, the same loss with a reference model, SimPO, and textbook DPO, each at its usual setting and, except the plain per-token loss, at three times the budget. We also ran SimPO with full-parameter fine-tuning on Llama-2-7B and Llama-3.1-8B. Table 2 and Appendix Figure 11 report each run’s demand, offset norm, dose, capture, share of degenerate samples, and win rate over the starting model under Skywork-Reward-V2 (Liu et al., [23]).

ObjectiveBudgetDemand (nat)∥bˉ∥\lVert\bar{b}\rVertDoseCaptureDegen.Win
Per-token ranking, β=0.12\beta= 0.121×1\times42841.9297%99%25%
with a reference model1×1\times42791.9186%99%26%
with a reference model3×3\times42914.53†^{\dagger}89%99%17%
SimPO1×1\times2.41140.290.4%0.7%7%
SimPO3×3\times2.4820.539%28%6%
SimPO, full-parameter1×1\times2.4650.333%0.7%10%
SimPO, full-parameter3×3\times2.4851.2927%80%7%
on Llama-3.1-8B1×1\times2.4750.520.3%30%13%
on Llama-3.1-8B3×3\times2.41011.30†^{\dagger}45%86%3%
from SFT, on Llama-3-8B1×1\times3.0330.152.5%1.3%63%
from SFT, on Llama-3-8B3×3\times3.0770.192.7%0%34%
Textbook DPO, β=0.1\beta= 0.11×1\times0.3140.084%3%68%
Textbook DPO, β=0.1\beta= 0.13×3\times0.3440.124%1%66%

Table 2.

Only the per-token ranking loss collapses at the usual budget, and adding a reference model does not stop it. SimPO forms no well at its usual setting, although its samples still degrade, and it passes the threshold with three times the budget under full-parameter fine-tuning on both Llama-2 and Llama-3.1 (Figure 3c). Textbook DPO stays far below the threshold even with three times the budget, because its offset stays short and spreads across the vocabulary.

This difference stems from how far each loss keeps pushing. A ranking loss stops pushing once the chosen response beats the rejected one by a large enough margin, and we call the per-token size of that margin the loss’s demand.2 Our per-token loss keeps pushing until the gap reaches about 42 nat per token, well beyond the base model’s own preference for its top token, so the dose passes one. SimPO stops at about 2.4 nat. Textbook DPO sums the score over the whole response, which spreads the same margin across hundreds of tokens and stops it at about 0.3 nat per token.

The reference model is therefore not what protects textbook DPO. Adding one to the per-token loss leaves a well in place on both datasets and at both budgets, with an offset as long as the untreated run’s. What protects textbook DPO from the well is its low demand. Summing the score is roughly equivalent to averaging it with β\beta multiplied by the response length, so at the usual β\beta the protection comes from the resulting low demand.

Table 2: Each loss at its usual setting and, where listed, with three times the budget, with doses above one in bold. Runs start from the base model unless marked SFT, and all train on the same UltraFeedback pairs. SimPO uses β=2.5\beta= 2.5 and γ=1.0\gamma= 1.0 from base models and the official β=2.0\beta= 2.0 and γ=1.0\gamma= 1.0 from the SFT checkpoint. Degen. counts repetitive or nearly empty samples, and Win compares a run’s samples with those of the model it starts from.

†^{\dagger}Computed for the run’s own leading token, D or Apache; all other doses are for Conf.

Between these extremes, SimPO within its published range grows an offset about as long as the collapsing run’s, but a quiet one whose dose remains below one, so no well forms. Nevertheless, its generation quality degrades. The released mass spreads deep into the vocabulary instead of settling on one token, and the samples devolve into strings of unrelated words that standard repetition counts fail to detect, while the reward model prefers the base model’s sample on about nine in ten prompts. However, adding the SimPO offset alone to the base model does not reproduce this diffuse form (Appendix § 12.13). This diffuse degradation is therefore a second route to damage, separate from the well. The missing fine-tuning stage alone does not explain this damage, since textbook DPO from the same base model wins against it on about two thirds of prompts.

Under a three-fold budget, however, the offset grows louder, concentrating its projection onto a narrow set of vocabulary rows. Its dose keeps rising, and under full-parameter fine-tuning it passes one on both Llama-2 and Llama-3.1. By the end of training, 80 to 86% of samples degenerate, and held-out ranking is the best of any UltraFeedback run on that base.

We also trained SimPO from its authors’ public SFT checkpoint of Llama-3-8B with their hyperparameters (Appendix § 12.12). From this start, no well forms within three times the base budget, and the median dose of every token stays below 0.4. The offset nonetheless grows from a norm of 33 at the base budget to 77 at three times the budget. It stays quiet, lifting no token by more than 4.3 nat, against about 18 nat for SimPO from the base models at the same budget. Meanwhile the SFT model’s per-token negative log-likelihood of the samples rises from 1.35 to 3.86. Two public reward models agree, preferring the one-fold run’s samples to the SFT model’s on 63 to 65% of held-out prompts but the three-fold run’s on only 33 to 34%. Supervised fine-tuning enlarges the base gaps of the candidate wells by at most 22%, so by the dose it should delay the well rather than prevent it.

Supervised fine-tuning already produces an offset of the same form, and the preference loss sets where it points and how far it grows (Appendix § 12.11). Small β\beta and listwise objectives therefore do not invent a new failure mode; they push harder on a shift that fine-tuning already produces.

Controlling the Offset

Centering hides the offset from the loss but leaves most of its norm within the model. Our recipe therefore keeps centering and adds two penalty terms that act on the offset itself, used alone or together, and we call the combination both. A norm penalty λnorm∥b∥\lambda_{\mathrm{norm}}\lVert b\rVert shrinks the offset, and a loudness penalty λloud⋅κlog⁡∑vexp⁡((Wbˉ)v/κ)\lambda_{\mathrm{loud}}\cdot\kappa\log\sum_v\exp((W\bar{b})_v/\kappa) flattens its vocabulary projection without necessarily shrinking its norm. On the driving lists the recipe also keeps the NLL term of weight 1.0, so it differs

of ∣y∣|y| tokens, with ∣y∣=182|y| = 182 on average in our UltraFeedback pairs. SimPO’s target margin γ\gamma adds γ/β\gamma/\beta to its demand. Appendix Figure 10 shows the full sweep over β\beta. from the NLL-only baseline only by offset control. Offset control matters most for objectives worth keeping despite a high demand. Listwise ranking, for example, also supervises the order among the rejected responses, which pairwise objectives never see, and on held-out driving lists it keeps that order at 0.82 to 0.92 while pairwise ranking inverts it (Appendix § 12.§ 6).

Cross-family validation up to three times the base budget. We applied the recipe to four base models at or beyond the budget where each untreated run collapses, with Llama-2, Qwen2.5-7B, and Llama-3.1 at three times the base budget (Table 3). No run forms a well. The largest share of the released mass, at most 20%, goes to an ordinary token such as the period, the tail stays at or below 5.5%, and sampled degeneration remains at or below about 12%. By contrast, at the same three-fold budget on Llama-2, the NLL-only baseline, N-DPO, and DPO-Positive leave 71 to 81% of the released mass in a single well (Table 3). The cost to ranking performance is small, since held-out pairwise accuracy settles between 0.94 and 0.96. On Llama-2, Mistral, and Qwen2.5-7B, this is within 2 points of a run of the same base and budget without offset control. Controlling the offset is therefore enough to prevent collapse up to three times the base budget, at almost no cost to ranking. At six times the budget on OLMo, the residual, not bˉ\bar{b}, lifts the surviving well, and a residual penalty closes the remaining routes on OLMo-3 (Appendix § 12.§ 4).

BasePenaltyStepsCapture / tailDegen.Held-out top-1 / pairwise
Llama-2-7Bloudness151219.8% / 1.5%6–12%1.00 / 0.95
Mistral-7Bboth50414.6% / 1.4%3.0%1.00 / 0.94
Qwen2.5-7Bboth15128.0% / 0.2%1.7%1.00 / 0.96
Llama-3.1-8Bboth151213.0% / 5.5%6.7%1.00 / 0.96
Llama-2-7B without offset control
NLL only–151270.6% / 71.0%2.0%1.00 / 0.97
N-DPO–151278.1% / 78.4%18.0%1.00 / 0.96
DPO-Positive–151281.4% / 82.2%24.3%1.00 / 0.96

Table 3. The recipe, centering plus a norm penalty, a loudness penalty, or both, at or beyond each base’s onset budget, where capture goes to an ordinary token. Three Llama-2 baselines without offset control keep 71 to 81% of the released mass on one rare token.

What Ranking Metrics Miss

An external judge disagrees with the ranking metrics. Held-out pairwise accuracy levels off at 0.93 to 0.97 within 300 to 500 steps, treated or not, while the margin keeps growing. The untreated run with the largest held-out margin typically collapses, in line with the over-optimization of direct alignment algorithms (Rafailov et al., 2024) (Appendix Figure 13). We evaluated one sample for each of 300 held-out UltraFeedback prompts using two public reward models, Skywork-Reward-V2 (Liu et al., [23]) and ArmoRM (Wang et al., 2024), across 17 Llama-2 runs (Appendix Table 7 and Figure 12a). Across these runs, the better a run’s held-out pairwise accuracy, the less often the external judge prefers its sample to the base model’s, with a Spearman correlation of −0.78 for Skywork-Reward-V2 and −0.79 for ArmoRM. The runs that rank best move the most probability away from the base model’s likely tokens, and the judge tracks how much stays there. The share of mass remaining on the base model’s ten most likely tokens correlates with the win rate at +0.74 and +0.87. Since the base models of these runs are not instruction-tuned, we use the win rate only to compare runs. By this measure, the recipe with an NLL term of weight 1.0 matches textbook DPO, winning 65% of prompts against 66 to 68%, as does the NLL-only baseline at 64%.

Two-panel scatter plot relating held-out ranking accuracy, judge preference, and training steps

Figure 12. (a) Across 17 UltraFeedback runs, the better a run ranks held-out responses, the less often a public reward model prefers its samples to the base model’s; the glyphs show where each group’s released mass goes. (b) In seven of nine collapsing lineages, the eventual well enters the eight largest entries of W‾b\overline{W}_b before it takes over the rejected positions.

The offset warns before the well forms. In seven of the nine lineages with snapshots every 50 steps, the eventual well enters the eight largest entries of WbˉW\bar{b} before it takes over, up to 1,200 steps ahead, even while held-out ranking has long saturated (Appendix Figure 12b). Tracking ∥bˉ∥\lVert\bar{b}\rVert, its loudness, and the leading entries of WbˉW\bar{b} during training therefore offers an early signal that standard ranking metrics do not give, although we have read it only from completed runs.

Conclusion

Preference optimization can collapse generation while held-out ranking stays high. We traced this failure to a common-mode offset, a shift of the final hidden state shared across positions and across chosen and rejected responses, which the ranking objective barely penalizes. A well forms only after the offset lifts one token above the tokens the base model would otherwise predict. Different ranking losses turn the same preference pressure into very different doses. Hiding the offset from the loss and penalizing it prevents collapse up to three times the base budget. Across runs, held-out ranking accuracy even anticorrelates with an external judge (ρ=−0.78\rho= -0.78). Ranking accuracy and preference margin measure only what the loss constrains, so the offset deserves monitoring.

Limitations. Our evidence spans 17 base-model lineages, two datasets, and adapter and full-parameter training, with two further seeds for the flagship settings and one seed elsewhere (Appendix § 12.16). The penalty strengths are calibrated on the driving data, the six-fold result covers the OLMo family only, and apart from the SFT-start runs the demand analysis starts from base models.

Reproducibility Statement

Training configurations (base models, LoRA rank, learning rate, batch composition, and the form of every penalty term), the exposure-ordering and offset-decomposition measurement code, and the held-out ranking probe are described in § 3–§ 7 and Appendix § 12; code with all penalty coefficients and the training and held-out preference data will be released upon publication. Held-out ranking numbers are computed on 600 lists, or on 300 of them where stated, whose prompts and chosen responses do not appear in any training split; the corpus counts in § 6 use the full training text and the WikiText-103 training split.

AI Use Statement

The authors set the research questions, made every final design decision, and checked every reported number against the logged experimental runs and every citation against its source. Within this process, we used generative AI tools (Claude Code and Gemini) to help develop the conceptual framework, propose and refine hypotheses, design experiments and give feedback on the methodology, formulate mathematical claims and assist in writing their proofs, implement the methods, clean and reformat datasets, interpret results, and assist with translation. We did not use these tools to generate synthetic datasets. We have reviewed all AI-assisted work and take responsibility for the final content of this work, including text, claims or artifacts produced with the aid of generative AI.

Acknowledgments

This work was supported by JSPS KAKENHI Grant Number JP26KJ0799 and by TIER IV, Inc.

Appendix Overview

Appendix A extends the related work of § 2. Appendix B collects the setup details, full results, and supporting measurements behind the main text, and ends with the formal statement of the common-shift invariance. The third column names the main-text section that each part supports.

PartContentsSupportsPage
AExtended related work. Iterated preference optimization, likelihood displacement, likelihood regularizers, and representation geometry.§ 216
B.1Full cross-family onset table. All 17 lineages, the budget at which each collapses if it does, and the collapse criterion.§ 417
B.2Setup and measurement details. β\beta per scene type, exposure, and the full definitions of every metric.§ 317
B.3Training curves and extended-budget trajectories. Flagship Llama-2 runs and six-fold-budget OLMo runs.§ 7, § 818
B.4Routes of the released mass at extended budget. Where the mass goes under each penalty, and the residual penalty.§ 718
B.5Penalty trajectories across families. Offset norm and its largest vocabulary entry under each penalty.§ 719
B.6A listwise advantage orthogonal to collapse. Ordering of unseen rejected responses by objective.§ 719
B.7Removing the offset from a collapsed model. Subtracting its logit bias at decoding time undoes the collapse.§ 519
B.8Absolute logit scale. A zz-loss compresses log ZZ without removing the well.§ 520
B.9Dose along training and the objective’s demand. Dose per lineage, the sweep over β\beta, and standard-recipe offsets.§ 622
B.10Full-parameter onset. Onset tracks the integrated learning rate.§ 4, § 523
B.11A supervised fine-tuning control. Plain fine-tuning builds a shared offset but no well.§ 5, § 623
B.12SimPO from a supervised fine-tuning checkpoint. The official recipe from a public SFT model, one-fold and three-fold budgets.§ 623
B.13External reward models. Judge protocol, win rates of all runs, the early-signal panel, and ranking against the margin.§ 6, § 825
B.14Shapes of well that standard rulers miss. Latent wells, non-repeating wells, and word-family rings.§ 3, § 626
B.15Supplementary measurements. Further numbers behind the claims of the main text.§ 5–§ 826
B.16Seeds and scope. Seed repeats, adapter ranks, and the ranges of the demand analysis.§ 927
B.17Common-shift invariance and the trained offset. Two propositions and the measured split of the offset.§ 3, § 527

Table 1 (PDF p. 16).

Extended Related Work

Iterated preference optimization. Recent pipelines repeat preference optimization over several rounds on continually refreshed preference data [12] (Yuan et al. 2024), although sampling more responses per prompt does not always help (Xiao et al., 2025).

Likelihood displacement, squeezing, and gradient entanglement. Recent studies have documented the probabilistic symptoms of preference optimization, but from different angles. Razin et al. (2025) demonstrate that DPO training can decrease the chosen response’s absolute likelihood. They trace this to the unembedding matrix’s geometry, using a centered hidden-embedding similarity score to predict when a dataset risks spilling rejected updates onto chosen ones. Yuan et al. (2025) prove that margin-based losses entangle the gradients of chosen and rejected log-probabilities, explaining why the two tend to move in tandem. While these works map the symptoms in probability space, we isolate in representation space the common-mode offset of the final hidden state, which produces the well. Furthermore, Ren & Sutherland (2025) show that the negative gradient on rejected tokens “squeezes” probability mass toward tokens the base model is already confident about. Pushing down the probability of unwanted tokens is also how unlikelihood training reduces repetition (Welleck et al., 2020), whereas in preference optimization the push on rejected responses can end in repetition. The vocabulary wells we discover represent a different phenomenon, since the base model initially ranks our well tokens far below its preferred tokens, typically by 17 to 39 nat. It is the accumulation of the common-mode offset that inflates the logits of these low-confidence tokens, until one of them overcomes this gap and dominates decoding at most positions.

Regularizing absolute likelihood. Several methods anchor the absolute probabilities that a relative ranking loss leaves free. DPO-Positive (Pal et al., 2024) penalizes the chosen log-probability if it falls below its reference value; Pang et al. (2024) inject an explicit NLL term for reasoning tasks; and Im et al. [16] regularize the length-normalized log-probability of the full response set toward the reference. However, because these interventions constrain the scalar probabilities rather than the underlying vector representations, they do not directly constrain the shared direction in the hidden state. In our experiments, an NLL term on the chosen response delays the growth of the common-mode offset without eliminating it (Figure 5). When it does prevent immediate collapse, the vocabulary well relocates to positions after the desired answer, where the NLL term exerts no constraint.

Six-fold-budget training trajectories of OLMo-2-7B and OLMo-3-7B under six training conditions

Figure 5. Six-fold-budget training trajectories of OLMo-2-7B (a–c) and OLMo-3-7B (d–f) in six conditions, namely untreated, NLL only, and four offset-control variants that penalize loudness, loudness and norm, per-position loudness, or per-position loudness and norm. OLMo-2 has no NLL-only run. (a) and (d) Well capture crosses the dotted 50% line in the untreated and NLL-only runs and ends at 81 to 94%, while no offset-control variant exceeds 45%. (b) and (e) The tail beyond rank 10,000 rises with the capture in those runs. With offset control it ends at 28 to 37% on OLMo-2, but near 5% on OLMo-3 once the norm penalty is added. (c) and (f) Pairwise accuracy on 300 held-out lists stays between 0.93 and 0.96 in every run.

Unembedding geometry and representation anisotropy. Prior work has mapped the geometry of pretrained representations. Word embeddings and contextual hidden states are anisotropic, sharing a common mean and a few dominant directions ([7, 2] (Mu & Viswanath 2018)), and standard training often squeezes them into a narrow cone ([9]). Furthermore, specific outlier dimensions are known to track token frequency ([25] (Puccetti et al. 2022)), encode massive activations (Sun et al., 2024), or regulate output entropy (Stolfo et al., 2024). However, these works describe the geometry of a pretrained model. The common-mode offset we study is constructed during preference training. It does not reshape the entire embedding space into a cone; instead, it acts as a single, shared displacement vector added to the hidden states, which increases the inner product with one specific vocabulary row. Functionally, this resembles activation steering, where adding a fixed vector at every position predictably alters a model’s behavior (Rimsky et al., 2024). Mechanistic studies of DPO also find that it changes behavior by shifting the residual stream, bypassing rather than removing capabilities the model already has (Lee et al., [20]). The key distinction is that preference optimization grows this steering vector on its own in the background, and the behavior it ultimately produces is generation collapse. Finally, Land & Bartolo [19] identify “under-trained” tokens with near-zero frequencies in base models. While many of the severe vocabulary wells in our experiments happen to emerge from this vulnerable population, these fragile tokens do not spontaneously cause collapse; they merely provide a path of least resistance for the common-mode offset. Decoding-time corrections that subtract unconditional logit terms (Ding et al., [6]) act after the fact and leave the training-time cause in place.

Additional Details

Full Cross-Family Onset Table

Table 4 reports every base-model lineage with at least one untreated run.

Setup and Measurement Details

On the driving lists, β\beta is fixed per scene type, 0.12 for the two thirds of lists in normal driving and 0.18 to 0.40 for the other scene types; on UltraFeedback, β=0.12\beta= 0.12. SimPO’s target margin γ\gamma is subtracted from the top response’s scaled score at each stage. Because a list of length KK yields K−1K - 1 comparisons, the response ranked kk-th is rejected in k−1k - 1 of them, a quantity we call its exposure. In the flagship run the rejected responses fall in order of their exposure, so the more often a response serves as a negative baseline, the steeper its drop.

We evaluate each trained run along four axes, since no single metric captures the whole failure. Mass flow is defined in § 3. Degeneration combines standard metrics for nn-gram repetition (Holtzman et al., [13]; Welleck et al. (2020)), non-ASCII runs, and abnormal length with two measures of our own. A word-stem ring rate catches token-family wells, which we call word-family rings, and the base model’s per-token negative log-likelihood of the samples catches degeneration that does not repeat. Held-out ranking uses 600 unseen lists of a prompt and its responses, or 300 of them for the Mistral runs, the six-fold OLMo-2 runs, and the training trajectories. Top-1 accuracy is the fraction of lists whose chosen response scores highest, and pairwise accuracy is the fraction of response pairs within a list that the model orders correctly. Kendall’s τ\tau (Kendall, [18]) is computed over unseen rejected responses, and the margin is the chosen response’s mean log-probability minus that of the highest-ranked rejected response. Offset geometry covers the norm ∥bˉ∥\lVert\bar{b}\rVert, the loudness, and, for candidate well tokens, their unembedding row-norm percentile and their counts in the training text and in the 10810^{8}-token WikiText-103 corpus ((Merity et al., 2017)). For centering, bˉ\bar{b} is computed on the current batch against the adapter-free forward pass. Collapse and ranking dynamics are measured across all lineages, while the computationally intensive analyses of §5 and §6 use the representative subsets named there.

BaseVocabularyOnset multipleCaptureDegen. / ringCollapse
Llama-2-7B ((Touvron et al., 2023b))32k1×89.4%96.0% / –well
LLaMA-1-7B ((Touvron et al., 2023a))32k1×53.0%86.3% / –well
TinyLlama-1.1B ((Zhang et al., 2024))32k6×99.0%100% / 100%well
Llama-3.2-1B-Instruct ([12])128k1×88.8%61.0% / –well
Llama-3.2-3B128knone within 6×43.7%10.7% / 7.7%no
Llama-3.1-8B128k1×87.6%18.3% / 22.0%well
Qwen2.5-0.5B ([10])152k6×93.9%7.7% / 6.7%well
Qwen2.5-1.5B152k6×93.8%99.7% / 1.3%well
Qwen2.5-3B152k6×89.6%91.3% / 0.3%well
Qwen2.5-7B152k3×93.9%99.3% / –well
Qwen2.5-14B152k3×27.0%80.3% / 78.7%word ring
Mistral-7B-v0.3 ([17])32k1×95.1%100% / 0%well
OLMo-2-7B ((Team OLMo et al., 2024))100k6×89.6%21.0% / 8.0%well
OLMo-3-7B ((Team Olmo et al., 2025))100k3×93.8%96.0% / 0%well
Gemma-2-2B ([10])256knone within 3×0.4%4.7% / 4.3%no
Gemma-2-9B256k1×12.5%0% / 11 of 12†word ring
Phi-3-mini ([1])32knone within 3×18.4%15.7% / 38.7%no

Table 4. Fourteen of the 17 base-model lineages collapse within six times the base budget (§ 4). A run collapses once one token takes most of the released mass, a well, or once most of its samples degenerate or ring on one word stem, a word ring. Each row shows the untreated run at the onset budget, or at the largest budget tested when no collapse occurs. The onset multiple is relative to the 504-step base budget under the fixed schedule. TinyLlama, the three smallest Qwen2.5 models, and Llama-3.2-3B were trained at one and six times the budget only, Qwen2.5-14B first at three times, Phi-3-mini and Gemma-2-2B at one and three times, and Gemma-2-9B at the base budget only. Ring rates marked – come from audits that predate the ring measure, and † marks a count on the 12 saved samples. Each family is cited at its first row.

Training Curves and Extended-Budget Trajectories

The runs without offset control drive the ranking loss lowest and the training margin highest, and almost all of their extra margin comes from pushing the rejected responses down (Figure 4). Figure 5 shows the six-fold-budget OLMo trajectories behind §7, including the per-position loudness penalty runs.

Training curves of six Llama-2-7B runs

Figure 4. Training curves of six Llama-2-7B runs with K=4K=4 sequential PL on the driving lists at three times the base budget, smoothed with a 25-step running median. (a) Total loss and (b) ranking loss fall lowest in the three runs without offset control, NLL only, N-DPO, and DPO-Positive (DPOP), which end with 71 to 81% of the released mass in a single well. The ranking loss of the SFT control (β=0\beta=0) stays at its initial value. (c) Without offset control the training margin reaches 67 to 87 nat, against 30 and 37 nat with it, and (d) almost all of this difference comes from pushing the rejected responses down.

Routes of the Released Mass at Extended Budget

At six times the base budget, each penalty on the OLMo family closes one route for the released mass, and the mass takes the next one (Figure 5). Penalizing loudness alone moves the well to another token of the same token family. Adding the norm penalty removes the well but sends about half of the released mass to punctuation and premature end-of-text tokens. Applying the loudness penalty at every position, rather than only to the mean offset, does not close this route either. Combined with the norm penalty, it still leaves about 40% of the mass on punctuation and end-of-text on OLMo-3, and without the norm penalty it turns most OLMo-2 samples into word-family rings. In the treated runs that still leak, the base state plus rˉ\bar{r} never makes the well the top token, while the base state plus rtr_t does so at 11 to 23% of positions, so the residual has become the component that lifts the surviving well. Penalizing the residual itself, with λres1T∑t∥rt∥\lambda_{\mathrm{res}}\frac{1}{T}\sum_t \lVert r_t\rVert, closes both routes on OLMo-3 and leaves no well, no tail, and unchanged held-out ranking. On OLMo-2, however, the mass moves on into word-family rings and the deep tail.

Penalty Trajectories Across Families

Figure 6 shows how each penalty shapes the offset over training on Llama-2 and Qwen. The norm penalty keeps the offset short, while the loudness penalty lets it grow but keeps its largest vocabulary entry small. On Qwen, the loudness penalty alone lets the offset grow to 490 to 550, more than ten times its length under the norm penalty, which is consistent with the family-specific penalty choice in Table 3.

Six line plots showing the effects of penalties on Llama-2-7B and Qwen2.5-7B runs

Figure 6. (a) Across seven Llama-2-7B runs, ∥bˉ∥\lVert\bar{b}\rVert stays between 70 and 85 without a penalty, grows to 116 under the loudness penalty, and falls to 6 to 8 once the norm penalty is added. (b) Divided by ∥bˉ∥\lVert\bar{b}\rVert, the largest entry of WbˉW\bar{b} stays near zero only under the loudness penalty. (c) NLL only and N-DPO lift the largest entry of WbˉW\bar{b} to 76 and 95 by the end of training, while the loudness penalty holds it below 4. (d) Across seven Qwen2.5-7B runs, ∥bˉ∥\lVert\bar{b}\rVert grows to 490 to 550 under every loudness penalty but stays near 40 once the norm penalty is added. (e) Divided by ∥bˉ∥\lVert\bar{b}\rVert, the largest entry of WbˉW\bar{b} falls below zero under every loudness penalty. (f) Without a penalty, the largest entry of WbˉW\bar{b} climbs to 66, while every penalized run stays below 5.

A Listwise Advantage Orthogonal to Collapse

Held-out evaluation reveals one respect in which listwise and pairwise objectives differ, independent of collapse. Beyond whether the single best response is identified, we ask whether a model preserves the graded ordering among the unseen rejected responses of a held-out list. Pairwise training never supervises this ordering, since it only compares the top response against one alternative. On the driving lists, pairwise ranking (K=2K=2) inverts the graded ordering, with ordering accuracy of 0.17 to 0.27 and Kendall τ\tau of −0.46-0.46 to −0.66-0.66, while every sequential listwise K=4K=4 run preserves it, at 0.82 to 0.92 (Figure 7). All-pairs and K=3K=3 training fall in between at about 0.58, and SFT alone falls slightly below chance, at 0.415. On UltraFeedback the effect does not appear, and pairwise training instead slightly improves the ordering of rejected responses over the base model. This listwise benefit is therefore domain-dependent and orthogonal to the collapse mechanism of § 5. It does not correlate with whether a run nucleates a well, and offset-controlled runs keep whichever ordering their underlying objective produces.

Ordering accuracy and Kendall tau for held-out lists

Figure 7. (a) On 600 held-out lists, pairwise K=2K=2 training orders the unseen rejected responses below chance (dashed line), with or without offset control (blue), while every sequential listwise K=4K=4 run orders them better than the base models. (b) The Kendall τ\tau over the rejected responses alone is negative for SFT and every pairwise run, and at least 0.64 for every listwise K=4K=4 run. Triangles mark runs that form a well.

Removing the Offset from a Collapsed Model

We subtract a multiple of the logit bias WbˉW\bar{b} from the untreated flagship model at decoding time, with bˉ\bar{b} measured on 80 rejected responses, and decode 60 held-out prompts greedily for 100 tokens.

Subtracting bˉ\bar{b} from the final hidden state gives the same result, since the unembedding is linear. Removing half of the bias stops the collapse, and removing all of it works nearly as well (Table 5). Removing only the well’s own logit is not enough, because the collapse then moves to the next token of the same direction, a rare Cyrillic token. The recovered model continues text normally but does not produce the answer format, which the untreated run never learned, so removing the offset stops the damage without aligning the model.

SubtractedDegen.Chosen log-prob (nat)Rejected log-prob (nat)
Nothing100%−13.37-13.37−65.01-65.01
Only the well’s logit22%−8.73-8.73−54.64-54.64
The eight largest entries of WbˉW\bar{b}65%−7.17-7.17−48.76-48.76
Half of WbˉW\bar{b}0%−4.70-4.70−39.44-39.44
All of WbˉW\bar{b}7%−5.34-5.34−24.98-24.98

Table 5. Subtracting the offset’s logit bias from the collapsed flagship model at decoding time. Degen. is the share of degenerate greedy samples, and the log-probabilities are per-token means under teacher forcing with the same subtraction.

Absolute Logit Scale

One alternative hypothesis is that collapse is a symptom of the output distribution’s absolute scale drifting upward, since a softmax is unchanged when the same constant is added to every logit. We test it by penalizing log⁡Z=log⁡∑vexp⁡(zv)\log Z = \log\sum_v \exp(z_v), a zz-loss as in Chowdhery et al. [4], at increasing strength on the Llama-2 flagship setting (Figure 8). The strongest coefficient compresses log⁡Z\log Z 2.5-fold, shrinks the relative peak logit and the common-mode vocabulary projection, and lowers the margin by about one third relative to the untreated run. Held-out ranking accuracy is essentially unchanged, and well capture stays at 65 to 66% of the released mass, against 74% untreated. At this coefficient the zz-loss gradient reaches 19 to 43% of the ranking-gradient norm, yet collapse persists. We treat this as supporting rather than conclusive evidence, because a raw gradient-norm

Four plots showing the effects of a z-loss penalty on log Z, relative peak logit, largest entry of $\overline{W}_b$, and released mass

Figure 8. A zz-loss penalty on log⁡Z\log Z compresses absolute logit scale, relative peak, and common-mode vocabulary projection without preventing well capture. Blue lines show three strengths of λz\lambda_z on the Llama-2 flagship setting, darker for stronger, and the gray line shows the untreated run. (a) The strongest penalty, λz=10−4\lambda_z = 10^{-4}, lowers log⁡Z\log Z from 76.0 to 30.5 by step 1500. (b) It lowers the relative peak logit by the same factor of 2.5, from 70.0 to 28.2. (c) It lowers the largest entry of W‾b\overline{W}_b from 74.6 to 31.1. (d) The top token still takes 65 to 66% of the released mass at every strength, against 74% untreated, and at λz=10−4\lambda_z = 10^{-4} the mass beyond rank 10,000 (dotted) rises to 80%.

ratio is not the same as its realized contribution to the optimizer update. Together with § 5, the result indicates that the operative quantity is not the overall output scale but the offset’s projection onto the vocabulary, which a zz-loss constrains only indirectly and incompletely.

Dose Along Training and the Objective's Demand

We summarize each snapshot by its median dose, the median gain divided by the median base gap over the rejected positions. In all nine lineages traced every 50 steps, the median dose passes one before the well takes over, and the takeover itself takes 100 to 300 steps (Figure 9). Across the sweep over β\beta, the final dose passes one only under per-token averaging with a demand above the base gap (Figure 10). The standard recipes build the same kind of offset as the collapsing runs, at a smaller dose (Figure 11).

Dose along natural training runs and the share of rejected positions with the well on top

Figure 9. Dose along natural training runs, with a dot at the first snapshot at which the well is the top token at a majority of rejected positions. (a) In all nine collapsing lineages, the median dose Dt=gt/mtD_t = g_t/m_t, measured every 50 steps, passes one before the dot and lies between 1.1 and 4.1 at the dot, while the three summed-score runs that stay healthy, in gray, remain below 0.2. (b) The share of rejected positions with the well on top climbs from under 10% to a majority within 100 steps in seven lineages, and within 200 and 300 steps in Qwen2.5-3B and TinyLlama.

Four-panel outcome of pairwise training on UltraFeedback

Figure 10. Outcome of pairwise training on UltraFeedback against the per-token margin the objective demands, δ/βeff\delta/\beta_{\mathrm{eff}} with βeff=β\beta_{\mathrm{eff}} = \beta for per-token-average scores and β∣y∣\beta|y| for summed scores. The dashed vertical line marks the median base gap at the rejected positions, 17.5 nat, and dotted lines mark the base model at 1.48 nat and 27%. (a) Under per-token averaging the well takes more than 95% of the released mass once the demand passes the base gap, whereas summed scores concentrate mass gradually and send almost none beyond rank 10,000. (b) Samples from the two lowest-demand per-token runs have a high negative log-likelihood under the base model, samples from per-token runs beyond the base gap a low one, and summed runs stay near the base model. (c) Every per-token run repeats under greedy decoding on at least 78% of prompts, against at most 37% for summed runs. (d) The final dose passes one only under per-token averaging with a demand above the base gap. Red marks runs with more than half of the released mass on one token and at least 30% degenerate samples, blue marks clean runs, and orange the runs in between. Doses are for each run’s leading token, or for Conf when that token takes less than 5% of the released mass.

Three-panel plots of offset norm, gain of $b$ on Conf, and dose $D$

Figure 11. Standard recipes build the same offset at a smaller dose. All runs train Llama-2-7B pairwise on UltraFeedback and are plotted against training step on log axes. Per-token averaging at β≤0.2\beta\le0.2 collapses, while textbook DPO at β=0.1\beta= 0.1 and SimPO at β∈{2.0,2.5}\beta\in\{2.0, 2.5\} do not within the base budget. (a) By the end of the base budget, SimPO has built an offset about as long as that of the collapsing runs, and textbook DPO one at most a sixth as long. (b) Only per-token averaging at β≤0.2\beta\le0.2 gives Conf, the well of the collapsing run, a logit gain above 20 nat, while textbook DPO and SimPO stay below 3 nat within the base budget. (c) By the end of the base budget, the doses of textbook DPO and SimPO sit four to twenty-five times below those of the collapsing runs, and full-parameter SimPO crosses D=1D = 1 at step 3000 under the three-fold budget. Squares mark full-parameter fine-tuning, and dashed lines mark three-fold-budget runs, which train for nine epochs under their own schedule. Full-parameter per-token averaging, which collapses onto the space token, and full-parameter textbook DPO appear only in (a).

Full-Parameter Onset

Under full-parameter fine-tuning, onset tracks the integrated learning rate, the learning rate summed over the steps taken. Two Llama-2 runs, at learning rates 5×10−75 \times10^{-7} and 10−610^{-6} under a 1512-step schedule, nucleated once it reached about 1.7×10−41.7 \times10^{-4}. Before running it, we then wrote down that a run at 10−610^{-6} under the 504-step schedule would nucleate between steps 200 and 300, and it nucleated at step 250. In all full-parameter runs with the per-token objective, the offset alone makes the well the top token at 98 to 100% of positions, while the fine-tuned unembedding rows alone do so at no more than 1.5%.

A Supervised Fine-Tuning Control

To check whether the offset needs a ranking term at all, we trained Llama-2-7B on the driving lists at three-fold budget with only the NLL term on the chosen response. Its offset reaches a norm of 56, and the offsets on chosen and rejected responses have a cosine of 0.99. However, the largest entry of W‾b\overline{W}^b is only 3.4 and the top token takes 8.6% of the released mass, so no well forms. Plain fine-tuning thus already builds a shared offset, while the ranking term decides where it points.

SimPO from a Supervised Fine-Tuning Checkpoint

We start from princeton-nlp/Llama-3-Base-8B-SFT, the Llama-3-8B model that the SimPO authors fine-tuned on UltraChat (Meng et al., 2024), and train it with their Llama-3-Base configuration of β=2.0\beta= 2.0, γ=1.0\gamma= 1.0, a learning rate of 6×10−76 \times10^{-7}, a global batch of 128, a cosine schedule with a warmup ratio of 0.1, and full-parameter updates. The data are our 15.7k UltraFeedback pairs rendered in the model’s chat template. The base budget of 369 steps makes three passes over these pairs, as many examples as about three quarters of one epoch of the official training data, and the three-fold budget of 1107 steps makes nine. We keep fp32 master weights with an 8-bit AdamW optimizer, where the authors use bf16 weights and AdamW.

Before training, we measured the base gap of three candidate wells at the rejected positions. Relative to Llama-3-8B, the SFT checkpoint enlarges the median base gap by 22% for Conf and by 10% for Apache and shrinks it by 2% for Sharper. Transplanting the offset of the Llama-3.1 full-parameter SimPO run at three-fold budget, whose well is Apache, gives a median dose of 1.12 on the SFT checkpoint against 1.27 on Llama-3-8B. By the dose, supervised fine-tuning should therefore delay a well rather than prevent it.

Table 6 reports both budgets. No well forms, and held-out ranking improves over the SFT checkpoint at both budgets. From one-fold to three-fold budget, the offset grows from 32.5 to 77.3 while its largest vocabulary entry stays small. At the base budget it lifts the ordinary Conf family by about 2 nat, and at three times the budget a flat spread of infrequent multilingual and code fragments by about 4 nat. SimPO from the base models lifts its leading tokens by about 18 nat. No token in the vocabulary reaches a median dose of 0.4, while SimPO from the base models carries Apache on Llama-3.1 and Conf on Llama-2 past one. Meanwhile the samples move away from the SFT model, whose per-token negative log-likelihood of them rises from 1.35 to 3.86. Against the SFT model’s own samples for the same 300 held-out prompts, Skywork-Reward-V2 and ArmoRM prefer the one-fold run’s samples on 63% and 65% of prompts and the three-fold run’s on 34% and 33%. The held-out chosen responses win on 73% and 72% (Table 7).

BudgetStepsTop tokenCapture / tailDegen.NLL∥bˉ∥\lVert\bar{b} \rVertDoseHeld-out top-1 / pairwise
1×369,2.5% / 0.6%1.3%1.3532.50.150.41 / 0.67
3×1107(2.7% / 4.1%0.0%3.8677.30.190.58 / 0.75

Table 6. SimPO from the public SFT checkpoint with the official hyperparameters. NLL is the SFT model’s per-token negative log-likelihood of sampled responses, the dose is for Conf, and held-out pairwise accuracy is 0.619 for the SFT checkpoint itself.

RunSkywork winArmoRM winPairwiseTop-10
Llama-2-7B
Per-token ranking, β=0.12\beta= 0.1225%6%0.680.02
with a reference model, 1× / 3×26% / 17%7% / 3%0.68 / 0.710.04 / 0.05
full-parameter24%45%0.700.14
SimPO, 1× / 3×7% / 6%3% / 4%0.67 / 0.730.04 / 0.04
SimPO, β=2.0\beta= 2.0, γ=1.6\gamma= 1.68%6%0.670.05
SimPO, full-parameter, 1× / 3×10% / 7%6% / 4%0.70 / 0.760.10 / 0.05
Textbook DPO, 1× / 3×68% / 66%67% / 65%0.63 / 0.640.77 / 0.77
Textbook DPO, full-parameter66%67%0.630.80
NLL only, λ=1\lambda= 164%64%0.630.75
Centering only20%23%0.680.42
Recipe, λNLL=1.0\lambda_{\mathrm{NLL}} = 1.0 / 0.165% / 52%66% / 55%0.63 / 0.660.74 / 0.45
Centering and loudness penalty17%15%0.690.17
Llama-3.1-8B
Per-token ranking, β=0.12\beta= 0.1211%5%0.740.02
SimPO, full-parameter, 1× / 3×13% / 3%14% / 5%0.78 / 0.790.01 / 0.03
Llama-3-8B from the SimPO authors’ SFT checkpoint
SimPO, official recipe, 1× / 3×63% / 34%65% / 33%0.67 / 0.750.67 / 0.46

Table 7. External reward models on 300 held-out UltraFeedback prompts. Win is the share of prompts on which the reward model prefers the run’s sample to that of the model the run starts from. Pairwise is held-out pairwise accuracy. Top-10 is the share of released mass that goes to the base model’s ten most likely tokens under teacher forcing. The base Llama-2-7B has a held-out pairwise accuracy of 0.60, the base Llama-3.1-8B 0.61, and the SFT checkpoint 0.62. The last four Llama-2 rows train on four-response lists and the others on pairs.

External Reward Models

Table 7 lists the reward-model evaluation behind § 8 and the Win column of Table 2. Each run generates one sample for each of the same 300 held-out UltraFeedback prompts, at temperature 0.7 with nucleus sampling at 0.95 and at most 256 new tokens. Each sample is compared with the sample of the model the run starts from for the same prompt, which is the base model except for the two runs from the SFT checkpoint. The chosen responses of the held-out lists win against the base model on 96% and 95% of prompts under the two reward models. Adding the SimPO offset alone to the base model, as in § 5, keeps its samples about as likely under the base model as the base model’s own samples. They cost about 0.6 nat per token, against 3.1 to 4.3 for the trained SimPO models, and when they fail they fall into ordinary repetition loops rather than strings of unrelated words.

Shapes of Well That Standard Rulers Miss

We compared the standard nn-gram repetition ruler with our mass-flow and word-stem measurements on every run. The standard ruler correctly flags the flagship run’s repetition loop, in which 96% of samples degenerate, but it misses three other shapes of well. A latent well appears only after the visible answer ends, meaning greedy decoding within a length-matched budget never reaches it. A non-repeating well keeps changing its surface form, for example a code-path continuation, while still capturing most of the released mass. The ruler then reports little repetition despite severe collapse. A word-family ring lacks a single dominant token, because a cluster of low-frequency, morphologically related tokens takes the released mass together. A ruler that counts single tokens therefore underreports the failure. Furthermore, these standard rulers miss the diffuse degradation from § 6, which leaves no repetitive patterns to count but is exposed by evaluating the base model’s likelihood of the generated samples. Any conventional ruler used in isolation can therefore pass a model that the offset and mass-flow measurements from § 3 reveal has already collapsed.

Supplementary Measurements

These measurements support claims of the main text, grouped by section.

The offset is shared (§ 5). Recomputed on held-out text with phrasing the model never saw, the flagship offset keeps a cosine of 0.85 with the original.

Dose threshold (§ 6). When the well takes over along natural training runs, it goes from under 10% of the rejected positions to a majority within 100 to 300 steps (Figure 9b).

Anomalous rows (§ 6). For every single-token well, we looked up the norm of its unembedding row and counted how often the token occurs in our training text and in WikiText-103. Most single-token wells of the adapter runs on the driving data never appear in our training text. In 13 bases, the highest-norm row belongs to a rare token, and its gain gtg_t can overtake base gaps of 17 to 39 nat. Under full-parameter fine-tuning on UltraFeedback, the Llama family collapses onto the rare base space token, and there too the offset alone makes the well the top token at nearly every position.

Losses and demand (§ 6). On the driving lists, the per-token loss with a reference model collapses onto text tt at the base budget and onto a rare Cyrillic token at the three-fold budget. Per-token training at β\beta of 1 and 10, whose demand is also low, degrades in the same diffuse way (Figure 10). With three times the budget, the SimPO offset concentrates its projection on a few vocabulary rows even where it becomes shorter. By the end of full-parameter training, its leading token takes 27 to 45% of the released mass (Table 2).

Ranking cost of the recipe (§ 7). Eight base models have treated runs in our held-out evaluation. We compare each treated run with the untreated run of the same base and budget, or with the NLL-only run where no untreated run was scored. Every treated run but one lies between 1.8 points below and 0.1 points above its comparison run. The exception is the loudness variant of the recipe on Gemma-2 9B at the base budget, which barely lowers the rejected responses in training and scores 3.1 points below the untreated run, while the combined penalty on the same base scores 0.3 points below it.

Margin along training (§ 8). The margin reaches three or four times its value at the point where ranking saturates. At a matched margin of 12 to 16 nat, Llama-2 runs without offset control have already channeled 42 to 51% of the released mass into a single well, against 8 to 20% for offset-controlled runs (Figure 13). The margin at which capture takes off also differs by base model, from 13 to 27 nat on six-fold OLMo-3 to 44 to 59 nat on OLMo-2, so the margin cannot serve as a stopping criterion.

Held-out ranking accuracy and top token's share of released mass against preference margin

Figure 13. (a) Held-out ranking accuracy and (b) the top token’s share of released mass against the preference margin along training. Ranking saturates early for every run, while the top token’s share keeps growing with the margin only in the runs without offset control.

Seeds and Scope

Both untreated seeds of the flagship settings collapse onto the same well as the first, textt on the driving lists and Conf on UltraFeedback. Their offset norms lie within 3% of the first seed’s, and their dose crosses one no later than the well takes over. Both treated seeds stay free of a well, with 10 to 12% residual degeneration and unchanged held-out ranking. Adapter rank from 8 to 128 and attention-only adapters change when a well forms rather than whether. Across these ranks, offset norms lie between 80 and 117 on Llama-2 and Llama-3.1, and rank 8 on Llama-2 nucleates the same well at three-fold budget. The demand analysis of § 6 covers pairwise UltraFeedback training under per-token averaging with β\beta from 0.05 to 10, under textbook DPO with β\beta from 0.1 to 0.0003, and under SimPO with β\beta and γ\gamma from its published tuning range. These runs use our adapter and full-parameter protocols for three or nine epochs.

Common-Shift Invariance and the Trained Offset

Write sθ(x,y)s_\theta(x,y) for the score of a response, the per-token average of § 3 or the summed log-ratio of textbook DPO. Both LPL\mathcal{L}_{\mathrm{PL}} and the DPO loss depend on the scores of a prompt’s responses only through their differences, since the DPO reference term does not depend on θ\theta.

Proposition 1 (Loss-level invariance). Let ℓ\ell be invariant under adding a constant to the scores of all responses of a prompt, and write a score perturbation as ε=εˉ+ε~\varepsilon=\bar{\varepsilon}+\tilde{\varepsilon} with εˉ(x)\bar{\varepsilon}(x) its mean over the prompt’s responses. Then ℓ(s+ε)=ℓ(s+ε~)\ell(s+\varepsilon)=\ell(s+\tilde{\varepsilon}), so the common part εˉ\bar{\varepsilon} cannot be identified from the objective with any amount of preference data, and ∑y∂ℓ/∂s(x,y)=0\sum_y \partial\ell/\partial s(x,y)=0 for every prompt.

Proof. Score differences within a prompt cancel εˉ\bar{\varepsilon}. Differentiating ℓ(s+c1)=ℓ(s)\ell(s+c\mathbf{1})=\ell(s) in cc at c=0c=0 gives the gradient identity.

Proposition 2 (First-order constraint on a shared hidden shift). Let WW be the unembedding with rows wvw_v, hth_t the final hidden state at position tt and pt=softmax⁡(Wht)p_t=\operatorname{softmax}(Wh_t). Shifting every hidden state of every response by the same δ\delta changes a per-token-average score by

s(x,y;δ)−s(x,y;0)=1∣y∣∑t[⟨wyt,δ⟩−log⁡Ev∼pte⟨wv,δ⟩]s(x,y;\delta)-s(x,y;0)=\frac{1}{|y|}\sum_t\left[\langle w_{y_t},\delta\rangle-\log\mathbb{E}_{v\sim p_t}e^{\langle w_v,\delta\rangle}\right]
=⟨gy,δ⟩−12∣y∣∑tVar⁡v∼pt⟨wv,δ⟩+O(∥δ∥3),=\langle g_y,\delta\rangle-\frac{1}{2|y|}\sum_t\operatorname{Var}_{v\sim p_t}\langle w_v,\delta\rangle+O(\lVert\delta\rVert^3),

with gy=1∣y∣∑t(wyt−Ev∼ptwv)g_y=\frac{1}{|y|}\sum_t(w_{y_t}-\mathbb{E}_{v\sim p_t}w_v) the mean gradient of the score with respect to the hidden states. Hence ∇δL∣δ=0=Ex∑y∂ℓ∂s(x,y)gy\nabla_\delta\mathcal{L}|_{\delta=0}=\mathbb{E}_x\sum_y\frac{\partial\ell}{\partial s(x,y)}g_y, which for a pairwise logistic loss equals −Ex[a(x)(gyw−gyl)]-\mathbb{E}_x[a(x)(g_{y_w}-g_{y_l})] with a(x)=βσ(−β(Δθ−Δref))>0a(x)=\beta\sigma(-\beta(\Delta\theta-\Delta_{\mathrm{ref}}))>0 and Δref=0\Delta_{\mathrm{ref}}=0 without a reference model. To first order the objective constrains δ\delta along this single direction, and on the orthogonal complement only the second-order term remains, which enters through the difference between the responses’ mean per-position variances.

Proof. The first equality is the change of log softmax under the logit shift WδW\delta, the second the cumulant expansion of log⁡Ee⟨wv,δ⟩\log\mathbb{E}e^{\langle w_v,\delta\rangle}, and the gradient follows from the chain rule with coefficients that sum to zero within each prompt by Proposition 1.

Three remarks bound the statement. The constraint is first-order and local, since the softmax is nonlinear and minibatch gradients differ from the expectation, so the complement is weakly rather than exactly unconstrained. The shift δ\delta is not a parameter but the image of parameter updates through Jacobians shared across positions and both responses. This sharing is how a component in the weakly constrained part arises at all, and gradient entanglement (Yuan et al., 2025) describes the coupling that produces it. Any term that depends on absolute scores, such as an NLL term on the chosen response, breaks the invariance, which is one reason such terms delay displacement (§ 2).

Table 8 measures how much of the trained offset lies in the weakly constrained part. For each run we add the offset δ\delta of § 3 to the base model’s final hidden states on 200 held-out lists. For each list we record the change of the chosen per-token score, a+a^{+}, and of the mean rejected per-token score, a−a^{-}, and split it into a common part c=(a++a−)/2c = (a^{+} + a^{-})/2 and a differential part d=(a+−a−)/2d = (a^{+} - a^{-})/2. The table reports the means of a+a^{+}, a−a^{-}, and cc over the lists and the mean of ∣d∣\lvert d\rvert, which exceeds half the difference of the mean scores because dd changes sign from list to list. We report the same two means for three norm-matched random directions, and the cosine between δ\delta and uˉ=1N∑x(gy(1)−1K−1∑j=2Kgy(j))\bar{u} = \frac{1}{N} \sum_x \left(g_{y^{(1)}} - \frac{1}{K-1} \sum_{j=2}^{K} g_{y^{(j)}}\right), the base model’s chosen-minus-rejected score gradient with every list and every rejected response weighted equally. For the three per-token UltraFeedback runs, which pair the chosen response with a randomly drawn rejected one, uˉ\bar{u} is the direction of Proposition 2 with equal weights, and at the base model those weights vary across lists by only 5 to 7% of their mean. For the driving run uˉ\bar{u} approximates that direction, since the K=4K = 4 loss weights the rejected responses by rank. In the four collapsing runs the common part is 20 to 530 times the differential part and the cosine is at most 0.16. The offset carries the collapse-producing component without accounting for all of the likelihood displacement. On the driving run it alone lowers the chosen score by 65 nat while training lowered it by 21 on the same lists on the same lists, so the position-specific residual partly compensates its effect on the chosen response, consistent with § 5. For the two standard-recipe runs both parts are small, below 0.02 nat under textbook DPO and 0.25 and 0.06 nat under SimPO, because their quiet offsets lift no token near the top.

BaseData, objective∥δ∥\lVert\delta\rVerta+a^{+}a−a^{-}cc∣d∣\lvert d\rvertrandom ∣c∣\lvert c\rvert / ∣d∣\lvert d\rvertcos
Llama-2-7BUltraFeedback, per-token β=0.12\beta= 0.12 (collapses)84−14.5−15.2−14.90.680.39 / 0.060.03
Llama-2-7Bdriving, K=4K = 4 (collapses)93−65.1−65.2−65.20.120.49 / 0.020.16
Qwen2.5-7BUltraFeedback, per-token β=0.12\beta= 0.12 (collapses)282−26.0−27.1−26.61.362.15 / 0.31−0.02
Llama-3.1-8BUltraFeedback, per-token β=0.12\beta= 0.12 (collapses)109−29.2−30.1−29.60.940.54 / 0.100.05
Llama-2-7BUltraFeedback, textbook DPO β=0.1\beta= 0.114−0.005−0.013−0.0090.0140.011 / 0.0080.07
Llama-2-7BUltraFeedback, SimPO β=2.5\beta= 2.5, γ=1.0\gamma= 1.0114−0.23−0.26−0.250.060.57 / 0.090.01

Table 8. The trained offset δ\delta added to the base model’s final hidden states on 200 held-out lists. a+a^{+} and a−a^{-} are the mean changes of the chosen and mean rejected per-token scores in nat, cc the mean of (a++a−)/2(a^{+} + a^{-})/2 and ∣d∣\lvert d\rvert the mean of ∣a+−a−∣/2\lvert a^{+} - a^{-}\rvert/2 over lists; “random” the same two means for three norm-matched random directions, and cos the cosine between δ\delta and uˉ\bar{u}, the base model’s chosen-minus-rejected score gradient.

References

  1. [1]Marah Abdin, Sam Ade Jacobs, Ammar Ahmad Awan, Jyoti Aneja, et al. Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phone. arXiv [cs.CL], 2024. URL https://arxiv.org/abs/2404.14219.
  2. [2]Daniel Biš, Maksim Podkorytov, and Xiuwen Liu. Too much in common: Shifting of embeddings in transformer language models and its implications. In Kristina Toutanova, Anna Rumshisky, Luke Zettlemoyer, Dilek Hakkani-Tur, Iz Beltagy, Steven Bethard, Ryan Cotterell, Tanmoy Chakraborty, and Yichao Zhou (eds.), Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, pp. 5117–5130, Stroudsburg, PA, USA, 2021. Association for Computational Linguistics. doi: 10.18653/v1/2021.naacl-main.403. URL http://dx.doi.org/10.18653/v1/2021.naacl-main.403.
  3. [3]Ralph Allan Bradley and Milton E Terry. Rank analysis of incomplete block designs: I. The method of paired comparisons. Biometrika, 39(3/4):324–345, 1952. ISSN 0006-3444,1464-3510. doi: 10.2307/2334029. URL http://dx.doi.org/10.2307/2334029.
  4. [4]Aakanksha Chowdhery, Sharan Narang, Jacob Devlin, et al. PaLM: Scaling Language Modeling with Pathways. Journal of Machine Learning Research, 2023. URL https://arxiv.org/abs/2204.02311.
  5. [5]Ganq u Cui, Lifan Yuan, Ning Ding, Guanming Yao, Bingxiang He, Wei Zhu, Yuan Ni, Guotong Xie, Ruobing Xie, Yankai Lin, Zhiyuan Liu, and Maosong Sun. UltraFeedback: Boosting language models with scaled AI feedback. In Proceedings of the 41st International Conference on Machine Learning, pp. 9722–9744, 2024. doi: 10.48550/arXiv.2310.01377. URL http://dx.doi.org/10.5555/3692070.3692454.
  6. [6]Yuanhao Ding, Meimingwei Li, Esteban Garces Arias, Matthias Aßenmacher, Christian Heumann, and Chongsheng Zhang. Breaking the likelihood trap: Variance-Calibrated Modulation for large language model decoding. arXiv [cs.CL], 2026. doi: 10.48550/arXiv.2606.22511. URL https://arxiv.org/abs/2606.22511. Accepted at EMNLP 2026.
  7. [7]Kawin Ethayarajh. How contextual are contextualized word representations? Comparing the geometry of BERT, ELMo, and GPT-2 embeddings. In Kentaro Inui, Jing Jiang, Vincent Ng, and Xiaojun Wan (eds.), Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP), pp. 55–65, Stroudsburg, PA, USA, 2019. Association for Computational Linguistics. doi: 10.18653/v1/d19-1006. URL http://dx.doi.org/10.18653/v1/D19-1006.
  8. [8]Kawin Ethayarajh, Winnie Xu, Niklas Muennighoff, Dan Jurafsky, and Douwe Kiela. KTO: Model alignment as prospect theoretic optimization. In Proceedings of the 41st International Conference on Machine Learning, 2 February 2024. doi: 10.48550/arXiv.2402.01306. URL http://dx.doi.org/10.5555/3692070.3692574.
  9. [9]Jun Gao, Di He, Xu Tan, Tao Qin, Liwei Wang, and Tie-Yan Liu. Representation Degeneration Problem in Training Natural Language Generation Models. In 7th International Conference on Learning Representations, ICLR 2019, New Orleans, LA, USA, May 6-9, 2019. OpenReview.net, 2019. URL https://dblp.org/rec/conf/iclr/GaoHTQWL19.html.arxiv.org/abs/1907.12009
  10. [10]Gemma Team, Morgane Riviere, Shreya Pathak, et al. Gemma 2: Improving open language models at a practical size. arXiv [cs.CL], 2024. doi: 10.48550/arXiv.2408.00118. URL http://arxiv.org/abs/2408.00118.
  11. [11]Mohammad Gheshlaghi Azar, Zhaohan Daniel Guo, Bilal Piot, Remi Munos, Mark Rowland, Michal Valko, and Daniele Calandriello. A general theoretical paradigm to understand learning from human preferences. In Sanjoy Dasgupta, Stephan Mandt, and Yingzhen Li (eds.), Proceedings of The 27th International Conference on Artificial Intelligence and Statistics, volume 238 of Proceedings of Machine Learning Research, pp. 4447–4455. PMLR, 2024. URL https://proceedings.mlr.press/v238/gheshlaghi-azar24a.html.
  12. [12]Aaron Grattafiori, Abhimanyu Dubey, Abhinav Jauhri, et al. The Llama 3 herd of models. arXiv [cs.AI], 2024. doi: 10.48550/arXiv.2407.21783. URL http://arxiv.org/abs/2407.21783.
  13. [13]Ari Holtzman, Jan Buys, Li Du, Maxwell Forbes, and Yejin Choi. The Curious Case of Neural Text Degeneration. In International Conference on Learning Representations, 2020. URL https://openreview.net/forum?id=rygGQyrFvH.
  14. [14]Jiwoo Hong, Noah Lee, and James Thorne. ORPO: Monolithic preference optimization without reference model. In Yaser Al-Onaizan, Mohit Bansal, and Yun-Nung Chen (eds.), Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing, pp. 11170–11189, Stroudsburg, PA, USA, 2024. Association for Computational Linguistics. doi: 10.18653/v1/2024.emnlp-main.626. URL http://dx.doi.org/10.18653/v1/2024.emnlp-main.626.
  15. [15]Edward J Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yanzhi Li, Shean Wang, Lu Wang, and Weizhu Chen. LoRA: Low-Rank Adaptation of Large Language Models. In *International Conference on Learning Representations*, 2022. URL https://openreview.net/forum?id=nZeVKeeFYf9.arxiv.org/abs/2106.09685
  16. [16]Shawn Im, Federico Danieli, Skyler Seto, Barry-John Theobald, and Katherine Metcalf. Normalized rewards for preference optimization. In *Proceedings of the 43rd International Conference on Machine Learning*, Seoul, South Korea, 2026. doi: 10.48550/arXiv.2607.16240. URL http://dx.doi.org/10.48550/arXiv.2607.16240.
  17. [17]Albert Q Jiang, Alexandre Sabl a yrolles, Arthur Mensch, et al. Mistral 7B. *arXiv [cs.CL]*, 2023. doi: 10.48550/arXiv.2310.06825. URL http://arxiv.org/abs/2310.06825.
  18. [18]M G Kendall. A new measure of rank correlation. *Biometrika*, 30(1/2):81–93, 1938. ISSN 0006-3444,1464-3510. doi: 10.2307/2332226. URL http://dx.doi.org/10.2307/2332226.
  19. [19]Sander Land and Max Bartolo. Fishing for magikarp: Automatically detecting under-trained tokens in large language models. In Yaser Al-Onaizan, Mohit Bansal, and Yun-Nung Chen (eds.), *Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing*, pp. 11631–11646, Stroudsburg, PA, USA, 2024. Association for Computational Linguistics. doi: 10.18653/v1/2024.emnlp-main.649. URL http://dx.doi.org/10.18653/v1/2024.emnlp-main.649.
  20. [20]Andrew Lee, Xiaoyan Bai, Itamar Pres, Martin Wattenberg, Jonathan K Kummerfeld, and Rada Mihalcea. A Mechanistic Understanding of Alignment Algorithms: A Case Study on DPO and Toxicity. In *International Conference on Machine Learning*, pp. 26361–26378. PMLR, 2024. URL https://proceedings.mlr.press/v235/lee24a.html.arxiv.org/abs/2401.01967
  21. [21]Yun Li, Ehsan Javanmardi, Simon Thompson, Kai Katsumata, Alex Orsholits, and Manabu Tsukada. Multi-PrefDrive: Optimizing Large Language Models for Autonomous Driving Through Multi-Preference Tuning. In *2025 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS)*, pp. 4347–4354, 2025. URL https://ieeexplore.ieee.org/abstract/document/11247608/.DOI
  22. [22]Tianqi Liu, Zhen Qin, Junru Wu, Jiaming Shen, Misha Khalman, Rishabh Joshi, Yao Zhao, Mohammad Saleh, Simon Baumgartner, Jialu Liu, Peter J Liu, and Xuanhui Wang. LiPO: List-wise preference optimization through learning-to-rank. In Luis Chiruzzo, Alan Ritter, and Lu Wang (eds.), *Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers)*, pp. 2404–2420, Stroudsburg, PA, USA, 2025. Association for Computational Linguistics. ISBN 9798891718696. doi: 10.18653/v1/2025.naacl-long.121. URL http://dx.doi.org/10.18653/v1/2025.naacl-long.121.
  23. [23]Yuhao Liu, Liang Zeng, Yuzhen Xiao, Jujie He, Jiacai Liu, Chaojie Wang, Rui Yan, Wei Shen, Fuxiang Zhang, Jiacheng Xu, and Yang Liu. Skywork-Reward-V2: Scaling Preference Data Curation via Human-AI Synergy. In *International Conference on Learning Representations*, volume 2026, pp. 133805–133838, 2026. URL https://proceedings.iclr.cc/paper_files/paper/2026/hash/d8836d0294651a867c2ca51012923997-Abstract-Conference.html.arxiv.org/abs/2507.01352
  24. [24]R D Luce. *Individual choice behavior: A theoretical analysis.* Dover Publications, New York, NY, 2005. ISBN 978084441368. URL https://books.google.com/books?hl=en&lr=&id=D7AqWAAQBAJ&oi=fnd&pg=PP1&dq=Individual+Choice+Behavior:+A+Theoretical+Analysis&ots=1h7LEH7B-g&sig=x5QbofqnxfQbtctkLNX13ZbPjBM.
  25. [25]Iuri Macocco, Nora Graichen, Gemma Boleda, and Marco Baroni. Not a nuisance but a useful heuristic: Outlier dimensions favor frequent tokens in language models. In Yonatan Belinkov, Aaron Mueller, Najoung Kim, Hosein Mohebbi, Hanjie Chen, Dana Arad, and Gabriele Sarti (eds.), *Proceedings of the 8th BlackboxNLP Workshop: Analyzing and Interpreting Neural Networks for NLP*, pp. 109–136, Stroudsburg, PA, USA, 2025. Association for Computational Linguistics. ISBN 9798891763463. doi: 10.18653/v1/2025.blackboxnlp-1.6. URL http://dx.doi.org/10.18653/v1/2025.blackboxnlp-1.6.

Paper details

vlnc:W_zjawfm7v
v1 · latestHistory (1)

published · Fresh — added to the Hub within the last 30 days

added ·

Usage

views · 0

downloads · 0

A view counts once per browser per day, a download each time the PDF is requested.

CC BY 4.0

Original source

  • Computer Science

Claude Code, Gemini · View statement →

Looking for related papers…
Contents