用大模型自动优化复杂系统,省去人工调参
POLCA: Stochastic Generative Optimization with LLM
- 让大模型当优化器,靠奖励和反馈找最优解
- 在噪声环境下仍能高效收敛,比现有方法更稳更快
- 适合需要反复调试的智能体、代码生成等场景
优化复杂系统(如大模型提示词、多轮智能体)传统上依赖人工反复试错。本文将此问题形式化为随机生成式优化任务,由生成式语言模型作为优化器,通过数值奖励与文本反馈寻找最优系统。提出优先级优化与局部上下文聚合框架(POLCA),可应对反馈噪声、采样批次及系统随机性,同时控制解空间无限膨胀。POLCA采用优先队列平衡探索与利用,并引入ε-网机制保持参数多样性,结合大模型摘要器实现历史试验的元学习。理论上证明其在随机环境下可收敛至近优解。在τ-bench、HotpotQA(智能体优化)、VeriBench(代码转换)、KernelBench(CUDA内核生成)等基准上验证,POLCA在确定性与随机性任务中均表现优异,样本与时间效率显著优于当前最优算法。代码已开源。
原文摘要 · Abstract (English)
Optimizing complex systems, ranging from LLM prompts to multi-turn agents, traditionally requires labor-intensive manual iteration. We formalize this challenge as a stochastic generative optimization problem where a generative language model acts as the optimizer, guided by numerical rewards and text feedback to discover the best system. We introduce Prioritized Optimization with Local Contextual Aggregation (POLCA), a scalable framework designed to handle stochasticity in optimization -- such as noisy feedback, sampling minibatches, and stochastic system behaviors -- while effectively managing the unconstrained expansion of solution space. POLCA maintains a priority queue to manage the exploration-exploitation tradeoff, systematically tracking candidate solutions and their evaluation histories. To enhance efficiency, we integrate an $\varepsilon$-Net mechanism to maintain parameter diversity and an LLM Summarizer to perform meta-learning across historical trials. We theoretically prove that POLCA converges to near-optimal candidate solutions under stochasticity. We evaluate our framework on diverse benchmarks, including $τ$-bench, HotpotQA (agent optimization), VeriBench (code translation) and KernelBench (CUDA kernel generation). Experimental results demonstrate that POLCA achieves robust, sample and time-efficient performance, consistently outperforming state-of-the-art algorithms in both deterministic and stochastic problems. The codebase for this work is publicly available at https://github.com/rlx-lab/POLCA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。