arXiv:2508.08641cs.LGcs.AI2025-08被引 4

让大模型推理时自我优化,无需外部数据

MiGrATe: Mixed-Policy GRPO for Adaptation at Test-Time

  • 用混合策略在线调整模型,结合贪婪与邻近采样提升搜索效率
  • 在词搜索、分子优化和ARC任务上均超越传统方法
  • 适合需要实时优化但无标注数据的黑盒任务场景

大型语言模型(LLMs)正被广泛应用于程序合成、分子设计等黑箱优化任务。以往方法依赖上下文学习迭代引导模型生成更优解,但难以平衡探索新空间与利用高回报区域。近期测试时训练(TTT)结合合成数据虽有效,但需为每项任务手工设计数据,限制了可扩展性。为此,我们提出MiGrATe——一种无需外部训练数据的在线TTT方法,使用GRPO作为搜索算法,在推理阶段自适应调整LLM。MiGrATe通过混合策略分组构造机制,融合在线策略采样与两种离线策略数据选择:贪婪采样(选取历史表现最佳完成项)和邻域采样(生成结构相似于高奖励解的输出),从而在保持探索性的同时,引导策略梯度向高潜力解空间聚焦。我们在三个挑战性任务上评估:词搜索、分子优化及在抽象推理基准(ARC)上的假设+程序归纳,结果表明其持续优于仅推理与传统TTT基线,证明了无监督在线TTT在复杂搜索任务中的潜力。

原文摘要 · Abstract (English)

Large language models (LLMs) are increasingly being applied to black-box optimization tasks, from program synthesis to molecule design. Prior work typically leverages in-context learning to iteratively guide the model towards better solutions. Such methods, however, often struggle to balance exploration of new solution spaces with exploitation of high-reward ones. Recently, test-time training (TTT) with synthetic data has shown promise in improving solution quality. However, the need for hand-crafted training data tailored to each task limits feasibility and scalability across domains. To address this problem, we introduce MiGrATe-a method for online TTT that uses GRPO as a search algorithm to adapt LLMs at inference without requiring external training data. MiGrATe operates via a mixed-policy group construction procedure that combines on-policy sampling with two off-policy data selection techniques: greedy sampling, which selects top-performing past completions, and neighborhood sampling (NS), which generates completions structurally similar to high-reward ones. Together, these components bias the policy gradient towards exploitation of promising regions in solution space, while preserving exploration through on-policy sampling. We evaluate MiGrATe on three challenging domains-word search, molecule optimization, and hypothesis+program induction on the Abstraction and Reasoning Corpus (ARC)-and find that it consistently outperforms both inference-only and TTT baselines, demonstrating the potential of online TTT as a solution for complex search tasks without external supervision.

大模型优化测试时训练自适应推理

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