arXiv:2604.08801cs.LGcs.CL2026-04被引 1

通过筛选高差异性用户提示,用更少提示实现更优系统提示优化。

$p1$: Better Prompt Optimization with Fewer Prompts

  • 筛选用户提示中对系统提示变化响应差异大的样本
  • 仅用2个AIME 24提示即实现跨任务良好泛化
  • 适用于需要高效优化且数据异质的推理任务

提示优化通过搜索更优系统提示来提升语言模型性能,而无需更新模型权重,但其效果在不同任务间差异显著。我们发现,系统提示间的奖励方差可分解为响应方差(生成随机性)与系统提示方差(提示质量差异)。当系统提示方差足够大时优化成功,反之则失败。令人意外的是,增加用户提示数量可能因降低系统提示方差而损害优化,尤其在异质数据集上。为此,我们提出 $p1$,一种简单用户提示过滤方法:选择在候选系统提示下响应差异大的少量用户提示。该子集能有效区分好坏系统提示,提升优化效率。实验表明,$p1$ 在推理基准测试中显著优于全量数据训练及GEPA等强基线;仅用两个AIME 24提示训练出的系统提示即可在其他推理任务上良好泛化。

原文摘要 · Abstract (English)

Prompt optimization improves language models without updating their weights by searching for a better system prompt, but its effectiveness varies widely across tasks. We study what makes a task amenable to prompt optimization. We show that the reward variance across different system prompts can be decomposed into two components: variance among responses, which captures generation stochasticity, and variance among system prompts, which captures differences in system prompt quality. Prompt optimization succeeds when variance among system prompts is sufficiently large, but fails when variance among responses dominates the variance of the system prompts. Surprisingly, we further show that scaling to more user prompts can hurt optimization by reducing variance among system prompts, especially on heterogeneous datasets where different user prompts favor different system prompts. Motivated by this insight, we propose $p1$, a simple user prompt filtering method that selects a small subset of user prompts with high variance across candidate system prompts. This subset of user prompts allows one to distinguish a good system prompt from a bad one, making system optimization easier. Experiments on reasoning benchmarks show that $p1$ substantially improves prompt optimization over training on the full dataset and outperforms strong baselines such as GEPA. Notably, training on only two prompts from AIME 24 yields a system prompt that generalizes well to other reasoning benchmarks.

提示优化小样本推理增强

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