cs.DSMar 24, 2026

Multi-LLM Query Optimization

Authors: Arlen Dean, Zijin Zhang, Stefanus Jasin, Yuqing Liu

Organizations: Washington University in St. Louis, Olin Business School · Boston College, Carroll School of Management · University of Michigan, Stephen M. Ross School of Business · University of Michigan, Industrial and Operations Engineering

Abstract

Deploying multiple large language models (LLMs) in parallel to classify an unknown ground-truth label is a common practice, yet the problem of optimally allocating queries across heterogeneous models remains poorly understood. In this paper, we formulate a robust, offline query-planning problem that minimizes total query cost subject to statewise error constraints which guarantee reliability for every possible ground-truth label. We first establish that this problem is NP-hard via a reduction from the minimum-weight set cover problem. To overcome this intractability, we develop a surrogate by combining a union bound decomposition of the multi-class error into pairwise comparisons with Chernoff-type concentration bounds. The resulting surrogate admits a closed-form, multiplicatively separable expression in the query counts and is guaranteed to be feasibility-preserving. We further show that the surrogate is asymptotically tight at the optimization level: the ratio of surrogate-optimal cost to true optimal cost converges to one as error tolerances shrink, with an explicit rate of O(log⁡log⁡(1/αmin⁡)/log⁡(1/αmin⁡))O\left(\log\log(1/α_{\min}) / \log(1/α_{\min})\right). Finally, we design an asymptotic fully polynomial-time approximation scheme (AFPTAS) that returns a surrogate-feasible query plan within a (1+ε)(1+\varepsilon) factor of the surrogate optimum.

Explore similar work

CardsList
  1. Circuits of LLM Classifiers

    Oct 21, 2026Hadi Khalaf, Sajani Vithana, Kunal Talreja +3Large Language Model-Based Text ClassificationLanguage Model Ensembles

  2. Multiple-Prediction-Powered Inference

    Mar 28, 2026Charlie Cowen-Breen, Alekh Agarwal, Stephen Bates +4Prediction-Powered Inference