Selective Tie Learning: Mitigating Spurious Correlations in Preference Optimization
Organizations: Purdue University · National University of Singapore
Abstract
Preference optimization can learn spurious correlations, and augmenting the preference data with ties (equally preferred responses) can reduce spurious reliance. However, ties differ in their corrective effect, making their selection for augmentation important. For log-linear policies, we derive each tie's influence on spurious reliance and show that taking the largest predicted reductions is first-order optimal among subsets of equal size. Computing these influences requires knowledge of the spurious features and the curvature of the training objective, which is often unavailable or expensive to estimate. To this end, we explore using the magnitude of a tie's implicit reward margin and characterize the conditions under which it is a useful proxy for predicted influence. To test the theory and its qualitative transfer to neural models, we estimate influence with a counterfactual proxy for spurious reliance and use a generalized Gauss--Newton approximation to curvature for language models. Across log-linear policies, neural models, and large language models, our selection rules consistently achieve comparable reductions in spurious reliance with fewer ties than uniform selection.
Figures & tables
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Environment | Interpretation | ||
|---|---|---|---|
| Training environment; both shortcut channels align. | |||
| Both shortcut channels are removed. | |||
| Both shortcut channels reverse. |
| Setting | Log-linear | Neural network |
|---|---|---|
| Feature dimensions | ||
| Condition numbers | ||
| Singular values | ||
| Signal strengths | ||
| Teacher coefficient | ||
| DPO coefficient |
| Setting | Hotels | Emojis |
|---|---|---|
| Base policy | Llama-3.2-1B-Instruct | |
| Arithmetic and attention | unquantized bfloat16; FlashAttention 2 | |
| LoRA rank; scale; dropout | ||
| LoRA targets | all attention and feed-forward projections | |
| DPO coefficient | ||
| Optimizer; step size | AdamW; | |
Preference Optimization
Preference optimization aligns AI systems with human or task-specific judgments by training them to favor better-ranked outputs.
See all 155 papers