Generalized Distributional Alignment Games for Unbiased Answer-Level Fine-Tuning
Not on the map. Why?
Abstract (arXiv)
The Distributional Alignment Game framework provides a powerful variational perspective on Answer-Level Fine-Tuning (ALFT). However, standard algorithms for these games rely on estimating logarithmic rewards from small batches, introducing a systematic bias due to Jensen's inequality that can destabilize training. In this paper, we systematically resolve this structural estimation bias. First, we generalize the alignment game to arbitrary Bregman divergences, showing that for a family of geometries inducing polynomial rewards, we can construct provably exact and unbiased estimators using U-statistics. Second, for the canonical KL divergence game where an exact solution is impossible, we derive a globally robust minimax polynomial estimator that is provably optimal, achieving the fundamental statistical error limit of Theta(1/K2), which we establish via the Ditzian-Totik theorem. Finally, we synthesize these two approaches to propose a novel Variance-Optimal Augmented Polynomial Optimization Program (AQP) Estimator, proving that by systematically reducing variance, our method achieves not only optimal bias but also provably accelerated game convergence, leading to more efficient and stable training with zero online computational overhead.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 13 of its 20 references that are in the map. First posted on arXiv 4 May 2026.
Builds on
13 of 20 references.
References · 13
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
- 2023Let's Verify Step by Stepcited by 3,598
- 2021Convex optimizationcited by 2,479
- 2022Solving math word problems with process- and outcome-based feedbackcited by 871
- 2018MINE: Mutual Information Neural Estimationcited by 680
- 2016CVXPY: A Python-Embedded Modeling Language for Convex Optimizationcited by 633
- 2014Convex Optimization: Algorithms and Complexitycited by 569
- 2019On Variational Bounds of Mutual Informationcited by 550
- 2016From Softmax to Sparsemax: A Sparse Model of Attention and Multi-Label Classificationcited by 337
1 more
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Online Learning-to-Defer with Varying Expertscited by 0
- 2026Online Learning-to-Defer with Varying Experts12 May 2026