Symbolic Prompt Program Search: A Structure-Aware Approach to Efficient Compile-Time Prompt Optimization

Document detail

ID

oai:arXiv.org:2404.02319

Topic

Computer Science - Computation and... Computer Science - Artificial Inte... Computer Science - Machine Learnin...

Author

Schnabel, Tobias Neville, Jennifer

Year

2024

listing date

7/3/2024

Keywords

sammo programs

Metrics

Abstract

In many modern LLM applications, such as retrieval augmented generation, prompts have become programs themselves.

In these settings, prompt programs are repeatedly called with different user queries or data instances.

A big practical challenge is optimizing such prompt programs.

Recent work has mostly focused on either simple prompt programs or assumed that the general structure of a prompt program is fixed.

We introduce SAMMO, a framework to perform symbolic prompt program search for compile-time optimizations of prompt programs.

SAMMO represents prompt programs on a symbolic level which allows for a rich set of transformations that can be searched over during optimization.

We show that SAMMO generalizes previous methods and improves the performance of complex prompts on (1) instruction tuning, (2) RAG pipeline tuning, and (3) prompt compression, across several different LLMs.

We make all code available open-source at https://github.com/microsoft/sammo .

Schnabel, Tobias,Neville, Jennifer, 2024, Symbolic Prompt Program Search: A Structure-Aware Approach to Efficient Compile-Time Prompt Optimization

Document

Open

Source

Articles recommended by ES/IODE AI

Computer Science