Search papers, labs, and topics across Lattice.
This paper introduces VDAR-Router, a novel routing framework for large language models (LLMs) that incorporates an explicit analysis of query difficulty to enhance model selection. By retrieving historical examples that match the difficulty profiles of incoming queries, VDAR-Router optimizes the cost-performance trade-off in LLM deployment. Experimental results across three datasets reveal that this difficulty-aware approach outperforms existing routing methods, leading to more efficient and reliable model allocations.
Difficulty-aware routing can significantly enhance LLM performance by ensuring that models are matched to queries based on their complexity rather than just surface semantics.
Large language models are increasingly used in practical systems, making efficient model selection important for reducing deployment cost. LLM routing has emerged as a practical solution for allocating each input query to an appropriate model under a desired cost-performance trade-off. Existing routing methods often estimate model suitability from the surface semantics or embedding similarity of the input query. However, such methods may ignore the underlying difficulty of a query, leading to suboptimal routing decisions. To address the challenge, we propose VDAR-Router, a difficulty-aware retrieval-based routing framework. For each input query, VDAR-Router first generates an explicit difficulty analysis. It then retrieves historical examples with similar difficulty profiles. Based on the retrieved records, it estimates candidate model suitability and selects the model using a reward function that considers both performance and cost. Experiments on three datasets show that VDAR-Router consistently achieves better cost-performance trade-offs than existing baselines. These results demonstrate the effectiveness of difficulty-aware retrieval for training-free LLM routing. Case studies further show that explicit query analysis helps retrieve more relevant examples and supports more reliable routing decisions.
