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 , while shift error
persists, demonstrating irreducibility. As predicted by
Theorem 6, empirical
deployment error
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
(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 ) 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 .
Table 2.
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 3 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 3. 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 4 reports key
hyperparameters for the deployment-error experiments.
Table 4. 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 5 reports key
hyperparameters for the tie-training experiments.
Table 5. 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 .
Increasing the proportion of tie examples () 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: and
Right:.
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 6 reports key
hyperparameters for the nonlinear experiments.
Table 6. 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 7. 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 8 reports key
hyperparameters for the LLM hotel experiments.
Table 8. 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.