arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.18449cs.CLcs.AIcs.LG
收藏

通过扩散语言模型实现提示优化

Prompt Optimization Via Diffusion Language Models

Shiyu Wang, Haolin Chen, Liangwei Yang, Jielin Qiu, Rithesh Murthy, Ming Zhu, Zixiang Chen, Silvio Savarese, Caiming Xiong, Shelby Heinecke, Huan Wang

AI总结:

本文提出利用扩散语言模型实现提示优化,通过迭代精炼提升LLM性能。

摘要:

我们提出了一种基于扩散的提示优化框架,利用扩散语言模型(DLMs)通过掩码去噪迭代优化系统提示。通过条件于交互轨迹,包括用户查询、模型响应和可选反馈,我们的方法使提示能够灵活地进行跨度级更新,而无需访问梯度或修改下游语言模型。在多样化的基准测试(例如τ-bench、SST-2、SST-5)中,DLM优化的提示在冻结的目标LLM(如GPT-4o-mini)上持续提高性能。我们进一步表明,适度的扩散步数在精炼质量和稳定性之间提供了最佳平衡。这些结果突显了基于扩散的提示优化作为一种通用、模型无关且可扩展的方法,通过迭代提示优化来提升LLM性能。

英文摘要:

We propose a diffusion-based framework for prompt optimization that leverages Diffusion Language Models (DLMs) to iteratively refine system prompts through masked denoising. By conditioning on interaction traces, including user queries, model responses, and optional feedback, our method enables flexible, span-level prompt updates without requiring gradient access or modifying the downstream language model. Across diverse benchmarks (e.g., $τ$-bench, SST-2, SST-5), DLM-optimized prompts consistently improve the performance of a frozen target LLM (e.g., GPT-4o-mini). We further show that moderate diffusion step counts provide the best balance between refinement quality and stability. These results highlight diffusion-based prompt optimization as a general, model-agnostic, and scalable approach for enhancing LLM performance through iterative prompt refinement.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑