Search papers, labs, and topics across Lattice.
This paper explores Neural Upper Confidence Bound (NeuralUCB) for online, cost-aware routing of LLM requests to different models. They address the limitations of supervised and partial-feedback routing methods by framing LLM selection as a contextual bandit problem. Experiments on RouterBench demonstrate that NeuralUCB outperforms random and min-cost baselines, achieving competitive reward with significantly lower inference costs compared to a max-quality reference model.
NeuralUCB can slash LLM inference costs while maintaining quality, offering a practical alternative to always using the biggest, most expensive models.
This study investigates the use of NeuralUCB for cost-aware large language model (LLM) routing. Existing routing approaches can be broadly grouped into supervised routing methods and partial-feedback methods, each with different tradeoffs in efficiency and adaptivity. We implement a NeuralUCB-based routing policy and evaluate it on RouterBench under a simulated online setting. Experimental results show that the proposed method consistently outperforms random and min-cost baselines in utility reward. Compared with the max-quality reference, our method achieves substantially lower inference cost while maintaining competitive reward. These findings suggest that NeuralUCB is a promising approach for cost-aware LLM routing, while also highlighting remaining challenges in action discrimination and exploration.
