YFPO: Yoked Feature Preference Optimization with Neuron-Guided Rewards

Yifan Le


Abstract
Preference optimization has become a widely used post-training paradigm for improving the reasoning abilities of large language models. Existing methods typically learn from preferred and dispreferred responses as external behavioral supervision, while largely ignoring capability-related signals encoded in the model’s internal representations. In this work, we study whether such internal signals can provide useful auxiliary supervision for mathematical reasoning. We introduce YFPO (Yoked Feature Preference Optimization), a neuron-guided preference optimization framework that couples response-level preference learning with neuron-level rewards. YFPO first uses AttnLRP to identify math-associated internal features, and then derives an auxiliary reward from the activation margin of these neurons between preferred and dispreferred responses. This reward is combined with the standard preference optimization objective, encouraging the model to align external preferences with internal math-related features. We conduct small-scale experiments on GSM8K with a compact language model. Results show that neuron-guided rewards can influence preference optimization dynamics and yield measurable improvements in several settings, suggesting that internal representations can serve as lightweight and interpretable signals for reasoning-oriented post-training.
Anthology ID:
2026.findings-aacl.116
Volume:
Findings of the Association for Computational Linguistics: IJCNLP-AACL 2026
Month:
November
Year:
2026
Address:
Hengqin, China
Editors:
Yue Zhang, Ruifeng Xu, Ayu Purwarianti, Tong Rong, Jun Suzuki
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
2291–2298
Language:
URL:
https://aclanthology.org/2026.findings-aacl.116/
DOI:
Bibkey:
Cite (ACL):
Yifan Le. 2026. YFPO: Yoked Feature Preference Optimization with Neuron-Guided Rewards. In Findings of the Association for Computational Linguistics: IJCNLP-AACL 2026, pages 2291–2298, Hengqin, China. Association for Computational Linguistics.
Cite (Informal):
YFPO: Yoked Feature Preference Optimization with Neuron-Guided Rewards (Le, Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-aacl.116.pdf