Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Donghao Li, Chengshuai Shi, Weijuan Ou, Cong Shen, Jing Yang• 2026

Related benchmarks

TaskDatasetResultRank
Best feasible prompt identificationCNN/DailyMail (test)
Average Soft Constrained Reward0.172
72
Prompt OptimizationXsum
Hypervolume (HV)0.1626
72
Pareto prompt set identificationCNN/DailyMail
Hypervolume (HV)18.03
36
Soft constrained reward optimizationXSum on Gemma-7B
Average Soft Constrained Reward0.147
36
Best feasible prompt identificationXsum
Average Soft Constrained Reward (b=3)0.161
18
Showing 5 of 5 rows

Other info

Follow for update