Preference learning methods like Direct
Preference Optimization (DPO) are known to induce reliance on spurious
correlations, leading to sycophancy and length bias in today’s language
models and potentially severe goal misgeneralization in future systems.
In this work, we provide a unified theoretical analysis of this
phenomenon, characterizing the mechanisms of spurious learning, its
consequences on deployment, and a provable mitigation strategy. Focusing
on log-linear policies, we show that standard preference-learning
objectives induce reliance on spurious features at the population level
through two channels: mean spurious bias and causal-spurious correlation
leakage. We then show that this reliance creates an irreducible
vulnerability to distribution shift: more data from the same training
distribution fails to reduce the model’s dependence on spurious
features. To address this, we propose tie training, a data augmentation
strategy using ties (equal-utility preference pairs) to introduce
data-driven regularization. We demonstrate that this approach
selectively reduces spurious learning without degrading causal learning.
Finally, we validate our theory on log-linear models and provide
empirical evidence that both the spurious learning mechanisms and the
benefits of tie training persist for neural networks and large language
models.
1. Introduction
Aligning large language models (LLMs) with human preferences is a key
challenge in building safe and useful AI systems. In current alignment
pipelines, Reinforcement Learning from Human Feedback (RLHF) learns a
reward model from preference data and optimizes a policy with respect to
that reward (Ziegler et al. 2019; Ouyang et al. 2022). Direct Preference
Optimization (DPO) simplifies this pipeline by directly optimizing the
policy on preference pairs (Rafailov et al. 2023). These approaches are
used to align widely deployed systems such as ChatGPT (Ouyang et al.
2022) and Claude (Bai et al. 2022). Despite their success, existing
theory provides limited insight into how preference optimization behaves
under the distributional structure of real-world human feedback.
Understanding this behavior requires examining the structure of
preference data itself. Preference optimization methods are trained on
datasets of human comparisons (Christiano et al. 2017; Rafailov et al.
2023), where annotators select preferred responses for given a prompt.
These datasets encode recurring patterns, reflecting consistent
annotator biases and shared superficial characteristics among preferred
responses. As a result, feature-level correlations arise between surface
attributes and preference labels that are not causally related to
response quality.
Surface-level attributes such as length, politeness, formatting, or
agreement with the user often correlate with preference labels during
training (Sharma et al. 2024; Casper et al. 2023), but may not reflect
true response quality. When these correlations shift at deployment,
models that rely on them fail to generalize, a behavior we call policy
misgeneralization: rather than learning to optimize response quality,
the policy learns to optimize a proxy that is non-causally correlated
with response quality on the training distribution.
Policy misgeneralization has important safety implications beyond
standard failures under distribution shift. Prior work has raised
concerns that AI systems may learn objectives that correlate with
intended goals during training but pursue misaligned proxy objectives
once those correlations break at deployment (Langosco et al. 2022; R.
Shah et al. 2022). In such cases, high training reward can reflect
alignment with proxy signals rather than improvements in true task
performance, masking failures that emerge under distribution
shift (Skalse et al. 2022). While much of this literature focuses on
hypothetical capable agents (Ngo, Chan, and Mindermann 2024; Bengio et
al. 2025), preference optimization in current LLMs provides a concrete
setting where this failure mode manifests even without objective
misspecification. Understanding the mechanisms by which spurious
correlations emerge in these systems is therefore essential for
developing robust alignment methods.
Despite these risks, existing work on spurious correlations in
preference optimization remains largely empirical. Prior studies report
failures such as verbosity (Saito et al. 2023), but describe symptoms
rather than identify underlying mechanisms. While supervised learning
has developed mathematical frameworks for analyzing spurious
correlations through shortcut learning (Geirhos et al. 2020), preference
optimization methods such as DPO lack analogous theory. Without such
understanding, mitigation strategies remain largely heuristic and lack
principled guarantees.
To address this gap, we develop a mathematical framework for spurious
correlation learning in preference optimization. We analyze log-linear
DPO as a representative and tractable testbed for pairwise preference
optimization, and characterize how feature correlations interact with
the optimization objective. Our contributions are:
(i) We characterize the mechanism of spurious learning by
analyzing the population equilibrium of the linearized log-linear DPO
objective. We prove that mean spurious bias or causal-spurious
correlation in the training distribution induces nonzero spurious
parameters (Theorem 3). This shows spurious learning
arises structurally from the data, not from finite-sample effects or
optimization noise.
(ii) We analyze deployment consequences when spurious
statistics shift between training and deployment. We use the expected
preference margin as a population-level deployment proxy to characterize
the shift term (Propositions 4 and
5). To understand
finite-sample behavior, we decompose deployment suboptimality into an
irreducible shift term driven by spurious parameters and a reducible
estimation term that decays as (Theorem 6).
This shows that scaling training data cannot eliminate shift-induced
error.
(iii) We propose tie training, a data augmentation strategy
that reduces spurious correlation reliance by adding preference pairs
with equal utility but differing spurious features. These ties inject
curvature along spurious directions, selectively regularizing spurious
parameters without affecting causal learning (Theorem 8 (i)). We prove that such ties
can reduce the irreducible shift error at deployment (Theorem 8 (iii)).
We validate our framework through controlled experiments that
progressively relax modeling assumptions. Linear models confirm
quantitative agreement with theory. Neural networks show the same
qualitative mechanisms persist despite hidden representations. In large
language models, tie training reduces spurious correlation
learning without compromising in-distribution accuracy.
2. Related Work
Spurious correlation learning. Spurious correlations
in supervised learning are a well-established failure mode (Singla and
Feizi 2022). Models trained via empirical risk minimization (ERM) often
exploit surface-level features that correlate with labels in the
training distribution but lack a causal relationship to the target task,
a phenomenon referred to as shortcut learning (Geirhos et al. 2020),
simplicity bias (H. Shah et al. 2020; Morwani et al. 2023), or spurious
feature reliance (Arjovsky et al. 2019). When spurious correlations
shift at deployment (C. Zhou et al. 2021), models suffer prediction
errors (Sagawa et al. 2020), biased outcomes (Geirhos et al. 2018), and
performance degradation (Xiao et al. 2021). Proposed mitigations include
data augmentation (Chang, Adam, and Goldenberg 2021; Plumb, Ribeiro, and
Talwalkar 2022), re-weighting of minority examples (Liu et al.
2021), and modified training dynamics (Izmailov et al. 2022; Kirichenko,
Izmailov, and Wilson 2023), though these approaches often require domain
knowledge or explicit annotation of spurious features. Theoretically,
spurious learning has been analyzed through optimization dynamics, where
gradient descent preferentially fits easier features early in training,
leading to gradient starvation (Rahaman et al. 2019; Kalimeris et al.
2019; Qiu, Kuang, and Goel 2024), as well as through NTK and linearized
analyses that characterize implicit biases (Pezeshki et al. 2021;
Hermann et al. 2024). Most closely related to our setting, (Bombari and
Mondelli 2025) study high-dimensional linear models under ERM, deriving
closed-form solutions that reveal how data covariance induces spurious
feature reliance. However, these theories assume pointwise loss
landscapes and do not extend to preference optimization, where pairwise
comparisons induce fundamentally different learning dynamics.
Empirical failures in preference optimization. Prior
work on preference optimization has documented spurious correlation
learning primarily through empirical observations of reward-hacking
behaviors. Studies show that RLHF and DPO models exploit surface-level
artifacts, such as verbosity bias (preferring longer responses
independently of quality (Singhal et al. 2024; Saito et al. 2023)),
sycophancy (agreeing with user beliefs to maximize reward (Sharma et
al. 2024)), and formatting bias (over-optimizing for numbered lists or
stylistic markers (Zhang et al. 2025)). Existing mitigations target
individual symptoms through ad-hoc interventions, including length
penalties (Park et al. 2024) for verbosity or synthetic data
filtering (L. Chen et al. 2024). These approaches treat each bias in
isolation without addressing the underlying learning dynamics that
produce them. In contrast, we provide a population-level analysis that
reveals the structural mechanisms driving these empirically observed
failures.
Theoretical analysis of preference optimization.
Theory for preference optimization has developed along several lines.
Work on linear contextual dueling bandits and dueling reinforcement
learning establishes regret minimization under realizable reward
assumptions (Dudík et al. 2015; Saha, Pacchiano, and Lee 2023). Recent
alignment work develops robust or safety-motivated analysis and training
procedures, including robust formulations (Xiong et al. 2024; Wu et al.
2025), noise-aware losses (Chowdhury, Kini, and Natarajan 2024),
privacy-preserving constraints (K. Chen et al. 2025; X. Zhou, Wu, and
Orabona 2025), and divergence-based alignment objectives that explicitly
separate preferred and rejected behaviors (Haldar et al. 2025).
Complementary analyses in linear or log-linear regimes motivate
simplified preference models and analyze learning behavior under
idealized assumptions (Zhu, Jordan, and Jiao 2023; Chowdhury, Zhou, and
Natarajan 2024; X. Zhou, Wu, and Orabona 2025). However, across these
lines of work, a critical assumption persists: that the learned feature
representation is valid for the target task. These approaches address
stochastic or adversarial failures through algorithmic modifications but
do not characterize systematic spurious correlation learning.
3. Preliminaries
3.1. Preference Learning Setup
Preference dataset. We consider a preference dataset
, where denotes the
human-preferred and rejected responses to prompt , following
standard pairwise preference supervision (Rafailov et al. 2023;
Christiano et al. 2017). We represent each prompt–response pair with a feature vector and train on
the feature difference
Causal and spurious feature decomposition. We
decompose each feature vector as . The
causal component captures true response utility, while the
spurious component correlates with observed preferences in the
training data through statistics that need not persist at deployment.
This decomposition induces a corresponding split of feature differences,
. In practice, spurious features arise from data
collection biases and domain-specific structure, such as annotators
preferring longer or more formal responses at fixed content quality. We
call spurious not because it
cannot influence annotators but because, by design, these correlations
should not determine policy behavior at test time. We formalize this
distinction through the following invariance assumption.
Assumption 1 (Invariance). Under interventions that
modify spurious features while holding causal features fixed, human
preferences remain unchanged.
3.2. Log-Linear Policy and Data
Generation
Policy model. We adopt a log-linear policy, a common
regime for theoretical analysis that enables tractable characterization
of learning dynamics (Zhu, Jordan, and Jiao 2023; X. Zhou, Wu, and
Orabona 2025). The policy takes the form where denotes the learnable parameter vector. We
decompose the parameter vector as to match the
causal–spurious feature split.
Data generation. Our analysis depends only on the
feature differences and
their stated moment conditions. We denote by the ground-truth parameter
encoding true utility. By Assumption 1,
spurious features do not affect human preferences, so .
3.3. Direct Preference
Optimization
We analyze Direct Preference Optimization (DPO) (Rafailov et
al. 2023) as a method for fitting the preference model described above.
DPO optimizes the policy relative to a reference policy , so we
express learning in terms of the deviation . The DPO loss for a single preference
example is (1) where controls the KL regularization strength. Equation (1) shows that, under the
log-linear parameterization, DPO reduces to logistic regression on
feature differences. While this reduction provides a well-defined
population objective, the sigmoid nonlinearity generally prevents
closed-form characterization of the population optimum.
Linearization regime. To enable analytical progress,
we work in a local regime where score differences remain moderate,
allowing a first-order Taylor expansion of the sigmoid. Formally:
Assumption 2 (Local regime). With high probability
under the data distribution,
This regime arises near initialization, for bounded feature
magnitudes, or when the scaled DPO margin
remains small. Under this linearization, the population optimum admits a
closed-form solution that reveals how data structure drives spurious
learning.
4. Population-Level Spurious
Learning
In this section, we analyze the population equilibrium of the DPO
objective under a local linearization. We show that spurious correlation
learning arises generically: mean spurious bias or causal–spurious
correlation in the training data leads to nonzero spurious parameters at
the population optimum.
4.1. Population Gradient and
Early-Training Drift
We study the population objective (2)
where the expectation is taken over preference data generated using the
ground truth parameters . A single preference pair
induces the gradient
where
denotes the predicted score difference and .
Hard or misclassified pairs receive larger weight , while confidently
satisfied preferences are downweighted. Taking expectations, the
population gradient is (3)Early-training drift. Near the reference policy (), score
differences are small, so . The
gradient simplifies to (4) where and are the mean feature differences.
Whenever , spurious
features are learned from the first gradient step: This shows that spurious features with
nonzero mean differences are immediately learned, moving the DPO update
in spurious directions.
Early-training drift shows that mean spurious bias drives immediate
spurious learning. However, this does not guarantee spurious parameters
remain nonzero at equilibrium. The gradient dynamics could drive them
back to zero. We now characterize the population equilibrium to
determine when spurious learning persists.
4.2. Linearized Equilibrium
Under Assumption 2, we
linearize the weighting function and substitute into the population
gradient. This yields (see Appendix A.1 for
details) (5) At the
linearized stationary point , the gradient vanishes:
(6)
Partitioning according to the causal-spurious split, we write (7) where ,, and captures
causal-spurious correlation. The equilibrium conditions are (8)Explicit solution via Schur
complement. The equilibrium conditions (8)–(8) couple causal and
spurious parameters through the cross-covariance . To isolate the spurious
component, we use the Schur complement method. Assume and the Schur complement
are
invertible. These conditions hold when . Under these assumptions,
the spurious component at equilibrium is (9) The full derivation is provided in Appendix A.2.
4.3. Main Mechanism Result
We now state our main result characterizing spurious learning at the
population equilibrium.
Theorem 3 (Spurious learning from mean bias and
correlation leakage). Under the linearized population dynamics
(Equation (5)),
assume .
If (mean spurious bias)
or
(causal-spurious correlation), then generically . That is,
the population optimum assigns nonzero weight to spurious
features.
(i) Mean spurious bias (): When spurious features are asymmetrically distributed
across preference pairs, they directly contribute to , even in the
absence of correlation ().
(ii) Correlation leakage (): When spurious features correlate with causal
features, weight intended for causal directions leaks into spurious
ones. This operates even when spurious features are unbiased ().
Deviation from ground truth. By Assumption 1, spurious features should
not determine test-time behavior. Theorem 3 shows
that the learned DPO update nevertheless satisfies under
generic conditions on the training distribution. This creates a
systematic bias: the learned policy deviates from the true preference
model in directions that do not affect utility. Whether this deviation
causes deployment failures depends on how spurious feature statistics
differ between training and deployment, which we formalize in Section 5.
5. Spurious Learning Deployment
Error
Section 4 showed that learned parameters
generically satisfy . We now study the consequences for deployment. Spurious
learning creates a potential vulnerability: learned parameters
depend on features that do not affect true utility. Whether this
vulnerability translates into deployment error depends on how spurious
statistics shift between training and deployment. We show that when
spurious statistics shift, the population objective becomes sensitive to
learned spurious parameters, with degradation occurring when the shift
has the harmful sign. We then decompose deployment suboptimality into a
shift component and an estimation component, showing that the former
persists regardless of training set size.
5.1. Distribution Shift Setup
Let denote the training
distribution and the deployment
distribution over preference pairs . These distributions may differ in their feature
statistics. We denote: (10) The model learns parameters by optimizing on
and deploys with these fixed
parameters on . Differences
between and can induce deployment error
through the learned spurious parameters . We
defer the full mathematical characterization of canonical shift
scenarios (suppression, adversarial, and rotation) to Appendix E.
5.2. Expected Margin as Deployment
Metric
To assess how shifts affect performance, we use the expected
preference margin, which provides a tractable first-order
characterization of model quality under the local regime.
Margin definition. The expected margin measures the
average score gap between preferred and dispreferred responses: The margin decomposes
into causal and spurious components: (11)
We now show that margin differences provide a first-order
approximation to objective differences.
Proposition 4 (First-order margin approximation).
Under the local regime (Assumption 2), the
DPO objective
satisfies (12)
The proof follows from Taylor expansion of the log-sigmoid; see
Appendix C.1.
Proposition 4 shows that margin differences
drive objective changes. We now isolate the spurious component by
considering shifts that preserve causal statistics.
Proposition 5 (Spurious margin drives shift).
When causal statistics are stable (), the objective
difference at is governed by
the spurious margin: (13) up to an
remainder.
Proof. Apply Proposition 4 with
.
From (11),
the margin difference decomposes as When , the causal term vanishes. ◻
Proposition 5 identifies when
deployment performance becomes sensitive to spurious statistics: when
learned spurious parameters
interact with shifts in spurious statistics . To first order,
degradation occurs when this interaction has the harmful sign. When
spurious statistics remain stable (), spurious
learning is benign despite .
We now show that this population-level vulnerability is irreducible with
respect to sample size by decomposing deployment suboptimality.
5.3. Suboptimality
Decomposition
We now formalize the irreducibility of deployment degradation by
decomposing suboptimality into a shift term and an estimation term. Let
denote the deployment-optimal
parameters, where .
Let denote the
finite-sample estimator trained on samples from . We define deployment suboptimality as
(14) Inserting the population
training optimum yields: (15)Irreducible shift vs. reducible
estimation. The decomposition separates two sources of
suboptimality. When is
consistent for the population training optimum , the
estimation term vanishes as . In contrast, the shift term is determined by
the population optimum and persists
regardless of sample size: (16) By Proposition 5, when causal
statistics are stable, the persistent vulnerability is driven by the
interaction between and
.
The magnitude depends on two factors: the learned spurious parameters
(which Section 4 showed emerge structurally from
training) and the spurious shift (which
depends on the deployment environment). Collecting more data from cannot eliminate this vulnerability, it
only reduces the vanishing estimation component while leaving
unchanged.
5.4. Main Deployment Bound
The margin analysis in Section 5.2
characterized deployment degradation at the population level, assuming
access to .
In practice, we learn from finite samples, obtaining an estimator that deviates from . We now bound the
estimation term to complete the decomposition in
Equation (15),
confirming that the irreducible shift dominates as .
Technical assumptions. We require: (A1) bounded
features,
almost surely; (A2) bounded parameters, for all ; (A3) local regime
(Assumption 2); and (A4) geometry transfer,
meaning there exists such that for all , where is the positive
definite local curvature of the training objective at . Assumption (A4)
bounds deployment variation by training curvature and holds when and are not too different. See Appendix C for detailed
discussion of these conditions.
We define as the
ridge-regularized MLE: (17) where is the regularization
parameter.
Theorem 6 (Deployment sub-optimality bound).
Under assumptions (A1)–(A4), with probability at least :(18) where ,, and .
Proof sketch. The decomposition (15)
holds by definition. For the estimation term, we expand around . The linear term
is bounded by ,
while the quadratic term is controlled by Assumption (A4). Concentration
gives . Full details are in Appendix 20. ◻
Interpretation. Theorem 6 reveals the structure of
deployment suboptimality. The estimation term decreases with more
training data and vanishes in the large-sample limit (up to the regularized bias). The
shift term persists: even with infinite training data, deployment error
remains bounded by the gap between and .
By Proposition 5, when causal
statistics are stable, learned spurious parameters make the deployment
objective sensitive to shifted spurious statistics. This decomposition
clarifies two sources of vulnerability: Section 4 showed
that arises structurally from the training distribution, while the
deployment shift depends on the environment. Spurious learning
creates latent vulnerability; harm materializes only when
spurious statistics shift. Crucially, scaling training data cannot
eliminate this vulnerability, as it leaves
unchanged.
6. Tie Training Reduces Spurious
Reliance
Section 5 showed that deployment error
contains an irreducible shift term driven by learned spurious parameters
.
We now describe a simple data-level intervention that directly targets
this mechanism. We introduce tie training, a data augmentation
strategy that reduces spurious reliance by adding curvature selectively
in spurious directions. The approach constructs preference pairs with
equal utility but differing spurious features, assigns labels randomly,
and mixes these ties with standard preference data during training.
6.1. Tie Construction and
Training
Definition 7 (Tie pair). A tie pair is a tuple with equal utility: .
We construct ties so that causal features match while spurious
features differ:
and .
For each tie, we assign the winner-loser label uniformly at random.
Thus, ties enter the standard DPO loss as hard-labeled pairs and require
no change to the objective. The random orientation gives
and (19) This covariance structure ensures ties add
curvature only in spurious directions.
Mixed training. We train on a mixture with . Under the local regime
(Assumption 2), the linearized equilibrium
satisfies (20)
Since has
support only on spurious coordinates, mixed training increases the
spurious curvature while scaling the raw causal–spurious second-moment
by . This shrinks the
spurious component relative to strict-only training. Full derivation is
in Appendix D.4.
6.2. Main Result
We now formalize the effect of tie training on spurious reliance and
deployment performance.
Theorem 8 (Tie training reduces spurious reliance
and deployment shift). Under the conditions of Theorem 6, the tie construction above,
and a regularity condition made explicit in Appendix D.5:
(i) Spurious shrinkage. Let denote the strict-only
population optimizer and the
mixed-training optimizer. Then (21) where . The
inequality is strict when tie training adds curvature along active
spurious directions.
(ii) Shift reduction. If , the first-order
deployment shift is bounded by the strict-only worst-case envelope:
where . In the scalar case (), this strengthens to
(iii) Finite-sample bound. Let be trained on
samples from . With probability at
least ,(22) where
and .
The proof combines the equilibrium characterization (Equation (20)) with the deployment
bound framework; see Appendix 23.
Corollary 9 (Quantitative reduction under isotropic
ties). If and , then (23)
Interpretation. Theorem 8 shows
that tie training directly targets the irreducible shift by shrinking
spurious parameters through selective regularization. Part (i)
establishes spurious weight reduction at the population level. Part (ii)
bounds the deployment shift contribution, with pointwise reduction in
the scalar case. Part (iii) provides finite-sample guarantees.
Corollary 9 gives an explicit reduction as a
function of .
6.3. Practical Considerations
We discuss three practical aspects of tie training next. Soft
ties: Exact utility equality is not required; near-ties with
yield similar effects. Random
labeling: Using both and
with equal probability ensures ;
single-direction labeling injects bias. Tie selection:
Systematic tie construction at scale remains an open problem; our
experiments (Section 7) use
manual construction or simple heuristics based on known spurious
features.
7. Experiments
We validate three theoretical predictions: (i) preference
optimization learns spurious correlations (Theorem 3), (ii) deployment error under
spurious shift is irreducible (Theorem 6),
and (iii) tie training reduces this error (Corollary 9). We evaluate these claims
across three regimes of increasing realism: linear models, neural
networks, and large language models. Code is available at https://github.com/cmoyacal/tie-training.
7.1. Linear Models: Quantitative
Validation
Setup. We generate Gaussian features with causal and spurious dimensions. Preference
labels are generated by a Bradley–Terry logistic teacher. Spurious
features correlate with preferences under training distribution and shift at deployment . Full details are in Appendix F.1.
Results.Figure 1
validates all three theoretical predictions. Panel (a) shows that
learned spurious parameters match the closed-form
prediction of spurious parameters (Theorem 3). When
the local regime assumption weakens and our quantitative predictions are
no longer valid, we find empirically that including second-order
curvature terms restores agreement (bottom panel). Panel (b-top) plots
deployment suboptimality versus sample size . The estimation error decays as , while the shift term
plateaus, confirming irreducibility (Theorem 6). Panel (b-bottom) reports
the tie-induced reduction ratio
across mixing fractions .
Empirical curves match the theoretical reduction factor (Corollary 9) for and ratio .
(a)
(b)
Figure 1.
Linear theory validation. (a) Learned spurious
parameters norm against theoretical prediction (Theorem 3) (top); second-order corrections
restore agreement when local regime is violated (bottom).
(b) Deployment suboptimality decomposition: estimation
error decays as n → ∞, while
shift error persists, demonstrating irreducibility. As predicted by
Theorem 6, empirical deployment error
SubOptQ remains
bounded as the number of training samples increases (top); tie reduction
ratio satisfies the theoretical formula (Corollary 9) across the fraction of strict
preferences α and the spurious
variance ratio σ2/λ0 = 5
(bottom).
These results confirm the population equilibrium characterization
(Section 4), the irreducibility decomposition
(Section 5), and the tie training guarantees
(Section 6).
7.2. Neural Networks: Qualitative
Persistence
Neural networks hide the causal–spurious decomposition inside
nonlinear representations, violating linear assumptions. We test whether
the same mechanisms persist qualitatively.
Setup. We sample latent causal and spurious
features, apply a random nonlinear mixing , and train an
MLP scorer with DPO. The causal–spurious decomposition is hidden from
the model. Details are in Appendix F.2.
Proxy metrics. Since is not observable, we
measure spurious reliance using (i) the spurious gap, where the two terms denote
pairwise preference accuracy when spurious features align with or oppose
true utility, respectively, and (ii) adversarial accuracy, the
pairwise preference accuracy under reversed spurious correlation.
Results. Strict training learns spurious reliance,
yielding spurious gap
and adversarial accuracy . Tie training () reduces the spurious gap to
and improves
adversarial accuracy to , while preserving in-distribution accuracy. Figure 2 shows that spurious gap
decreases monotonically with tie mixing fraction , and tie training breaks the
adversarial accuracy plateau observed under strict training as sample
size increases. Consistent with Theorems 6
and 8, tie training has limited
visible effect at small , where
estimation error dominates, but its advantage emerges as estimation
error decays and shift error remains. Although the theory does not apply
exactly, the qualitative behavior matches the linear predictions.
Figure 2.
Neural network validation. Left: Spurious gap (accuracy
on aligned minus misaligned spurious conditions) decreases monotonically
with tie mixing fraction α.Right: Strict training exhibits a persistent
adversarial accuracy plateau despite increasing data; tie training (with
α = 0.75) breaks this plateau,
improving robustness from to
.
7.3. LLMs: Synthetic Hotel
Preferences
Dataset. We construct a controlled hotel preference
benchmark. Causal attributes affecting utility include price, distance,
and rating. Spurious attributes (not causally affecting utility) include
building age, renovation year, chain tier, lobby size, and employee
count. These correlate with utility during training but do not affect
true quality. Full details are in Appendix F.3.
Tie construction. Informative ties satisfy three
properties: (i) near-equal utility (), (ii) strong spurious contrast (maximal ), and (iii) random labels.
As an ablation, in Appendix F.3, we also evaluate
non-informative ties that use monotonic spurious assignment correlated
with utility, providing weak regularization signal.
Results. Standard DPO achieves in-distribution accuracy but
degrades to (suppressed
correlation) and (adversarial
correlation) under spurious shift. Informative tie training maintains
in-distribution while
improving to (suppressed) and
(adversarial). Table 1 reports the results.
Table 1. Informative tie training (Tie-I) improves robustness under
spurious shift without sacrificing in-distribution accuracy.
Method
In-Distr. ()
Sup. ()
Adv. ()
Strict
92.25%
74.00%
64.20%
Tie-I
92.40%
82.85%
86.70%
8. Discussion
On our assumptions. Our closed-form results on
spurious reliance (Section 4),
irreducible deployment error (Section 5), and
tie training (Section 6) hold in
the local regime (Assumption 2) for
log-linear policies. Within this regime, the analysis identifies mean
spurious bias and causal–spurious leakage as the mechanisms behind
spurious learning, links them to irreducible deployment error, and
motivates tie training as a targeted intervention.
Linear analysis is standard in the shortcut- and spurious-learning
literature for isolating shift-induced failure modes. For
example, (Nagarajan, Andreassen, and Neyshabur 2021) use linear models
to characterize spurious-feature failure under distribution shift in
classification. Here, we study the corresponding question for pairwise
preference optimization. To make the regime boundary explicit, we report
the diagnostic in our log-linear experiments
(Appendix F.1), confirming the
parameter ranges where the local approximation holds.
Beyond these assumptions, the same qualitative pattern appears in
neural network and LLM experiments outside the local regime: spurious
reliance emerges under correlated training data, additional
in-distribution data fails to remove the resulting deployment error, and
tie training reduces it. Thus, the theory remains useful as a diagnostic
guide for designing the intervention.
Finally, the finite-sample deployment result also invokes boundedness
and a transfer condition (Section 5.4). These
assumptions control finite-sample behavior under shift but do not change
the qualitative conclusion. The theory does not require the
Bradley–Terry (Bradley and Terry 1952) annotator model; it depends only
on feature-difference statistics.
On the controlled experimental design. Our
experiments are deliberately controlled: they separate causal and
spurious attributes, vary spurious statistics across training and
deployment, and keep the underlying preference structure fixed. This
control is necessary to directly test the paper’s central claims and is
generally unavailable in real preference data, where causal and spurious
factors are entangled. The method itself does not require explicit
identification of spurious factors. The construction is used only to
make the mechanism empirically testable. The spurious correlations we
study are also reported in deployed alignment systems (Casper et
al. 2023), so the isolated mechanisms remain practically relevant
despite the synthetic benchmark. Broader evaluation on natural
preference data remains an important next step.
One might expect the controlled construction to require idealized
ties. However, the ties in our LLM and nonlinear experiments are already
approximate: near-ties can preserve a weak causal margin while injecting
larger variation along spurious directions, showing that exact ties are
not essential for the mechanism. The local log-linear framework also
extends to this regime by replacing ideal tie statistics with
approximate ones. Appendix F.1 gives this
characterization and a log-linear near-tie sweep.
Tie construction and the role of . A limitation of the
method is that the theory does not prescribe how to construct ties.
However, tie training is not restricted to hand-crafted ties. The
mechanism suggests three practical regimes. (i) Known spurious
attribute: when the spurious factor is identified, as in our LLM
experiments, ties can be constructed by direct perturbation. (ii)
Candidate axes: when the spurious direction is unknown but
plausible candidates can be listed (e.g., verbosity, style, formatting),
ties are constructed by content-preserving perturbations across these
axes. (iii) Fully agnostic: cross-prompt constructions form
tie-like pairs that can perturb many spurious directions at once.
This data-level view connects to RRM (T. Liu et al. 2025), which
addresses spurious correlations in reward learning through a causal
framework and feature-agnostic augmentation. RRM targets reward-model
training with soft labels for neutral pairs, whereas we use symmetric
hard-labeled pairs in the unmodified DPO loss as a direct consequence of
the mechanism we analyze. We view RRM’s augmentation as introducing
preference-neutral signal at the data level, complementary to ours.
The parameter controls
the strict–tie mixture: decreasing increases tie mass, and
Corollary 9 establishes monotone spurious
shrinkage as tie mass grows. In practice, a moderate tie budget can
induce spurious-direction regularization. We used an unoptimized in the LLM experiment, with a
full -sweep reported in
Appendix F.3. Tie effectiveness also
depends on which pairs are added. Suppression scales with the alignment
between ties and , so
under a fixed budget, selecting informative pairs can yield greater
suppression than uniformly increasing tie mass. Appendix F.1 further examines
the -dependence of causal
components. Two problems remain open: automatic tie construction and
scalable allocation of a fixed tie budget toward high-impact
directions.
9. Conclusion
This work analyzes spurious correlation learning in preference
optimization and establishes three main results. First, under standard
preference objectives, spurious correlations can be learned at the
population level through mean spurious bias and causal–spurious
correlation leakage (Theorem 3). Second,
when spurious statistics shift between training and deployment, such
reliance can induce deployment degradation that persists even with
unlimited training data drawn from the same distribution (Theorem 6). Third, tie training, a
simple data augmentation strategy using equal-utility preference pairs,
can reduce spurious reliance by selectively regularizing spurious
directions (Theorem 8).
Together, these results suggest that robustness to spurious
correlations in preference optimization depends not only on model
capacity or data scale, but also on the structure of supervision. Tie
training provides one concrete instance of this approach within standard
preference objectives. Our analysis focuses on a local regime and
assumes access to informative ties, enabling tractable characterization;
extending these results beyond this regime and developing scalable tie
construction methods remain important directions for future work.
Further discussion appears in Appendix H.
Impact statement
This paper presents work whose goal is to advance the field of
machine learning. There are many potential societal consequences of our
work, none which we feel must be specifically highlighted here.
Acknowledgments
We thank the anonymous reviewers for their helpful comments. We would
also like to thank the Future Impact Group for their help in initiating
this project. GL would like to thank the support of National Science
Foundation (DMS-2533878, DMS-2053746, DMS-2134209, ECCS-2328241,
CBET-2347401 and OAC-2311848), and U.S. Department of Energy (DOE)
Office of Science Advanced Scientific Computing Research program
DE-SC0023161, the SciDAC LEADS Institute, and DOE–Fusion Energy Science,
under grant number: DE-SC0024583.
References
Arjovsky, Martin, Léon Bottou, Ishaan Gulrajani, and David Lopez-Paz.
2019. “Invariant Risk Minimization.” arXiv Preprint
arXiv:1907.02893.
Bombari, Simone, and Marco Mondelli. 2025. “Spurious Correlations in
High Dimensional Regression: The Roles of Regularization, Simplicity
Bias and over-Parameterization.” In Proceedings of the 42nd
International Conference on Machine Learning, 267:4839–73.
Proceedings of Machine Learning Research. PMLR.
Bradley, Ralph Allan, and Milton E Terry. 1952. “Rank Analysis of
Incomplete Block Designs: I. The Method of Paired Comparisons.”
Biometrika 39 (3/4): 324–45.
Chang, Chun-Hao, George Alexandru Adam, and Anna Goldenberg. 2021.
“Towards Robust Classification Model by Counterfactual and Invariant
Data Generation.” In Proceedings of the IEEE/CVF Conference on
Computer Vision and Pattern Recognition, 15212–21.
Chen, Lichang, Shiyang Li, Jun Yan, Hai Wang, Kalpa Gunaratna, Vikas
Yadav, Zheng Tang, et al. 2024. “AlpaGasus: Training a Better Alpaca
with Fewer Data.” In The Twelfth International Conference on
Learning Representations.
Chowdhury, Sayak Ray, Xingyu Zhou, and Nagarajan Natarajan. 2024.
“Differentially Private Reward Estimation with Preference Feedback.” In
International Conference on Artificial Intelligence and
Statistics, 4843–51. PMLR.
Christiano, Paul F, Jan Leike, Tom Brown, Miljan Martic, Shane Legg, and
Dario Amodei. 2017. “Deep Reinforcement Learning from Human
Preferences.” Advances in Neural Information Processing Systems
30.
Dudík, Miroslav, Katja Hofmann, Robert E. Schapire, Aleksandrs Slivkins,
and Masrour Zoghi. 2015. “Contextual Dueling Bandits.” In
Proceedings of the 28th Conference on Learning Theory,
40:563–87. Proceedings of Machine Learning Research. Paris, France:
PMLR.
Geirhos, Robert, Jörn-Henrik Jacobsen, Claudio Michaelis, Richard Zemel,
Wieland Brendel, Matthias Bethge, and Felix A Wichmann. 2020. “Shortcut
Learning in Deep Neural Networks.” Nature Machine Intelligence
2 (11): 665–73.
Geirhos, Robert, Patricia Rubisch, Claudio Michaelis, Matthias Bethge,
Felix A Wichmann, and Wieland Brendel. 2018. “ImageNet-Trained CNNs Are
Biased Towards Texture; Increasing Shape Bias Improves Accuracy and
Robustness.” In International Conference on Learning
Representations.
Haldar, Rajdeep, Ziyi Wang, Guang Lin, Yue Xing, and Qifan Song. 2025.
“LLM Safety Alignment Is Divergence Estimation in Disguise.” In
Advances in Neural Information Processing Systems, 38:94010–40.
Hermann, Katherine, Hossein Mobahi, Thomas Fel, and Michael Curtis
Mozer. 2024. “On the Foundations of Shortcut Learning.” In The
Twelfth International Conference on Learning Representations.
Hu, Edward J, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li,
Shean Wang, Lu Wang, and Weizhu Chen. 2022. “LoRA: Low-Rank Adaptation
of Large Language Models.” In International Conference on Learning
Representations.
Izmailov, Pavel, Polina Kirichenko, Nate Gruver, and Andrew G Wilson.
2022. “On Feature Learning in the Presence of Spurious Correlations.”
Advances in Neural Information Processing Systems 35: 38516–32.
Kalimeris, Dimitris, Gal Kaplun, Preetum Nakkiran, Benjamin Edelman,
Tristan Yang, Boaz Barak, and Haofeng Zhang. 2019. “Sgd on Neural
Networks Learns Functions of Increasing Complexity.” Advances in
Neural Information Processing Systems 32.
Kirichenko, Polina, Pavel Izmailov, and Andrew Gordon Wilson. 2023.
“Last Layer Re-Training Is Sufficient for Robustness to Spurious
Correlations.” In The Eleventh International Conference on Learning
Representations.
Langosco, Lauro, Jack Koch, Lee D Sharkey, Jacob Pfau, and David
Krueger. 2022. “Goal Misgeneralization in Deep Reinforcement Learning.”
In International Conference on Machine Learning, 12004–19.
PMLR.
Liu, Evan Z, Behzad Haghgoo, Annie S Chen, Aditi Raghunathan, Pang Wei
Koh, Shiori Sagawa, Percy Liang, and Chelsea Finn. 2021. “Just Train
Twice: Improving Group Robustness Without Training Group Information.”
In International Conference on Machine Learning, 6781–92. PMLR.
Liu, Tianqi, Wei Xiong, Jie Ren, Lichang Chen, Junru Wu, Rishabh Joshi,
Yang Gao, et al. 2025. “RRM: Robust Reward Model Training Mitigates
Reward Hacking.” In The Thirteenth International Conference on
Learning Representations.
Morwani, Depen, Jatin Batra, Prateek Jain, and Praneeth Netrapalli.
2023. “Simplicity Bias in 1-Hidden Layer Neural Networks.” Advances
in Neural Information Processing Systems 36: 8048–75.
Nagarajan, Vaishnavh, Anders Andreassen, and Behnam Neyshabur. 2021.
“Understanding the Failure Modes of Out-of-Distribution Generalization.”
In International Conference on Learning Representations.
Ngo, Richard, Lawrence Chan, and Sören Mindermann. 2024. “The Alignment
Problem from a Deep Learning Perspective.” In The Twelfth
International Conference on Learning Representations.
Ouyang, Long, Jeffrey Wu, Xu Jiang, Diogo Almeida, Carroll Wainwright,
Pamela Mishkin, Chong Zhang, et al. 2022. “Training Language Models to
Follow Instructions with Human Feedback.” Advances in Neural
Information Processing Systems 35: 27730–44.
Park, Ryan, Rafael Rafailov, Stefano Ermon, and Chelsea Finn. 2024.
“Disentangling Length from Quality in Direct Preference Optimization.”
In Findings of the Association for Computational Linguistics: ACL
2024, 4998–5017.
Pezeshki, Mohammad, Oumar Kaba, Yoshua Bengio, Aaron C Courville, Doina
Precup, and Guillaume Lajoie. 2021. “Gradient Starvation: A Learning
Proclivity in Neural Networks.” Advances in Neural Information
Processing Systems 34: 1256–72.
Plumb, Gregory, Marco Tulio Ribeiro, and Ameet Talwalkar. 2022. “Finding
and Fixing Spurious Patterns with Explanations.” Transactions on
Machine Learning Research.
Rafailov, Rafael, Archit Sharma, Eric Mitchell, Christopher D Manning,
Stefano Ermon, and Chelsea Finn. 2023. “Direct Preference Optimization:
Your Language Model Is Secretly a Reward Model.” Advances in Neural
Information Processing Systems 36: 53728–41.
Rahaman, Nasim, Aristide Baratin, Devansh Arpit, Felix Draxler, Min Lin,
Fred Hamprecht, Yoshua Bengio, and Aaron Courville. 2019. “On the
Spectral Bias of Neural Networks.” In International Conference on
Machine Learning, 5301–10. PMLR.
Sagawa, Shiori, Aditi Raghunathan, Pang Wei Koh, and Percy Liang. 2020.
“An Investigation of Why Overparameterization Exacerbates Spurious
Correlations.” In International Conference on Machine Learning,
8346–56. PMLR.
Saha, Aadirupa, Aldo Pacchiano, and Jonathan Lee. 2023. “Dueling Rl:
Reinforcement Learning with Trajectory Preferences.” In
International Conference on Artificial Intelligence and
Statistics, 6263–89. PMLR.
Shah, Harshay, Kaustav Tamuly, Aditi Raghunathan, Prateek Jain, and
Praneeth Netrapalli. 2020. “The Pitfalls of Simplicity Bias in Neural
Networks.” Advances in Neural Information Processing Systems
33: 9573–85.
Sharma, Mrinank, Meg Tong, Tomasz Korbak, David Duvenaud, Amanda Askell,
Samuel R. Bowman, Newton Cheng, et al. 2024. “Towards Understanding
Sycophancy in Language Models.” In The Twelfth International
Conference on Learning Representations.
Singhal, Prasann, Tanya Goyal, Jiacheng Xu, and Greg Durrett. 2024. “A
Long Way to Go: Investigating Length Correlations in RLHF.” In First
Conference on Language Modeling.
Singla, Sahil, and Soheil Feizi. 2022. “Salient ImageNet: How to
Discover Spurious Features in Deep Learning?” In International
Conference on Learning Representations.
Skalse, Joar, Nikolaus Howe, Dmitrii Krasheninnikov, and David Krueger.
2022. “Defining and Characterizing Reward Gaming.” Advances in
Neural Information Processing Systems 35: 9460–71.
Wu, Junkang, Yuexiang Xie, Zhengyi Yang, Jiancan Wu, Jiawei Chen,
Jinyang Gao, Bolin Ding, Xiang Wang, and Xiangnan He. 2025. “Towards
Robust Alignment of Language Models: Distributionally Robustifying
Direct Preference Optimization.” In The Thirteenth International
Conference on Learning Representations.
Xiao, Kai Yuanqing, Logan Engstrom, Andrew Ilyas, and Aleksander Madry.
2021. “Noise or Signal: The Role of Image Backgrounds in Object
Recognition.” In International Conference on Learning
Representations.
Xiong, Wei, Hanze Dong, Chenlu Ye, Ziqi Wang, Han Zhong, Heng Ji, Nan
Jiang, and Tong Zhang. 2024. “Iterative Preference Learning from Human
Feedback: Bridging Theory and Practice for RLHF Under KL-Constraint.” In
Proceedings of the 41st International Conference on Machine
Learning, 235:54715–54. Proceedings of Machine Learning Research.
PMLR.
Zhang, Xuanchang, Wei Xiong, Lichang Chen, Tianyi Zhou, Heng Huang, and
Tong Zhang. 2025. “From Lists to Emojis: How Format Bias Affects Model
Alignment.” In Proceedings of the 63rd Annual Meeting of the
Association for Computational Linguistics (Volume 1: Long Papers),
26940–61.
Zhou, Chunting, Xuezhe Ma, Paul Michel, and Graham Neubig. 2021.
“Examining and Combating Spurious Features Under Distribution Shift.” In
International Conference on Machine Learning, 12857–67. PMLR.
Zhou, Xingyu, Yulian Wu, and Francesco Orabona. 2025. “A Unified
Theoretical Analysis of Private and Robust Offline Alignment: From RLHF
to DPO.” In Forty-Second International Conference on Machine
Learning.
Zhu, Banghua, Michael Jordan, and Jiantao Jiao. 2023. “Principled
Reinforcement Learning with Human Feedback from Pairwise or k-Wise
Comparisons.” In International Conference on Machine Learning,
43037–67. PMLR.
Ziegler, Daniel M, Nisan Stiennon, Jeffrey Wu, Tom B Brown, Alec
Radford, Dario Amodei, Paul Christiano, and Geoffrey Irving. 2019.
“Fine-Tuning Language Models from Human Preferences.” arXiv Preprint
arXiv:1909.08593.
A. Full Proof of Spurious Learning
Mechanism (Theorem 3)
This section provides the full proof of Theorem 4.1 by progressively
isolating the mechanisms that drive spurious learning in direct
preference optimization for log-linear policies. We first restate the
population objective and derive its exact gradient, then linearize the
dynamics under the local regime assumption. We next decompose the
resulting covariance structure into causal and spurious blocks and solve
the corresponding equilibrium in closed form. Finally, we interpret the
solution to separate and quantify the mean-bias and correlation-leakage
contributions, completing the proof.
Our analysis relies on the local linearization assumption introduced
in Section 3 (Assumption 2), which we restate below for
completeness.
Assumption 10 (Local regime). We assume that the
scaled DPO margin satisfies (24) with high probability under the data distribution.
This regime arises in several settings. First, during early training
the policy remains close to the reference policy. Second, the margin is
small when competing responses receive similar scores under the current
policy. Finally, the regime naturally arises in regions of
near-indifference. In these settings, the DPO objective admits a
first-order linear approximation, enabling analytical characterization
of the training dynamics.
A.1. Population Objective and
Gradient
We study the population objective: (25)
where
and is the feature difference vector.
A.1.1. Full DPO pairwise
gradient.
It follows from the chain rule and . In particular, (26) Defining the adaptive weight , where , and applying the chain rule with ,
the gradient is: (27)
A.1.2. Linearized population
gradient.
Taking the expectation of the per-example gradient, (28) We decompose
the expectation of this product via the vector–scalar covariance
identity, (29) and
linearize each term in turn (Steps 1–2) before recombining (Step 3).
Under the local regime (Assumption 10), we derive the
linearized population gradient next. To this end, let us first define
the feature second moment matrix: (30) This is related to the covariance matrix by where . We use
second moments (rather than centered covariances) because they lead to
simpler final formulas.
A.1.3. Step 1: We linearize the
expected weight:
(31) This approximation
follows from the first-order Taylor expansion of the sigmoid around
zero: , which is valid under Assumption 10.
A.1.4. Step 2: We compute the
covariance term as follows:
(32) Using the
identity for covariance between a vector and a linear form yields: (33) Therefore: (34)
A.1.5. Step 3: We combine the
terms:
(35) The terms
and
cancel exactly (this cancellation is why using (rather than ) gives a clean final formula). Thus,
the linearized population gradient takes the form (36) accurate up to third-order terms in the scaled
margin.
A.2. Solving the Equilibrium
We now derive the explicit equilibrium expressions for the causal and
spurious components of the linearized population optimum stated in
Section 4.
Assumption 11 (Spurious-Causal Feature
Decomposition). We decompose the feature map into causal and spurious
components, i.e., (37) where contains
causal features and contains spurious features. This
induces the same decomposition for parameters and pairwise differences:
,.
A.2.1. Setup.
Using the linearized population gradient (36),
the equilibrium condition yields (38) where .
Writing ,,
and (39) the equilibrium equations decompose as (40)
A.2.2. Solving for the spurious
component.
Assume is
invertible. From (40), (41) Substituting into (40) gives (42)
A.2.3. Schur complement.
Define the Schur complement: (43) Assuming
is invertible, the causal
component satisfies (44)
A.2.4. Explicit spurious
equilibrium.
Substituting the expression for back yields (45) This completes the derivation of the explicit
spurious equilibrium reported in Section 4.
A.3. Interpretation of the
Equilibrium Solution
We now interpret the equilibrium solution derived above and state our
main result characterizing spurious learning at the population level.
The following theorem formalizes when and why the population optimizer
assigns nonzero weight to spurious features under the linearized
dynamics.
Theorem 12 (Spurious learning from mean bias and
correlation leakage). Under the linearized population dynamics
(Equation (36)),
assume (equivalently, and ). If either
(mean spurious bias) or (causal–spurious leakage), then
genericallyOutside a Lebesgue measure-zero set of problem
instances. the population optimum satisfies .
Proof. The linearized population gradient takes the form
(see Appendix A.1 for
details) (46) so the equilibrium
satisfies ,
i.e. (47) Using the spurious-causal feature
decomposition (Assumption 11), we can write the
equilibrium condition in block form as (48) Assume and so the solution is unique.
Solving by Schur complement (see Appendix A.2 for details)
yields, (49)
Substituting into (48)
yields (50) This proves the claimed decomposition into a
component driven by and a
component driven by . It
remains to show: if either or ,
then generically .
Case 1: . The mean-bias term in (50)
equals with The
matrix is invertible: indeed,
is similar to
whose second term is positive semidefinite. Hence when . Since also contains the
leakage term, can occur only
through exact cancellation, This
equality defines a proper affine constraint on the problem instance.
Thus, generically .
Case 2: but
. Then
.
Substituting into (48)
gives (51) If
, then for generic
this expression is nonzero
(failing only when lies in
the nullspace of , again a measure-zero
condition). Thus, generically .
Combining the two cases establishes the theorem. ◻
B. Deployment Error and
Margin-Based Analysis
This appendix analyzes how spurious correlations learned during
training induce vulnerabilities at deployment. Let denote the training preference
distribution over triples , and let denote the deployment preference
distribution over the same space. We write (52) for the population optimizer of the
log-linear DPO surrogate under ,
i.e., .
The learned parameters are fixed after training; deployment changes the
distribution from to , not the parameters.
Our analysis operates directly on the log-linear DPO surrogate and
the induced margins. In the local regime, the surrogate admits a
first-order expansion in terms of the expected preference margin, which
provides a transparent characterization of deployment degradation under
distribution shift.
C. Assumptions for Deployment
Analysis
We start by providing detailed discussion of the assumptions used in
the deployment suboptimality analysis.
Assumption 13 (Bounded feature differences). We
assume that feature differences are uniformly bounded: (53)
This is a normalization convention rather than a substantive
restriction. Specifically, if the feature map satisfies (54) we rescale and reparametrize
,
preserving all margins: . In practice, the assumption holds whenever
features are normalized (e.g., embeddings projected onto the unit
sphere), bounded by construction (e.g., indicator features, clipped
activations), or supported on a compact set.
Assumption 14 (Bounded parameters). The deviation
from the reference parameters is bounded: (55)
This assumption restricts optimization to a compact parameter set and
ensures that preference score differences
remain bounded. The bound may be
enforced explicitly (e.g., via constrained optimization) or implicitly
(e.g., via early stopping).
C.1. Joint implication.
Under Assumptions 13 and 14,
preference score differences satisfy (56) Thus the local regime (Assumption 15) holds uniformly
over whenever .
The local regime assumption is used in two places.
It justifies the first-order expansion of the surrogate in terms
of the margin (Proposition 17).
It ensures that the pairwise Fisher curvature remains close to a
constant multiple of the unweighted second moment, enabling the geometry
transfer (Assumption 16).
The local regime is also naturally satisfied for tie and near-tie
pairs, since these pairs have small margins by construction.
C.3. Quantitative scale.
The linearization has absolute error below for and below for
. The approximation is
accurate when .
C.4. Beyond the local
regime.
When , the weight
function varies strongly with the margin and
saturates toward zero for large . In that case, second-order
expansions around no longer provide
uniform control over the objective, and first-order margin
approximations remain qualitatively informative but the quantitative
bounds derived below require modification.
C.5. Deployment curvature.
Let for
any positive semi-definite matrix . Define the (pairwise) Fisher
information under deployment ,
evaluated at :(58)
Define analogously by replacing with . Let for regularization parameter
.
Informally, in the local regime, , so (59)
Thus, locally, the deployment Fisher is controlled by the same
second-moment structure that appears in the population gradient analysis
(Appendix A.1).
Assumption 16 (Geometry transfer). There exists
such that for all
,(60)
This assumption controls how much the local curvature of the loss can
change between training and deployment. Without it, the estimation error
measured under would not transfer
to : the Hessian under could be arbitrarily larger, making the
quadratic bound on the estimation term vacuous.
C.6. Sufficient conditions for
geometry transfer.
Assumption 16 holds with
an explicit constant
under any of the following conditions.
Condition 1 (Direct curvature domination). If there
exists such that (61) then Assumption 16 holds with
.
Condition 2 (Bounded density ratio). If and almost
surely, then by change of measure: (62) which bounds the worst-case reweighting of
preference pairs under . To obtain
Assumption 16 in the
form above, one applies the same argument to the raw second moment,
, together with a
curvature lower bound as supplied by Condition 3. The resulting constant
is .
Condition 3 (Covariance transfer under local
regime). Assume the local regime holds uniformly, so that and therefore , where . If the feature covariances satisfy (63) for some , then (64) Moreover,
since
uniformly on ,(65) If additionally (which holds
in particular whenever ,
since then ), then (66) where the
last step uses . Combining, (67) so, provided ,
Assumption 16 holds with
.
C.1. Deployment Proxy: Expected
Margins
We now characterize deployment degradation through the expected
preference margin.
C.1.1. Margin definition.
The expected margin under distribution measures the average score gap between
preferred and dispreferred responses: The margin decomposes
into causal and spurious components: (68)
where and .
Under the training distribution , the spurious term may contribute
positively to the learned classifier. Under deployment , this term may shrink, flip sign, or
become pure noise, depending on the shift scenario (Section E).
C.1.2. Margin approximation of
objective differences.
We now restate Proposition 4 and
provide the margin approximation proof.
Proposition 17 (First-order margin approximation).
Under the local regime (Assumption 15), the pairwise
surrogate objective
satisfies (69)
Proof. Let . Under Assumption 15, we have with high
probability. Since is
, a Taylor expansion at gives (70) because and
hence .
Taking expectations under
yields (71) Under Assumption 13, , so (72) and therefore (73) Applying this with and and subtracting cancels the constant
, giving (74) as claimed. ◻
We now show that spurious margin drives shift under stable causal
statistics.
Proposition 18 (Spurious Margin Drives Shift).
When causal statistics are stable (), the objective
difference at is determined by
the spurious margin: (75)
Proof. Apply Proposition 17 with . By (68):
(76) When , the causal term vanishes: (77) Substituting into Proposition 17 yields the
result. ◻
C.1.3. Interpretation.
Deployment degradation is driven by the spurious margin because the
training optimizer was
optimized to exploit the spurious correlation under . When the deployment distribution breaks this correlation, the same
parameter becomes a source of vulnerability. The magnitude of
degradation is controlled by ,
the same quantity that tie training reduces (see Appendix D).
C.2. Deployment Suboptimality
Definition
Let (78) denote the
deployment-optimal parameters, where . Let denote the empirical
estimator trained on samples from
. Define the deployment
suboptimality: (79)
C.2.1. Decomposition.
Inserting the population training optimizer :(80)
C.2.2. Irreducible and reducible
components.
The shift term depends on and does not
depend on sample size. As , and the estimation term vanishes (as
demonstrated in Appendix C.3): (81) Proposition 18 shows that,
when causal statistics are stable, the deployment objective evaluated at
is
first-order sensitive to .
Thus, when the -optimal parameter
removes or reverses this spurious reliance, the persistent shift term is
driven by the same learned spurious component. The estimation term
vanishes as , so
additional samples from do not
remove this structural bottleneck. The fundamental bottleneck is
spurious component , which
arises structurally from the training distribution (Section 4).
C.3. Proof Deployment Suboptimality
Bound
This appendix provides the proof of the deployment suboptimality
bound (Theorem 6). For simplicity, we state
the following high-probability concentration bound as an assumption. It
follows from the self-normalized MLE analysis of (Zhu, Jordan, and Jiao
2023), specialized to our regularized estimator and bounded-feature
setting.
Assumption 19 (Training-estimator
concentration (Zhu, Jordan, and Jiao 2023)). With probability at least
,(82) where (83)
This is a standard self-normalized concentration bound for
regularized generalized linear estimators (Zhu, Jordan, and Jiao 2023);
in our setting, plays the role of the regularized Fisher matrix. The
first term captures statistical error and vanishes as ; the second reflects
regularization bias and vanishes as .
Theorem 20 (Deployment sub-optimality bound).
Under Assumptions 13, 14, 15, 16, and 19, with probability
at least :(84) where .
The estimation term decomposes into a first-order contribution
controlled by the deployment gradient norm and a second-order contribution
controlled by the geometry transfer constant . The gradient norm measures
how far is from being
optimal under : when , this quantity is small near ; under
distribution shift, it captures the first-order cost of deploying a
-optimal estimator under .
Proof. Let denote the
(population) optimizer of the training surrogate under over , and let denote the (population)
optimizer under . We organize the
proof into six steps.
C.3.1. Step 1
(Decomposition).
From (80):
(85) It remains to
bound the estimation term .
C.3.2. Step 2 (Smoothness upper
bound).
For each , the
negative Hessian of
is (86)
Since for all ,(87) Hence, by the standard
smoothness inequality for concave functions, applied around ,(88)
Rearranging gives (89)
C.3.3. Step 3 (Bound the linear
term).
By Cauchy–Schwarz in the Euclidean inner product: (90) This uses the
identity , which holds for any positive
definite without requiring
geometry transfer.
C.3.4. Step 4 (Geometry transfer
for the quadratic term).
By Assumption 19, with probability
at least :(92)
C.3.6. Step 6 (Combine).
Substituting Steps 3–5 into (89):
(93) Inserting into the
decomposition from Step 1 yields the stated bound. ◻
C.3.7. Interpretation.
The bound reveals three distinct sources of deployment
suboptimality:
Shift term: The irreducible gap between the
-optimal and -optimal parameters, driven by when causal statistics are stable
(Proposition 18).
Gradient term: The first-order cost of deploying
a -trained estimator under . This quantity measures how far the population
training optimizer is from
stationarity under . It is small
when near and grows with
distribution shift.
Quadratic term: The second-order estimation
penalty, controlled by the geometry transfer constant and the statistical rate .
Tie training (Appendix D)
reduces the shift term by suppressing
and simultaneously reduces the gradient term by making the estimator
less sensitive to spurious distributional changes.
D. Tie Training Theory
Theorem 12 identifies two drivers of
spurious learning: spurious mean bias and causal–spurious leakage . Both are properties
of the training distribution, so the natural remedy is a data-level
intervention. Pairs with matched causal features (), varying spurious
features (), and
zero spurious mean bias () dilute the effective contribution of both drivers under
mixture training. This section formalizes this construction as tie
training, derives the resulting mixed-training equilibrium, proves
that tie training reduces the spurious parameter magnitude (Theorem 23), with the reduction
being unconditional in the scalar spurious case (), and derives a quantitative
reduction bound (Corollary 24) that sharpens the
main theorem.
D.1. Tie Construction Details
We formalize the tie data-generating process and labeling scheme. We
start by defining a tie pair:
Definition 21 (Tie pair). A tie pair is a tuple
with equal utility,
.
D.1.1. Feature structure.
We construct tie pairs so that causal features match while spurious
features differ: and . Thus, ties vary
only in spurious coordinates.
D.1.2. Random labeling.
For each tie pair, we assign the winner-loser label uniformly at
random. With probability , we
label and set ; with probability
, we label and set . The zero-mean
property
follows from the symmetric random labeling, not from any assumption on
the distribution of . Then
(94)
D.1.3. Example.
For hotel recommendations: =
“Excellent service, prime location” (concise); = “Excellent service, prime location,
with detailed amenities list” (verbose). Both have equal utility but
differ in length, a common spurious feature.
D.2. Expected Gradient from
Ties
We now derive the expected gradient contribution of tie examples.
Under Assumption 10, the linearized
weight function is: (95)
D.2.1. Expected tie
gradient.
Then, for a single tie example, the DPO gradient is (96) Taking expectation over random tie
labeling and using
gives (97) The second equality uses , so . Under the
local regime (Assumption 10), this gives (98)
D.3. Linearized Tie Dynamics
We now characterize the structural effect of ties on the population
gradient.
D.3.1. Tie covariance.
Let us define the tie covariance matrix as (99) By the tie construction
(Appendix D.1), its block
structure is (100) We assume .
When , tie training provides curvature in all spurious directions.
In general, may be
low-rank, in which case the additional suppression from ties acts only
along the subspace spanned by the tie perturbations, while the remaining
spurious directions are regularized by the strict preference data.
D.3.2. Interpretation.
To first order, the tie gradient acts as a
data-dependent quadratic regularizer on the spurious parameters while
leaving causal parameters unaffected. The regularization strength is
controlled by , which is a
design choice determined by the tie distribution.
D.4. Mixed Training
Equilibrium
We now solve for the equilibrium parameters under mixed strict–tie
training. We start by defining the data model.
D.4.1. Mixed data model.
We consider training on a mixture of strict preference pairs from
distribution and tie pairs from
. Let denote the fraction of
strict preferences, and define the mixed distribution (101) All expectations below are taken
with respect to
unless stated otherwise.
D.4.2. Population gradient under
mixed training.
Under the above data model and local regime (Assumption 10), the population
gradient decomposes linearly: (102) Using the local expansions
(103) we obtain
(104) where the
effective covariance is (105) Since and , the
effective covariance satisfies for all
.
D.4.3. Structure of the effective
covariance.
Writing the block decomposition of the covariance (106) and using the tie construction (),
we obtain (107) Thus tie training
scales the raw causal–spurious second-moment blocks by and adds positive semidefinite
mass to the spurious block.
D.4.4. Population
equilibrium.
At the population equilibrium, we have , yielding (108)
and therefore (109)
D.4.5. Spurious block.
Applying the same Schur complement algebra as in Appendix A.2 to the mixed
system , the spurious component
satisfies (110) where .
D.4.6. Effect on the causal
block.
Tie training does not introduce direct causal bias, since the tie
construction contributes neither causal mean nor causal–spurious
cross-moments. The effect on the causal block is indirect through the
Schur complement.
Equipped with the mixed-training equilibrium (109)
derived in Appendix D.4, we now provide the
full proof of Theorem 8. To
this end, we use the following regularity assumption on the interaction
between the tie covariance and the spurious driving term.
Assumption 22 (Spurious driving term stability). The
spurious driving term satisfies (111) where and .
This regularity condition requires that the contraction of the raw
causal–spurious block
is not offset by amplification through the Schur complement and
spurious-block inverses. In the scalar spurious case (), the theorem below gives the
stronger pointwise shift reduction whenever the corresponding scalar
monotonicity condition
holds. This condition is immediate in pure mean-bias settings with , where ; in
pure-leakage and mixed regimes it is the scalar instance of
Assumption 22. For general , Assumption 22 rules out adversarial
tie designs that increase the effective spurious driving term despite
adding spurious curvature.
Theorem 23 (Tie training reduces spurious reliance
and deployment shift). Assume the conditions of Theorem 6, the tie construction in
Appendix D.1, and Assumption 22. Then:
(i) Spurious shrinkage (population level). Let
denote the strict-only
population optimizer under , and
the
mixed-training optimizer under . Then (112) where
and
denotes the weighted norm. The -norm measures the
contribution of spurious parameters to the variance of implicit reward
margins; the theorem thus guarantees that tie training reduces the
reward-variance attributable to spurious features. The inequality is
strict whenever tie curvature acts on the mixed spurious driving
direction and the driving-term stability bound is not tight.
(ii) Shift reduction under stable causal
statistics. Suppose and let .
**Scalar case (). If the scalar driving term is monotone under mixing,
i.e., (113) then the first-order deployment shift satisfies
(114) up to the
same local-regime remainder .*
**General case (). The first-order spurious shift contribution satisfies
(115) and the right-hand
side is controlled by the same-matrix norm shrinkage established in the
proof below.*
(iii) Deployment bound. Let be the
ridge-regularized estimator trained on with samples. Then, with probability at
least ,(116)
Proof. Let be the mixed training
distribution, and let denote the
corresponding population optimizer (equilibrium) in the local regime.
Let denote the
strict-only population optimizer under .
We use the mixed linearized equilibrium derived in Appendix D.4: (117)
and the strict-only equilibrium (118) From the Schur
complement decomposition (Appendix D.4), the spurious
components satisfy (119) where
and .
D.5.1. Proof of Part (i):
spurious weight reduction.
We first establish that (123) Let and . From the
equilibrium expression: (124)
Since , we have
.
Applying congruence by :(125) so (126) where the last step uses
from the PSD inverse bound in Part (i). Substituting: (127)
By Assumption 22: (128) so (129)
establishing (123).
D.5.8. Scalar case ().
When , all spurious
quantities are scalar. From the equilibrium expressions: (130) Since
:(131)
Under the condition :(132) The shift inequality follows
immediately: (133)
D.5.9. General case ().
For general , Cauchy–Schwarz
in the -inner
product gives (134) By the same-matrix
norm reduction (123):
(135) This bounds the
mixed-training shift by the strict-only worst-case envelope: the
right-hand side equals the maximum of over all
deployment shifts with , so tie training
guarantees the mixed model stays within this envelope.
D.5.10. Proof of Part (iii):
deployment bound.
Apply Theorem 20 with the training
distribution replaced by and the corresponding
population optimizer and ridge-MLE
. The same
proof yields, with probability ,(136)
where is
the same self-normalized radius but computed with the mixed-training
curvature (and
the same ). This
proves Part (iii). ◻
D.6. Quantitative Reduction
Bound
We conclude this appendix by providing the full proof of Corollary 9. For completeness, we restate
the corollary below and then provide the proof.
Corollary 24 (Quantitative reduction under isotropic
ties). Under the conditions of Theorem 23, if ties are
isotropic, i.e., , and , then (137)
Proof. Under isotropy, (138) so (139) Hence (140)
From the equilibrium expressions, (141) Therefore (142) Under
Assumption 22, which reduces to
Euclidean contraction in the isotropic case, (143) yielding the claimed bound. ◻
E. Distribution Shift
Scenarios
This appendix provides detailed analysis of the three shift scenarios
from Section 5. Throughout, we assume
and stable causal statistics ().
E.1. Suppression ()
Spurious correlation absent at deployment while .
E.1.1. Margin behavior.
.
Systematic spurious bias vanishes.
E.1.2. Variance-induced accuracy
degradation.
Zero mean does not imply robustness. The spurious variance (144) adds
noise to predictions.
E.1.3. Examples.
Hotels: Training has longer reviews for quality
hotels; test set balanced length adds noise.
Code: Training has verbose correct solutions;
test balanced concise
correct solutions underrated.
Safety: Training has hedged safe responses; test
balanced direct safe
responses underrated.
E.2. Adversarial Reversal ()
Spurious correlation flips sign.
E.2.1. Margin behavior.
(145) If spurious features helped at training,
they hurt equally at deployment.
E.2.2. Worst-case analysis.
For constrained :(146) attained when (adversarial
direction).
E.2.3. Examples.
Hotels: US prefers long reviews; international
region has long complaint reviews low-quality
preferred.
Code: Expert code is verbose; beginner incorrect
code is also verbose
incorrect preferred.
Safety: Safe responses hedge; adversarial unsafe
responses hedge more
unsafe rated as safe.
Spurious correlation changes direction (relevant when ).
E.2.4. Alignment analysis.
The spurious margin depends on alignment: (147) where .
Effect
Aligned: no degradation
Orthogonal: spurious noise only
Opposite: maximum degradation
(reversal)
E.2.5. Examples.
Hotels: Length correlates positively in US,
negatively in Europe; star rating correlates oppositely partial
misalignment.
Code: Comment density and line count have
different correlation patterns across languages rotation.
Safety: Formality and hedging evolve differently
over time temporal
rotation.
F. Experimental Details and
Additional Results
This appendix provides a comprehensive empirical validation of the
theoretical results presented in the main paper, progressing from
settings that exactly match the theory to increasingly realistic and
expressive model classes. We begin with linear preference models, where
the assumptions of the theory hold and we obtain precise quantitative
agreement with predicted bounds and scaling laws. We then move to
nonlinear neural networks, where the causal–spurious decomposition is
hidden inside nonlinear representations, and finally to large language
models trained with DPO on synthetic preference data. While exact
quantitative predictions no longer apply beyond the linear regime, we
demonstrate that the core qualitative mechanisms identified by the
theory, spurious correlation learning, irreducible deployment error
under distribution shift, and mitigation via tie training, persist
across all stages. Together, these experiments show that the linear
analysis captures essential dynamics of preference learning systems even
when deployed with rich nonlinear models. Across all experiments, shaded
regions denote standard
deviation. The number of seeds varies by setting, with many runs in the
linear and deployment-error experiments and fewer runs for the costly
LLM experiments. We report the exact number in the corresponding
hyperparameter table for each setting. Code is available at https://github.com/cmoyacal/tie-training.
F.1. Linear Models (Theoretical
Ground Truth)
F.1.1. Dataset Construction
F.1.1.1. Feature
decomposition.
We construct a synthetic preference dataset with an explicit
causal–spurious feature decomposition. Each example is represented by a
Gaussian feature vector , where denotes causal features and denotes spurious features. Causal
features determine true preference utility, while spurious features do
not affect utility. Under the training distribution , spurious features are correlated with
causal features. At deployment, this correlation changes under a shifted
distribution .
F.1.1.2. Strict preference
pairs.
Strict preference pairs are generated by (148) where decomposes into causal and
spurious components. In the experiments, we use . The preferences depend on
the full feature difference, inducing spurious correlations when is correlated with . Spurious features are drawn as
(149) where
controls the variance scale of spurious features in strict data. This
construction ensures that spurious cues are predictive during training
despite being non-causal.
F.1.1.3. Tie examples.
Tie examples are defined by a small causal margin (150) which produces pairs with weak causal signal. Labels for
ties are randomized as , removing any causal dependence.
Spurious features for ties are drawn from (151)
where controls the
variance injected by tie data. This construction decorrelates spurious
features from labels while amplifying spurious variance, creating
targeted negative evidence against spurious reliance.
F.1.2. Model and Objective
F.1.2.1. Model class.
We train a log-linear preference model (152) which
matches the assumptions of the theoretical analysis. The parameter
vector decomposes into
causal and spurious components. This setting allows direct measurement
of spurious reliance through parameter norms. As a result, population
predictions can be tested exactly.
F.1.2.2. Training objective
(DPO-style).
Training minimizes a DPO-style pairwise logistic loss (153) The parameter
controls the sharpness of
preference separation. We vary the fraction of strict versus tie
examples using . This setup
exactly matches the assumptions of Theorems 3, 6, and 8. The same parameterization also
subsumes RLHF-style reward learning, which corresponds to this objective
with and zero reference. We
show in Appendix G that our mechanisms and shift
vulnerabilities therefore apply to standard reward modeling as well.
F.1.3. Metrics
F.1.3.1. Spurious reliance.
We measure spurious reliance using the norm . This quantity directly
quantifies how much the learned model relies on spurious features.
Because the model is linear, this metric has a clear population
interpretation. It provides a precise test of theoretical
predictions.
F.1.3.2. Reduction ratio.
We compare empirical and theoretical reductions in spurious reliance
under tie training. Empirically, we compute (154) The theory
predicts (155) which
depends on the strict fraction and the spurious variance ratio
.
F.1.4. Results
F.1.4.1. Spurious correlation
learning
Strict-only training learns nonzero spurious parameters. Figure 3 compares
empirical spurious parameter norms to the population prediction from
Theorem 3. Empirically, we found that
correcting for curvature leads to the correct scaling across .
Figure 3.
Population scaling of spurious parameters in DPO. We compare empirical
spurious parameter norms with the population prediction from
Theorem 3.Left: Including
curvature yields accurate predictions across β.Right: Ignoring
curvature systematically underestimates spurious reliance, leading to
large relative error even with infinite data. This confirms that
curvature helps correct population scaling when the local regime
assumption (Assumption 10) is not valid.
Remark 25 (Curvature-corrected linear prediction). The local
linear approximation replaces the logit curvature of the logistic loss
by its value at the origin, , giving with and .
This is accurate when is small.
Away from this regime, the loss no longer has effective logit curvature
: logistic saturation reduces
the average curvature to .
We therefore use the curvature-corrected approximation .
When logits remain near zero, and the correction is negligible. In a more saturated
regime, , so the same
mean bias produces a larger
displacement; the factor corrects this curvature
mismatch while preserving the direction predicted by the linear local
theory.
F.1.4.2. SGD ablation: monitoring
the local regime.
We perform a stochastic-gradient ablation that exposes the dynamics
underlying the equilibrium prediction. We consider three data-generating
levels of increasing complexity. In no_BT, the feature differences are drawn from a
Gaussian with nonzero mean and arbitrary covariance. In BT_causal, features are Gaussian with a
correlation between causal and
spurious blocks, and the Bradley–Terry annotator depends only on the
causal block (). In BT_full, features are
independent isotropic Gaussian and the Bradley–Terry annotator depends
on both blocks (). For each level we run SGD on the DPO loss with learning
rate for iterations, repeated over seeds with a single fixed ground
truth. We fix and use
with training samples per run. To
diagnose whether the linearized analysis applies along the trajectory,
at every SGD step we record the worst-case magnitude on a fixed -point subsample: this is the
local-regime indicator, since the first-order Taylor expansion of the
sigmoid that underlies the closed form
holds quantitatively while this quantity remains below .
Figures 4–Figure 6 report
three panels per setting: the spurious-norm trajectory against the
closed-form predictions, the margin diagnostic with horizontal references at and , and a bar comparison of the final
spurious norm against the Linear, DPO, and (where applicable)
curvature-corrected closed forms. In no_BT (Fig. Figure 4), the
margin stays below throughout
training and the SGD trajectory rises toward the Linear/DPO equilibrium,
with the empirical norm lying within sampling variability of the
closed-form predictions at
iterations. In BT_causal (Fig. Figure 5), the margin grows past the
reference threshold during training, exiting the local regime; the
trajectory overshoots and then drifts downward, and the final SGD norm
lies above the closed-form predictions, indicating that the local regime
no longer captures the late-time training dynamics. In BT_full with curvature correction (Fig. Figure 6), the margin sits well above the
boundary of the local regime; the curvature-corrected prediction (-corr) recovers the empirical norm
where the uncorrected Linear bar undershoots. Table 2 reports key
hyperparameters for the log-linear mechanism experiments.
Spurious-norm trajectory.
Local-regime margin.
Final-norm comparison.
Figure 4.
SGD ablation, no_BT. The margin stays
below throughout training and the
trajectory approaches the Linear/DPO prediction.
Spurious-norm trajectory.
Local-regime margin.
Final-norm comparison.
Figure 6.
SGD ablation, BT_full. The margin sits
well above the boundary of the local regime; the curvature-corrected
prediction (ζ-corr) recovers
the empirical norm where the uncorrected Linear bar undershoots.
Spurious-norm trajectory.
Local-regime margin.
Final-norm comparison.
Figure 5.
SGD ablation, BT_causal. The margin grows
past , the trajectory overshoots, and
the final SGD norm lies above the closed-form predictions, reflecting
departure from the local regime.
Table 2. Key hyperparameters for the log-linear mechanism experiments.
Remaining settings follow the defaults in our released code.
Hyperparameter
Value
Data-generating process
Causal feature dim
Spurious feature dim
Training samples
KL strength
Training (SGD ablation)
Optimizer
SGD on DPO loss
Learning rate
Iterations
KL strength
DGP levels
no_bt,
bt_causal, bt_full
Seeds
F.1.4.3. Deployment error.
Spurious learning induces a deployment vulnerability that persists
under distribution shift. Figure 7 shows
in-distribution () and
out-of-distribution () accuracy as
a function of training set size ,
in a setup where the spurious feature flips sign between and . In-distribution accuracy grows with
while out-of-distribution
accuracy plateaus far below it, illustrating the irreducibility of
deployment failure under spurious shift. Figure 8 reports
the same phenomenon for ,
validating Theorem 20: as grows, the empirical suboptimality
converges to the shift floor while the parameter-space
estimation error decays at the expected rate. The empirical
suboptimality remains below the theoretical upper bound at every , consistent with the theoretical
guarantee. Table 3
reports key hyperparameters for the deployment-error experiments.
Table 3. Key hyperparameters for the deployment-error experiments.
Remaining settings follow the defaults in our released code.
Hyperparameter
Value
Data-generating process
Causal feature dim
Spurious feature dim
Spurious correlation
Label model
vs
sign flip on spurious block
Training and evaluation
Training sizes
Test pairs per distribution
Ridge regularization
Parameter ball radius
Confidence
Seeds
Figure 7. In-distribution () vs out-of-distribution () accuracy under spurious shift, as a
function of training set size .
Both test sets have the same size; under the spurious feature flips sign.
In-distribution accuracy grows mildly with , while out-of-distribution accuracy
plateaus far below it (just below the random-guess line), demonstrating
that spurious reliance learned under cannot be fixed by additional training
data from .Figure 8. Empirical deployment suboptimality and
its decomposition under distribution shift, for . The figure shows four
quantities as a function of the number of training samples : (i) empirical deployment suboptimality
; (ii)
shift error ; (iii) measured
parameter estimation error ; (iv) theoretical upper bound
from Theorem 20. As grows, the estimation error decays at
the expected rate,
demonstrating that deployment error is irreducible with additional
training data from . The empirical
suboptimality remains below the theoretical upper bound at every , consistent with the theoretical
guarantee.
F.1.5. Tie Training
Tie training suppresses spurious reliance in a predictable,
population-level way. Figure 9 plots the theoretical
reduction factor for different
spurious variance ratios , showing monotone
suppression as tie fraction increases. This prediction holds
independently of sample size and isolates the irreducible effect of tie
training on spurious learning. We use this curve to interpret empirical
reductions in
across .Table 4 reports key
hyperparameters for the tie-training experiments.
Table 4. Key hyperparameters for the tie-training experiment. Remaining
settings follow the defaults in our released code.
Hyperparameter
Value
Data-generating process
Causal feature dim
Spurious feature dim
Training samples
Spurious variance
Spurious variance ratio
KL strength
Mixing
Strict fraction
Tie construction
soft (spurious-only, random labels)
regularization
Seeds
Figure 9.
Theoretical prediction for spurious reliance under tie training. The
curve shows the reduction factor as a
function of the strict-preference fraction α, for different spurious variance
ratios σ2/λ0.
Increasing the proportion of tie examples (1 − α) monotonically suppresses
reliance on spurious features, with stronger suppression when ties
inject higher spurious variance. This bound holds independently of
sample size and captures the irreducible effect of tie augmentation on
spurious learning. KL-strength Left:β = 0.3 and Right:β = 0.7.
F.1.6. Log-Linear Experiment:
Causal Decontamination and Near Tie Robustness
F.1.6.1. Intuition for causal
decontamination.
Tie training does not necessarily shrink the causal parameters.
Instead, it reduces the contamination of the causal block induced by
causal–spurious coupling. This can be seen in the same block setting
used in the log-linear experiments. Let (156) denote the second moment and mean of the feature
differences. Adding isotropic ties in the spurious block gives mixed
mean and effective
spurious covariance , since ties have zero mean. In the
linearized equilibrium ,
the causal block can then be written via the Schur complement as (157) Thus, in
this isotropic setting, the influence of the spurious block on the
causal solution enters through the single scalar , which decreases from
toward as the tie contribution increases. Smaller weakens the
causal–spurious coupling, and in the limit the causal block
coincides with the pure-causal solution obtained from the causal
features alone, (158) This motivates the
metric used below: we compare the causal block to the pure-causal
solution rather than measuring the norm of the causal parameters, since
the latter may move in either direction and does not capture
decontamination.
F.1.6.2. Experimental setup.
We sample feature differences directly, .
This lets us set the local regime by selecting small. We use ,, isotropic blocks , with , and a diagonal
causal–spurious coupling with . The mean
has small nonzero entries in
both blocks, so the spurious features leak into the strict-preference
solution. Strict data are drawn from . Tie data have
and
with . For each mixing
level we solve DPO on the
pooled strict-plus-tie data to obtain , on the strict
data alone for , and on the
strict data with the spurious coordinates zeroed for . We report the
relative distance (159)
averaged over random seeds.
F.1.6.3. Results.
Figure 10 shows that increasing
the tie contribution moves
the causal block substantially closer to the pure-causal solution: the
relative distance falls from in
the strict-only case () to
roughly , closely tracking the
isotropic factor
predicted by the linearized solution (the small residual gap at large
tie contribution is consistent with finite-sample variation in the
distance estimate). In contrast, the causal norm ratio
changes much less and does not track the decontamination curve. This
supports the interpretation that tie training decontaminates the causal
component by weakening causal–spurious coupling, rather than by
shrinking causal weights directly.
Figure 10. Causal decontamination (). As the tie contribution increases, the causal block
moves toward the pure-causal solution (relative distance, red), closely
following the linearized prediction (blue) and the isotropic factor
.
The causal norm ratio (grey) changes much less and does not track this
contraction, confirming that tie training decontaminates rather than
merely shrinks the causal block.
F.1.6.4. Near ties (robustness
under imperfect ties).
We also test an imperfect-tie setting in which tie pairs differ
slightly in causal features. Near ties have zero mean but nonzero
variance in both blocks, with causal variance and spurious variance , where :(160) with
the two blocks sampled independently. Thus near ties still add far more
curvature in spurious directions than in causal directions, but they no
longer leave the causal block exactly unchanged. The mixed moments
become (161) and the linearized causal block
can be written with two parameters, (162)
Here controls the
reduction of causal–spurious coupling, exactly as for exact ties, while
is the cost of
imperfect ties: a small extra causal-curvature term arising from causal
leakage. When the
dominant effect is still spurious suppression and decontamination; as
the leakage grows, causal
preservation weakens. Near ties therefore serve as a robustness test of
the exact-tie mechanism: they should suppress spurious reliance while
still moving the causal block toward the pure-causal solution, provided
the causal leakage is small relative to the spurious variation.
Accordingly we report two quantities together, the spurious norm ratio
and the causal decontamination ratio .
F.1.6.5. Results.
Figure 11 shows that the same
mechanism persists under near ties. Although near ties introduce small
causal leakage, increasing the tie contribution suppresses spurious
reliance: the spurious norm ratio decreases sharply and closely follows
the linearized prediction. At the same time, the causal decontamination
ratio also decreases, showing that the causal block moves closer to the
pure-causal solution. In contrast, the causal norm ratio changes much
less and does not track either suppression curve. Thus, near ties do not
merely shrink all parameters; they primarily reduce spurious reliance
while still improving causal decontamination when the causal leakage is
small relative to the spurious variation.
Figure 11. Near-tie robustness (). Near ties
introduce small causal leakage but much larger spurious variation. As
the tie contribution
increases, spurious reliance is suppressed and the causal block moves
closer to the pure-causal solution. Both DPO curves closely follow the
corresponding linearized predictions. The causal norm ratio changes much
less, showing that the effect is not simple parameter
shrinkage.
F.1.7. Takeaway (Linear)
The linear experiments provide confirmation of Theorems 3, 6,
and 8. Strict-only training learns
persistent spurious parameters and induces a vulnerability to
irreducible deployment shift error under . Tie training acts as a selective
regularizer on by injecting
spurious variance. This reduces and lowers shift-induced
deployment error without relying on additional samples from . As a result, tie training can remove
the irreducible vulnerability as predicted by our mathematical
analysis.
F.2. Neural Networks (Nonlinear
Regime)
F.2.1. Motivation.
Real-world preference models are nonlinear and do not expose an
explicit causal–spurious feature decomposition. As a result, the
developed linear theory fails to apply exactly in this regime. Thus, our
goal in this appendix is mechanism validation rather than exact
prediction. In particular, we test whether the qualitative
spurious-learning mechanisms identified in the linear analysis persist
when representations are nonlinear and hidden. This allows us to assess
whether the theory captures dominant dynamics rather than model-specific
artifacts.
F.2.1. Dataset Construction
F.2.1.1. Latent variables.
We generate data from a latent quality variable that determines
true preference ordering. Causal features are constructed as nonlinear
functions of with additive noise,
(163) where
can include transformations
such as ,, or (we use in our
experiments). These features contain information about but are not linearly related to it. As
a result, recovering quality requires nonlinear processing.
Spurious features are generated as correlated but non-causal
functions of ,(164) where is correlated with and is independent noise. The parameter
controls the strength of
spurious correlation during training. These features are predictive
in-distribution but have no causal relationship to preference labels.
This construction mirrors spurious cues in real preference datasets.
F.2.1.2. Nonlinear mixing.
The observed input to the model is a nonlinear mixture of causal and
spurious features, (165) where is a fixed, nonlinear function
unknown to the model (we use as a random MLP in the experiments): a
generic, architecture-agnostic entangling of causal and spurious
features that prevents the model from exploiting a hand-designed
structure. The mechanisms we study do not depend on this specific
choice. This mixing prevents direct access to or . As a result, the model must learn
representations internally. This setting tests whether spurious reliance
emerges even when the decomposition is hidden.
F.2.1.3. Tie construction.
We construct tie examples by enforcing for a small threshold
. Preference labels for ties
are assigned randomly, so causal signal is intentionally weak. We
manipulate spurious features independently of , either by assigning opposing extremes
or by randomizing them. This creates pairs with minimal causal
difference and strong spurious contrast. These ties provide targeted
gradient signal against spurious reliance.
F.2.2. Model and Objective.
F.2.2.1. Model.
We train a multilayer perceptron (MLP) reward model that maps nonlinear inputs
to scalar scores. The model has no architectural bias toward separating
causal and spurious components. All structure must be learned from data.
This setting reflects realistic nonlinear preference models. It
therefore provides a stringent test of the theory.
F.2.2.2. Training loss.
Training uses a pairwise logistic objective, (166) which
matches the standard preference optimization loss (we use in the experiments). We vary
the fraction of strict versus tie comparisons using the parameter . When , training uses only strict
comparisons. As decreases, a
larger fraction of tie data is introduced.
F.2.3. Proxy Metrics
F.2.3.1. Spurious gap.
We measure the spurious gap as the difference between accuracy on
pairs where spurious features align with true quality and accuracy on
pairs where they conflict. Let and denote
these accuracies. Their difference quantifies reliance on spurious cues.
A large gap indicates strong spurious dependence.
F.2.3.2. Adversarial accuracy.
We evaluate adversarial accuracy under a distribution where spurious
correlations are reversed. Performance in this setting isolates failure
due to spurious reliance. Because the deployment objective is nonlinear,
we use accuracy as a proxy for utility. Persistent degradation under
this shift indicates misgeneralization.
F.2.3.3. Counterfactual
margin.
We measure counterfactual sensitivity by flipping spurious features
while holding causal features fixed. The counterfactual margin is
defined as (167)
Large values indicate that the learned reward depends strongly on
spurious features. Tie training is expected to reduce this margin.
F.2.4. Results
F.2.4.1. Spurious learning.
Models trained without ties exhibit clear spurious learning. They
show a large spurious gap, indicating substantially different
performance when spurious cues align or conflict with quality. Figure 12 reports this gap
across training conditions.
Figure 12. Spurious gap (accuracy difference between
aligned and misaligned spurious conditions) as a function of the
fraction of strict preferences . Note that as decreases, the number of ties
increases. Thus, tie training reduces spurious reliance despite hidden
representations.
These models also perform poorly under adversarial evaluation,
showing that spurious reliance translates into deployment failures.
F.2.4.2. Tie training.
Introducing tie training reduces sensitivity to spurious features.
Figure 13 shows that
the counterfactual margin decreases sharply as the fraction of tie data
increases, indicating reduced dependence on spurious cues.
Figure 13.Tie training reduces spurious
reliance. Counterfactual margin
as a function of the fraction of strict preferences . As decreases (more tie comparisons),
the counterfactual margin drops sharply, indicating reduced sensitivity
of the learned model to spurious features.
Figure 14 illustrates
that tie training also improves adversarial accuracy under adversarial
reversed correlations.
Figure 14.Strict-only training plateaus
under distribution shift; tie training improves robustness.
Adversarial accuracy on , where spurious
correlations flip, as a function of the number of training samples.
Strict-only training ()
exhibits a persistent accuracy plateau despite increasing data. In
contrast, tie training () improves adversarial
accuracy, breaking the plateau.
These trends qualitatively match the predictions of the linear
analysis.
F.2.4.3. Takeaway (Nonlinear).
In the nonlinear regime, the exact linear theory no longer applies.
Nonetheless, the same qualitative spurious-learning mechanisms persist.
Tie training reduces spurious reliance and improves robustness under
distribution shift. These results suggest that the linear analysis
captures dominant dynamics of preference learning. The theory therefore
provides useful guidance beyond the linear setting.
F.2.4.4. Hyperparameters.
Table 5 reports key hyperparameters
for the nonlinear experiments.
Table 5. Key hyperparameters for the nonlinear synthetic experiments
(spurious gap vs. ,
counterfactual margin vs. ,
and adversarial accuracy vs. sample size ). Remaining settings (initializations,
loss reduction, dataloader shuffling) follow the defaults in our
released code.
Hyperparameter
Value
Data generating process
Causal latent dim
Spurious vector dim
Shortcut scalar dim
Spurious-vector correlation
Shortcut– correlation
Noise scales
Bradley–Terry teacher
Mixer
Frozen MLP, Linear–Tanh–Linear–Tanh
Mixer hidden width
Scorer model & training
Scorer architecture
3-layer MLP, ReLU
Hidden width
Loss
Pairwise BCE-with-logits
Model temperature
Optimizer
AdamW
Learning rate
Weight decay
Epochs
Batch size
Tie construction & data
Tie construction
Spurious flip: vs. , label
Training pairs (gap / CF-margin
sweeps)
Training pairs (-sweep)
Eval pairs (spurious gap, adv. acc.)
Eval pairs (CF margin)
Mixing ratio (-sweeps)
Mixing ratio (-sweep)
(strict baseline), (tie
training)
Test distributions
(),()
Seeds
F.3. Large Language Models
(Synthetic Hotel Benchmark)
We study a large-scale synthetic language-model setting, where
preferences are expressed in natural language and spurious attributes
resemble real-world surface cues.
F.3.1. Dataset: synthetic hotel
preferences.
We construct a synthetic hotel comparison dataset designed to study
spurious correlation learning and the effect of tie training in large
language models. Each example consists of a user context, two hotel
options , and a binary
preference label indicating which hotel is preferred. The dataset
explicitly separates causal utility, which determines true quality, from
spurious features, which are surface attributes correlated with utility
during training. This separation allows controlled experiments in which
correlations can be manipulated without changing the underlying task. As
a result, robustness under distribution shift can be evaluated in
isolation.
Each hotel is assigned a latent true utility computed from task-relevant
attributes, including price,
distance_to_destination, star_rating, and
context-dependent amenities. Preference labels are
generated by a teacher that depends only on this true utility, so higher
always corresponds to higher
quality. The true utility is never directly observed by the model and
must be inferred from preference supervision. This ensures that causal
signal is present only implicitly. Consequently, any reliance on
non-causal attributes reflects spurious learning.
We designate the following hotel attributes as spurious features:
street_number (100–9999)
floor_number (1–20)
building_age (1–50; lower is better)
renovation_year (2000–2024)
hotel_chain_tier {Budget,
Standard, Premium}
lobby_size_sqft (500–5000)
employee_count (10–200)
These attributes do not affect true utility but are correlated with
utility during training. Their values are explicitly manipulated to
create different correlation regimes at deployment. This design ensures
that spurious cues are strong, structured, and controllable. As a
result, failures under shift can be directly attributed to spurious
reliance.
Spurious features are assigned as deterministic functions of a
normalized utility level . In the normal correlation mode, higher-utility
hotels receive systematically better spurious attributes, such as newer
buildings, premium chains, larger lobbies, and more employees. In
suppression mode, spurious attributes are decorrelated from
utility. In adversarial mode, the mapping is inverted using
, so high-utility
hotels receive worse spurious attributes. This construction induces
sharp distribution shifts without altering the causal preference
structure.
For standard (non-tie) training examples, we sample two hotels , compute their true utilities , assign spurious features
according to the chosen correlation mode, and label the pair by the
utility ordering. This procedure induces strong correlations between
spurious attributes, preference labels, and true quality under the
training distribution. These correlations are systematic rather than
noisy. As a result, standard preference optimization objectives are
incentivized to rely on spurious cues. An example of a hotel preference
sample is:
You are helping someone choose the right hotel for their stay. ...
--- Option A ---
Hilton Plaza is prominently located at 4126 Second Ave. ...
--- Option B ---
Hampton Inn Central is prominently located at 6560 Park Blvd. ...
--- Task ---
Which of these two options is the better choice for the user?
A tie is defined as a hotel pair such that the utility difference
satisfies for
a small threshold . In these
pairs, the causal signal is intentionally weak by construction.
Preference labels are assigned randomly, with . This
ensures that labels are independent of both utility and spurious
features. Ties therefore isolate non-causal learning signals.
F.3.3. Informative ties.
We make tie examples informative by explicitly decorrelating spurious
features from utility. Our default informative strategy assigns one
hotel maximal spurious features and the other minimal spurious features.
The assignment is random between
and , so spurious direction is
uninformative. This produces pairs with near-zero causal margin but
maximal spurious contrast. As a result, gradients from these examples
penalize spurious reliance.
Informative ties satisfy three properties simultaneously.
First, the causal signal is weak because is small. Second, the spurious
contrast is large because spurious features are maximally separated.
Third, labels are independent of spurious attributes by construction.
Together, these properties ensure that tie gradients push against
spurious features while preserving causal learning.
We also evaluate an alternative tie construction strategy. In a
standard-monotonic strategy, we assign spurious features monotonically
from utility. These non-informative ties provide weak
regularization signal.
F.3.4. Model and training.
We fine-tune a fixed base language model
Llama-3.2-1B-Instruct using Direct Preference Optimization
(DPO) (Rafailov et al. 2023) and Low-Rank Adaptation (LoRA) (Hu et
al. 2022) for one epoch. All experiments use the same architecture and
optimization settings. We compare strict training, which uses only
standard preference pairs, with tie training, which augments the dataset
with ties.
F.3.5. Evaluation metrics.
We measure in-distribution accuracy on standard test pairs
drawn from the training distribution . This evaluates whether models trained
with ties retain performance on the original task. High in-distribution
accuracy indicates that causal learning is preserved. We report overall
accuracy and per-option accuracy. This allows us to detect asymmetric
degradation.
We evaluate robustness under deployment distributions where spurious correlations are
suppressed or adversarially reversed. These settings isolate failures
caused by spurious reliance. Accuracy is measured using the same
preference labels derived from true utility. Performance degradation
under reflects reliance on
spurious features. Robust models should maintain accuracy across
shifts.
F.3.6. Results.
Table 6 shows that Tie Training (TT)
DPO with informative ties improves robustness to spurious distribution
shift. While standard DPO performs well in-distribution (), its accuracy degrades under
suppressed and adversarial spurious correlations (). Informative tie training preserves
high accuracy under both shifts without sacrificing in-distribution
performance. The mixture variants (TT-Mixture) sample informative ties
with probability and otherwise
emit non-informative ties, interpolating between the two regimes. We
report and . Non-informative ties inject
spurious contrast in the same direction as the causal signal. The modest
robustness they do provide stems from the random tie label and near-tie
causal perturbation, which act as weak regularizers.
Table 6. Tie Training (TT) DPO with informative ties improves robustness
to spurious distribution shift. While standard DPO performs well
in-distribution (), its accuracy
degrades under suppressed and adversarial spurious correlations (). Informative tie training preserves
high accuracy under both shifts without sacrificing in-distribution
performance. The mixture variants (TT-Mixture) sample informative ties
with probability and otherwise
emit non-informative ties, interpolating between the two regimes. We
report and . Non-informative ties inject
spurious contrast in the same direction as the causal signal. The modest
robustness they do provide stems from the random tie label and near-tie
causal perturbation, which act as weak regularizers. Results are
reported for overall accuracy and per-option accuracy (Hotel A/B),
illustrating that robustness gains are systematic rather than
label-specific. In all experiments, we use augmented ties.
Accuracy (overall)
Accuracy (Hotel A)
Accuracy (Hotel B)
DPO-Strict-In-Distr.()
92.25%
89.54%
94.88%
DPO-Strict-Suppressed()
74.00%
74.48%
77.81%
DPO-Strict-Adversarial()
64.20%
55.72%
72.76%
DPO-TT-Informative-In-Distr.()
92.40%
88.02%
96.65%
DPO-TT-Informative-Suppressed()
82.85%
83.17%
82.50%
DPO-TT-Informative-Adversarial()
86.70%
80.60%
92.86%
DPO-TT-Mixture ()-In-Distr.()
89.40%
92.76%
86.34%
DPO-TT-Mixture ()-Suppressed()
83.40%
90.81%
76.30%
DPO-TT-Mixture ()-Adversarial()
83.20%
85.84%
80.47%
DPO-TT-Mixture ()-In-Distr.()
90.75%
91.92%
89.68%
DPO-TT-Mixture ()-Suppressed()
84.10%
86.72%
81.59%
DPO-TT-Mixture ()-Adversarial()
81.70%
79.84%
83.62%
DPO-TT-Non-Informative-In-Distr.()
91.20%
92.13%
90.35%
DPO-TT-Non-Informative-Suppressed()
81.85%
83.04%
80.71%
DPO-TT-Non-Informative-Adversarial()
76.90%
72.47%
81.49%
F.3.7. Ablation of .
We fine-tune Llama-3.2-1B-Instruct with DPO (LoRA, ,) on strict preference pairs and sweep
the tie augmentation ratio , the
fraction of tie pairs in the augmented training set. Ties are near-tie
pairs in which the spurious feature is decorrelated from utility; is the strict-only baseline
(no ties) and lower injects
more ties. All models are evaluated on a fixed adversarial test set
(, pairs, spurious correlation
strength ) held constant across
seeds. We report mean standard
deviation over 5 seeds.
Results. Adversarial accuracy rises monotonically as
ties are added, from at
to at , a -point gain over the strict-only
baseline (Figure 15). The largest improvement comes
from the first increment of ties (, points), with diminishing returns
thereafter. Variance also shrinks as decreases (std ), indicating that tie
augmentation yields not only higher but also more consistent robustness
across seeds.
Figure 15. Tie augmentation improves adversarial
robustness. Adversarial accuracy (, anti-correlated
spurious feature) of Llama-3.2-1B-Instruct fine-tuned with DPO, as a
function of the tie fraction , where is
the share of strict pairs in the training set. The leftmost point (, i.e. ) is the strict-only baseline
(dashed line). Adding decorrelated ties raises accuracy from to , with the gain concentrated in the
first increment and variance shrinking as more ties are added. Markers
show the mean over 5 seeds; error bars show standard deviation.
F.3.8. Takeaway (LLMs).
Spurious correlation learning persists in large language models
trained with preference optimization. Increasing data alone does not
eliminate spurious reliance or improve robustness under distribution
shift. Informative tie training provides targeted robustness gains by
directly penalizing spurious features. These effects mirror those
observed in linear models and neural networks. Together, the results
suggest that the linearized theory captures core dynamics of preference
learning under spurious correlations.
F.3.9. Hyperparameters.
Table 7 reports key hyperparameters
for the LLM hotel experiments.
Table 7. Key hyperparameters for the LLM Hotels experiments. Remaining
settings (LoRA dropout, target modules, optimizer, sequence lengths, LR
schedule) follow the defaults in our released code.
Hyperparameter
Value
DPO training
Base model
Llama-3.2-1B-Instruct
Fine-tuning
LoRA (4-bit), ,
DPO
Learning rate
Epochs
Effective batch size
Data
Training pairs (strict)
Test pairs per distribution
Test distributions
,,
Spurious correlation strength
Tie construction
near-tie
Mixing ratio
(baseline), (tie training)
Seeds
G. RLHF-Style Reward Learning with
Greedy Decoding
G.1. Reward Learning Setup
We study linear reward learning under RLHF and show that it is a
special case of Direct Preference Optimization (DPO). Specifically, RLHF
reward learning corresponds to DPO with scaling parameter and reference parameter , so
that the effective parameter satisfies . Under this
specialization, the pairwise reward learning loss reduces to (168) which matches the DPO
objective. This equivalence implies that the reward learning dynamics
are unchanged relative to DPO in our linear analysis. Thus, the same
spurious-learning mechanisms and distribution-shift vulnerabilities
apply to standard RLHF reward learning.
G.2. Deployment with Greedy
Policies
Given a learned linear reward ,
deployment uses greedy decoding to select outputs. This induces the
greedy policy (169) which is optimal under but can be suboptimal under the
true deployment utility. To measure this gap under a shifted deployment
distribution , we evaluate the
true deployment value functional rather than the learned reward.
Let denote the optimal
policy for the deployment problem under and , and define the deployment
suboptimality as (170) This definition turns reward
mislearning into a policy-level metric that we can track under
distribution shift.
G.3. Spurious Learning under RLHF
Reward Learining
We now show that spurious correlation learning persists under RLHF
reward learning and directly impacts greedy deployment.
G.3.1. Setup.
We evaluate the RLHF reward learning dynamics using a synthetic
environment where features are decomposed into causal features and a spurious
feature .
The full feature vector is given by the concatenation .
We construct a set of five items with a
fixed spurious correlation coefficient . The feature representations
are defined as follows: (171) Ground truth
preference data is generated via a Bradley-Terry model . The ground truth parameter is set to .
Sampling Strategy. We generate a dataset of pairwise comparisons with a fixed
distribution of outcome types: strict preferences (75%). The samples are
drawn evenly from three specific comparison pairs to create the
experimental structure: vs
(difference: ), vs (difference: ), and vs (difference: ). Ties (25%). The remaining
samples are labeled as ties.
G.3.2. Results.
Figure 16 shows that
training with
yields a reward parameter whose spurious component is
nonzero, i.e., . Similarly, Figure 17 illustrates that
increasing the number of samples from reduces estimation error around this
optimum but does not remove the spurious component.
G.4. Effect of Tie Training
We next evaluate how spurious learning induces deployment
suboptimality and how tie training mitigates this effect.
G.4.1. Results.
Without tie training, Figure 18 shows that the greedy
policy achieves low error on but incurs nonzero when the
deployment distribution
suppresses or reverses spurious correlations. As the number of training
samples increases, the shift-induced component of the error persists.
This persistence is strongest in adversarial reversals. Tie training
reduces by
shrinking
while preserving the causal component, so the greedy policy becomes less
sensitive to spurious shifts.
Figure 16. Under standard RLHF learning training,
the learned policy exhibits nonzero reliance on spurious features (), and this
reliance does not vanish with additional data drawn from the training
distribution . Tie training
explicitly counteracts this effect, driving spurious reliance toward
zero.Figure 17. As the number of training samples
increases, estimation error, defined as the weighted norm
decreases at comparable rates for strict MLE training and tie training,
showing that tie training reduces spurious reliance without sacrificing
estimation accuracy.Figure 18. Greedy decoding of a log-linear RLHF
policy does not introduce additional error mechanisms, but exposes
spurious reward learning under shift: performance, measured as degrades in both adversarial and suppression
settings, and this error does not vanish with more data from . Tie training reduces this
shift-induced error.
G.5. Discussion and Conclusion
Our results show that greedy decoding does not affect what reward
learning fits, but it could amplifies the behavioral impact of spurious
reward errors. Under distribution shift, small spurious weights can flip
greedy decisions and induce nonzero deployment suboptimality. Without
tie training, this error persists in the infinite-data limit. Additional
samples reduce estimation error but do not remove spurious reliance. In
contrast, tie training shrinks spurious weights and reduces the
resulting deployment error.
H. Limitations and Future Work
Local regime and linearization. Our theoretical
analysis relies on a local regime where the learned policy remains close to the reference
policy , enabling
linearization of the KL-regularized objective.
Tie construction and approximate equality. The
analysis assumes access to informative ties, preference pairs , where responses have
near-equal utility but differing spurious features. In practice, exact
utility equality is difficult to verify, since true utility is latent
and must be estimated from noisy human feedback or imperfect reward
models.
However, the mechanism requires only that causal utility differences
are small relative to spurious feature variability, not exact equality.
Formally, what matters is that is small compared to in
expectation over the tie distribution. Our experiments (Appendix F) demonstrate that
ties constructed by selecting pairs with similar scores effectively
reduce spurious learning, even when utilities are not exactly equal.
Nevertheless, formal guarantees for imperfect tie construction remain
an open problem. How much utility mismatch can be tolerated before tie
training becomes ineffective? How should one trade off the number of
ties versus their quality? Can adaptive tie construction algorithms
identify informative ties online during training? We will study these
questions in our future work.
Analytical assumptions and feature decomposition. We
assume features decompose as into causal and spurious
components. This decomposition makes the spurious learning mechanism
explicit and enables clean theoretical statements, but it is an
idealization.
Importantly, this assumption is used only for analysis, not for the
method. The decomposition serves as an analytical tool to understand why
tie training works, not as a prerequisite for its application. Extending
the theory to settings without a clear causal–spurious decomposition
remains open.
Scale and scope of validation. Our experiments
validate the theoretical mechanisms we derive: we demonstrate that
predictions from the population-level theory match finite-sample
behavior, that tie training reduces spurious parameters as predicted,
and that these reductions translate to improved deployment robustness.
However, these experiments are designed to test theoretical predictions
in controlled settings, not to optimize end-to-end performance of
production alignment systems.
Large-scale validation in production alignment pipelines remains
important future work. This includes: applying tie training to frontier
language models with billions of parameters, developing practical
methods for tie construction from human feedback at scale, evaluating
robustness improvements on diverse downstream tasks and distribution
shifts, and comparing tie training to other robustness interventions
such as distributionally robust optimization or causal regularization.
Such validation would determine whether the gains observed in controlled
experiments translate to meaningful improvements in deployed
systems.