Automatic Prompt Optimization with Language Models

published on 07 June 2023

Welcome everyone to this dive into a recent paper on Natural Language Processing (NLP), titled "Automatic prompt optimization with gradient descent and beam search," authored by a team from Microsoft. The paper explores a new approach to automatically optimising the prompts of large language models, using LLMs to do the job. 

Image generated with Dall-E. 
Image generated with Dall-E. 

We also made a video about this paper, check it out here

The Power and Challenge of Prompts

In recent years, prompts have emerged as a critical tool for interacting with large language models (LLMs). They allow us to condition LLMs to perform specific tasks. By providing the LLM with instructions and examples through prompts, we can drive it to produce the desired output. This approach has proven successful across many applications such as chatbots and classification.

However, prompt engineering comes with its challenges. Typically, the process involves starting with a simple prompt, letting examples guide the construction, and gradually adding complexity. As you iterate, the prompt often becomes overly complex, and you can find yourself stuck, wondering how to improve it further.

Introducing a Framework for Prompt Optimization

This paper by Microsoft introduces a quantitative framework for prompt optimization. The authors suggest using a dataset of inputs and outputs to automatically tune and enhance a prompt. They discuss the dynamic nature of these models, emphasizing that a prompt that is ideal for one model might not work as well for another. With new versions of language models continually being released, like GPT-5 or other open-source models, the challenge of optimizing prompts is evident.

Their proposed solution involves using the language model itself to edit the prompt, a process they've coined as "prompt optimization with gradient descent." It mimics an optimization process, harnessing the power of LLMs to refine and improve the prompts.

How It Works: From Initial Prompt to Optimized Version

Let's break down this method. You begin with an initial prompt, say, for detecting a "jailbreak" attack in AI system protections. After providing a set of examples, an NLM computes "gradients" which guide how to edit the prompt to make it more effective.

From these gradients, new prompt candidates are generated, tweaking the original to make it better. They then employ a bandit selection algorithm to pick the best prompt for the task from these candidates. This cycle repeats several times, gradually evolving the prompts and improving their effectiveness.

The researchers experimented with several classification tasks like jailbreak, sarcasm detection, and others, and found improvements of up to 31% over the original prompt.

Comparisons and Experiments

The authors compared their method against various baselines, such as Monte-Carlo, reinforcement learning, AutoGPT, and evolutionary search. AutoGPT, a popular library on GitHub, is somewhat similar as it uses only prompts for improvement. However, the paper's method differs by using a supervised dataset for direct prompt improvement.

The Automatic Prompt Optimization (APO) method outperforms all other tested methods in terms of F1 score on classification tasks. The improvement can be quite significant, with performance jumps of nearly 20 percentage points in some cases.

The Future of Prompt Optimization

In essence, this paper offers a novel approach to prompt optimization, a problem faced by many engineers working with large language models today. Despite the widespread use of LLMs, prompt engineering still remains more art than science due to its lack of consistency and rigor.

The methods proposed in this paper promise to bring more systematization and rigor to the field, making it easier for engineers to achieve better results with less manual trial and error. As we continue to make strides in natural language processing, this type of thinking will be critical for maximizing the utility and effectiveness of large language models.

That's all for this review. Stay tuned for more insights

Read more