arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2302.11520cs.CL
收藏

通过方向性刺激提示引导大型语言模型

Guiding Large Language Models via Directional Stimulus Prompting

University of California Santa Barbara(加州大学圣巴巴拉分校);Microsoft(微软);Microsoft Research(微软研究院)

Zekun Li, Baolin Peng, Pengcheng He, Michel Galley, Jianfeng Gao, Xifeng Yan

AI总结:

本文提出方向性刺激提示框架,通过小型策略模型生成实例特定提示来引导黑盒LLM输出,在摘要、对话和推理任务上以极少标注数据显著提升性能。

摘要:

我们引入了方向性刺激提示(Directional Stimulus Prompting),这是一种用于引导黑盒大型语言模型(LLM)产生特定期望输出的新型框架。与直接调整LLM不同,我们的方法采用一个可调节的小型策略模型(例如T5)为每个输入实例生成一个辅助性的方向性刺激提示。这些方向性刺激提示充当细微的、实例特定的提示和线索,以引导LLM生成期望的结果,例如在生成的摘要中包含特定关键词。我们的方法通过优化策略模型来探索与LLM期望行为对齐的方向性刺激提示,从而规避了直接调整LLM的挑战。策略模型可以通过以下两种方式进行优化:1)使用标注数据进行监督微调;2)基于LLM输出的离线或在线奖励进行强化学习。我们在摘要生成、对话响应生成和思维链推理任务上评估了我们的方法。我们的实验表明,该框架使用最少的标注数据持续提升了LLM(如ChatGPT、Codex、InstructGPT)在这些监督任务上的性能。值得注意的是,在MultiWOZ数据集上仅使用80个对话,我们的方法就将ChatGPT的性能提升了41.4%,达到或超过了某些完全监督的最先进模型。此外,与人工设计或自动生成的提示相比,我们方法生成的实例特定思维链提示提高了InstructGPT的推理准确性。代码和数据公开于\url{https://github.com/Leezekun/Directional-Stimulus-Prompting}。

英文摘要:

We introduce Directional Stimulus Prompting, a novel framework for guiding black-box large language models (LLMs) toward specific desired outputs. Instead of directly adjusting LLMs, our method employs a small tunable policy model (e.g., T5) to generate an auxiliary directional stimulus prompt for each input instance. These directional stimulus prompts act as nuanced, instance-specific hints and clues to guide LLMs in generating desired outcomes, such as including specific keywords in the generated summary. Our approach sidesteps the challenges of direct LLM tuning by optimizing the policy model to explore directional stimulus prompts that align LLMs with desired behaviors. The policy model can be optimized through 1) supervised fine-tuning using labeled data and 2) reinforcement learning from offline or online rewards based on the LLM's output. We assess our method across summarization, dialogue response generation, and chain-of-thought reasoning tasks. Our experiments demonstrate that the framework consistently improves LLMs' (e.g., ChatGPT, Codex, InstructGPT) performance on these supervised tasks using minimal labeled data. Notably, using just 80 dialogues on the MultiWOZ dataset, our approach enhances ChatGPT's performance by an impressive 41.4%, matching or surpassing some fully supervised start-of-the-art models. Additionally, the instance-specific chain-of-thought prompt generated by our approach improves InstructGPT's reasoning accuracy compared to human-crafted or automatically generated prompts. The code and data are publicly available at \url{https://github.com/Leezekun/Directional-Stimulus-Prompting}.

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

↑