Efficient Multi-objective Prompt Optimization via Pure-exploration Bandits
About
Prompt engineering has become central to eliciting the capabilities of large language models (LLMs). At its core lies prompt selection -- efficiently identifying the most effective prompts. However, most prior investigations overlook a key challenge: the inherently multi-faceted nature of prompt performance, which cannot be captured by a single metric. To fill this gap, we study the multi-objective prompt selection problem under two practical settings: Pareto prompt set recovery and best feasible prompt identification. Casting the problem into the pure-exploration bandits framework, we adapt provably efficient algorithms from multi-objective bandits and further introduce a novel design for best feasible arm identification in structured bandits, with theoretical guarantees on the identification error in the linear case. Extensive experiments across multiple LLMs show that the bandit-based approaches yield significant improvements over baselines, establishing a principled and efficient framework for multi-objective prompt optimization.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Best feasible prompt identification | CNN/DailyMail (test) | Average Soft Constrained Reward0.172 | 72 | |
| Prompt Optimization | Xsum | Hypervolume (HV)0.1626 | 72 | |
| Pareto prompt set identification | CNN/DailyMail | Hypervolume (HV)18.03 | 36 | |
| Soft constrained reward optimization | XSum on Gemma-7B | Average Soft Constrained Reward0.147 | 36 | |
| Best feasible prompt identification | Xsum | Average Soft Constrained Reward (b=3)0.161 | 18 |