Back to AI News
arxiv

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.

AI Brief

Provides a concrete method for improving small LM self-improvement using trajectory preferences, with empirical gains on reasoning tasks and transfer benchmarks.

arXiv:2610.41296v1 Announce Type: new Abstract: Answer-level rewards cannot distinguish reasoning traces with equally correct outcomes. We study whether auxiliary trajectory preferences improve small-language-model self-improvement and transfer under limited supervision. We compare confidence-based (SC) and entropy-based metrics (UID-Conj.), which combines global entropy variation and local smoothness, through online reward augmentation and offline training on correct responses. Online experiments adapt Qwen3-1.7B on 30 AIME problems without reference answers; offline experiments use historical AIME problems. UID-Conj. improves the online macro-average across the adaptation benchmark and three transfer benchmarks by 2.08 percentage points over unmodified TTRL. The strongest evidence comes from offline weighted SFT: UID-Conj. exceeds uniform weighting on the same response pool by 1.93 points, with a paired prompt-bootstrap 95\% CI of [0.44, 3.44]. Benefits are task-dependent: SC rewards judged trace quality and instruction following, while UID-Conj. rewards yield stronger training outcomes in several settings. We find that auxiliary preferences can expose learning value among equally correct traces not captured during inference time, motivating their evaluation through downstream gains.
Be the first to rate this article.

Comments & Questions (0)

Captcha
Please be respectful — let's keep the conversation friendly.

No comments yet

Be the first to comment!