Supervisory Prompt Training
Jean Ghislain Billa, Min Oh, Liang Du
TL;DR
SPT introduces a self-improving, dual-LLM framework in which a generator and a corrector iteratively refine prompts to boost performance and reduce hallucinations. It adds impact scores to quantify sentence-level contributions, guiding subsequent improvements and enabling the corrector to evolve its feedback. Across four benchmarks, SPT yields substantial gains—most notably a $28.3$-point GSM8K improvement for GPT-4—and demonstrates that refining prompts can rival traditional fine-tuning as a scalable alternative. The approach advances prompt-based optimization, offering a practical path to more reliable LLM behavior in high-stakes and diverse tasks.
Abstract
The performance of Large Language Models (LLMs) relies heavily on the quality of prompts, which are often manually engineered and task-specific, making them costly and non-scalable. We propose a novel approach, Supervisory Prompt Training (SPT). SPT automates the generation of highly effective prompts using a dual LLM system. In this system, one LLM, the generator, performs a task while the other, the corrector, provides feedback and generates improved prompts. In contrast to earlier techniques, both the generator and corrector collaboratively and continuously improve their prompts over time. We also introduce the concept of \textit{impact scores} to measure the sentence-level effectiveness of the prompts. Our method was tested on four benchmarks, testing the level of hallucinations in LLMs. Notably, we were able to increase the accuracy of GPT-4 on GSM8K from 65.8\% to 94.1\% (28.3\% increase). SPT advances LLMs by refining prompts to enhance performance and reduce hallucinations, offering an efficient and scalable alternative to traditional model fine-tuning.
