arXiv:2511.06430cs.LGcs.CL2025-11Transactions of th…被引 1

让大模型在设备端测试时更聪明地自我优化,提升答题准确率。

CG-TTRL: Context-Guided Test-Time Reinforcement Learning for On-Device Large Language Models

  • 用上下文动态引导两次采样,改进伪标签生成
  • 3步内提升8%准确率,比原方法快且准
  • 适合移动端实时推理,效率高

测试时强化学习(TTRL)在测试阶段适应基础模型以完成复杂任务方面展现出巨大潜力,显著提升性能。TTRL采用两阶段采样策略:先通过多轮采样和多数投票生成伪标签,再通过降采样与基于奖励的微调,鼓励模型探索并学习多样有效解,同时伪标签调节奖励信号。然而,当前TTRL在两阶段中对上下文引导利用不足,可能影响伪标签准确性并限制探索。为此,本文提出上下文引导的TTRL(CG-TTRL),将上下文动态融入两个采样阶段,并设计适用于设备端的高效上下文选择方法。在数学与科学问答基准上的评估显示,CG-TTRL优于TTRL(例如相对准确率额外提升7%),且仅需少量测试时训练步骤即实现强性能(如3步后相对提升8%,而TTRL仅为1%)。

原文摘要 · Abstract (English)

Test-time Reinforcement Learning (TTRL) has shown promise in adapting foundation models for complex tasks at test-time, resulting in large performance improvements. TTRL leverages an elegant two-phase sampling strategy: first, multi-sampling derives a pseudo-label via majority voting, while subsequent downsampling and reward-based fine-tuning encourages the model to explore and learn diverse valid solutions, with the pseudo-label modulating the reward signal. Meanwhile, in-context learning has been widely explored at inference time and demonstrated the ability to enhance model performance without weight updates. However, TTRL's two-phase sampling strategy under-utilizes contextual guidance, which can potentially improve pseudo-label accuracy in the initial exploitation phase while regulating exploration in the second. To address this, we propose context-guided TTRL (CG-TTRL), integrating context dynamically into both sampling phases and propose a method for efficient context selection for on-device applications. Our evaluations on mathematical and scientific QA benchmarks show CG-TTRL outperforms TTRL (e.g. additional 7% relative accuracy improvement over TTRL), while boosting efficiency by obtaining strong performance after only a few steps of test-time training (e.g. 8% relative improvement rather than 1% over TTRL after 3 steps).

测试时学习大模型优化设备端推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。