English

Beyond Query Memorization: Large Language Model Routing with Query Decomposition and Historical Matching

Artificial Intelligence 2026-05-26 v1

Abstract

Optimizing the trade-off among predictive performance and computational cost is a central focus in the deployment of Large Language Models (LLMs). Current routing methods primarily rely on direct mapping from queries to models based on surface-level features, making them susceptible to the memorization trap and leading to poor generalizability on out-of-distribution (OOD) data. In this paper, we propose DecoR, a novel routing framework that recasts the routing task as a matching process of sifting similar queries from historical logs, effectively mitigating the memorization trap. To enhance matching accuracy, we introduce a query capability deconstruction method that decouples linguistic surface forms from task-intrinsic requirements, directing matching toward capability dimensions to ground decisions in essential task attributes. Furthermore, we develop CodaSet, a comprehensive benchmark for assessing routing generalization, where experimental results demonstrate that DecoR maintains superior accuracy while substantially lowering inference costs across both in-distribution and OOD settings. All the codes and data are available at https://github.com/lvbotenbest/DecoR.

Keywords

Cite

@article{arxiv.2605.25558,
  title  = {Beyond Query Memorization: Large Language Model Routing with Query Decomposition and Historical Matching},
  author = {Bo Lv and Jingbo Sun},
  journal= {arXiv preprint arXiv:2605.25558},
  year   = {2026}
}
R2 v1 2026-07-22T07:32:01.269Z