Self-Improvement with Auxiliary Rewards: Learning from Verified Traces and Evaluating Transfer
This paper investigates using auxiliary trajectory preferences, specifically confidence-based (SC) and entropy-based (UID-Conj.) metrics, to enhance small language model self-improvement and transfer under limited supervision. Online experiments with Qwen3-1.7B on 30 AIME problems show UID-Conj. improves macro-average by 2.08 percentage points over unmodified TTRL, while offline weighted SFT yields a 1.93 point gain with a 95% CI of [0.44, 3.44]. The benefits are task-dependent, with SC rewarding instruction following and UID-Conj. producing stronger training outcomes in several settings.
Provides a concrete method for improving small LM self-improvement using trajectory preferences, with empirical gains on reasoning tasks and transfer benchmarks.
Comments & Questions (0)
No comments yet
Be the first to comment!