Cost-Aware Best-LLM Identification using Dueling Feedback
Organizations: Centre for Machine Intelligence and Data Science (C-MInDS) Indian Institute of Technology Bombay · Department of Electrical Engineering Indian Institute of Technology Bombay
Abstract
Inspired by the problem of identifying the best model from a collection of large language models (LLMs) with heterogeneous querying costs, we formulate and analyse a variant of the multi-armed bandit (MAB) with (i) dueling feedback, where pairwise comparisons between model responses provide robust preference signals, and (ii) heterogeneous sampling costs, reflecting the differing costs of querying different LLMs. Assuming the existence of a Condorcet winner, a condition we empirically validate across multiple real-world datasets, we propose a Track-and-Stop style algorithm for best-arm identification with prescribed confidence. We prove that the algorithm almost surely achieves the asymptotically optimal cost as the error tends to zero. Finally, we extensively evaluate our approach on both synthetic and real-world instances, demonstrating consistent improvements over classical cost-unaware algorithms and their cost-aware extensions.
Figures & tables
| Arm Pairs | Cost Unaware | Cost Aware |
|---|---|---|
| 1 and 2 | ||
| 1 and 3 | ||
| 2 and 3 |
| T2I | T2T | Vision | Search | |
|---|---|---|---|---|
| DCTAS | ||||
| TAS | ||||
| DPCA | ||||
| CRR | ||||
| DCTAC |
| T2I | T2T | Vision | Search | |
|---|---|---|---|---|
| DCTAS | ||||
| TAS | ||||
| reduction |
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| Assumption | Text-to-Image | Text-to-Text | Vision | Search |
|---|---|---|---|---|
| LNM | ||||
| TO | ||||
| SST | ||||
| -RST | ||||
| MST | ||||
| WST |
| Assumption | Violating Example |
|---|---|
| Low Noise Model | i=C, j=B |
| Strong Stochastic Transitivity | i=A, j=D, k=C |
| Relaxed Stochastic Transitivity | i=B, j=D, k=C |
| Moderate Stochastic Transitivity | i=B, j=D, k=C |
| Weak Stochastic Transitivity | i=B, j=D, k=C |
| Assumption | Violating Example |
|---|---|
| Low Noise Model | i=DeepSeek R1, j = DeepSeek V3 |
| Strong Stochastic Transitivity | i=LLAMA 4 Maverick, j=DeepSeek R1, k=DeepSeek V3 |
| Stochastic Triangle Inequality | i=LLAMA 4 Maverick, j=Gemini 2.0 Flash, k=Gemma 3 27b |
| Assumption | Violating Example |
|---|---|
| Low Noise Model | i = Gemini 2.0 Flash j = Gemini 1.5 Pro |
| Strong Stochastic Transitivity | i=Gemini 2.5 Pro Exp, j=Gemini 2.0 Flash, k=Gemini 1.5 Pro |
| Relaxed Stochastic Transitivity | i=Gemini 2.0 Flash, j=Gemini 1.5 Pro, k=Claude 3.7 Sonnet |
| Moderate Stochastic Transitivity | i=Gemini 2.0 Flash, j=Gemini 1.5 Pro, k=Claude 3.7 Sonnet |
| Weak Stochastic Transitivity | i=Gemini 2.0 Flash, j=Gemini 1.5 Pro, k=Claude 3.7 Sonnet |
| Assumption | Violating Example |
|---|---|
| Low Noise Model | i=PPL Sonar Pro, j=PPL Sonar |
| Strong Stochastic Transitivity | i=PPL Reasoning Pro, j=PPL Reasoning, k=PPL Sonar |
| Moderate Stochastic Transitivity | i=PPL Reasoning Pro, j=PPL Reasoning, k=PPL Sonar |
| Stochastic Triangle Inequality | i=PPL Reasoning Pro, j=PPL Sonar, k=GPT 4O Search High |
| Assumption | Chatbot Arena | Text to Video | Text to Image |
|---|---|---|---|
| LNM | |||
| TO | |||
| SST | |||
| -RST | |||
| MST | |||
| WST |
| Assumption | Violating Example |
|---|---|
| Low Noise Model | i = chatglm-6b, j = gpt4all-13b-snoozy |
| Strong Stochastic Transitivity | i=RWKV-4-Raven-14B, j=alpaca-13b, k=dolly-v2-12b |
| Relaxed Stochastic Transitivity | i=RWKV-4-Raven-14B, j=chatglm-6b, k=gpt4all-13b-snoozy |
| Moderate Stochastic Transitivity | i=RWKV-4-Raven-14B, j=chatglm-6b, k=gpt4all-13b-snoozy |
| Weak Stochastic Transitivity | i=RWKV-4-Raven-14B, j=chatglm-6b, k=gpt4all-13b-snoozy |
| Assumption | Violating Example |
|---|---|
| Low Noise Model | i = AnimateDiff, j = VideoCrafter2 |
| Strong Stochastic Transitivity | i=AnimateDiff, j=VideoCrafter2, k=OpenSora |
| Relaxed Stochastic Transitivity | i=AnimateDiff, j=VideoCrafter2, k=T2VTurbo |
| Moderate Stochastic Transitivity | i=AnimateDiff, j=VideoCrafter2, k=T2VTurbo |
| Weak Stochastic Transitivity | i=AnimateDiff, j=VideoCrafter2, k=T2VTurbo |
| Assumption | Violating Example |
|---|---|
| Strong Stochastic Transitivity | i=PixArtAlpha, j=PixArtSigma, k=OpenJourney |
| Moderate Stochastic Transitivity | i=PixArtSigma, j=SDXLLightning, k=SDXLTurbo |
| Stochastic Triangle Inequality | i=PlayGroundV2.5, j=PixArtSigma, k=SDXLLightning |