FlexRouter: Learning Complementary Model Sets for Flexible LLM Routing

Hugging Face Daily Papers Papers

Summary

FlexRouter proposes a coverage-oriented LLM routing framework that uses Determinantal Point Processes to model complementarity among models, maximizing the probability that at least one selected model answers correctly while avoiding redundant selections and fixed budgets.

Existing Large Language Model (LLM) routing methods score LLMs independently to select top-k models. However, this ignores model correlations and enforces a rigid computational budget. Consequently, routers often select redundant models that share failure modes, limiting the overall probability of success. To address this, we propose FlexRouter, a routing framework that explicitly models model complementarity. FlexRouter optimizes for answer coverage, maximizing the probability that at least one selected model yields a correct response. This objective aligns with practical inference pipelines where multiple candidate outputs are generated and a downstream verifier or user selects the final one. We formulate routing as a coverage-oriented subset selection problem and model the routing policy using Determinantal Point Processes (DPPs), which naturally capture both model competence and redundancy. To directly optimize coverage without requiring a ground-truth target subset, we introduce a training objective based on marginalizing over failure sets. During inference, we employ a greedy strategy based on marginal log-determinant gains, enabling the router to adaptively determine subset sizes without a predefined budget. Extensive experiments on the large-scale RouterEval benchmark demonstrate that our proposed FlexRouter achieves higher coverage with lower redundancy across both in-domain and out-of-domain tasks than strong baselines while maintaining flexible inference cost.
Original Article
View Cached Full Text

Cached at: 10/02/26, 04:26 AM

Paper page - FlexRouter: Learning Complementary Model Sets for Flexible LLM Routing

Source: https://huggingface.co/papers/2609.38585

Abstract

ExistingLargeLanguageModel(LLM)routingmethodsscoreLLMsindependentlytoselecttop-kmodels.However,thisignoresmodelcorrelationsandenforcesarigidcomputationalbudget.Consequently,routersoftenselectredundantmodelsthatsharefailuremodes,limitingtheoverallprobabilityofsuccess.Toaddressthis,weproposeFlexRouter,aroutingframeworkthatexplicitlymodelsmodelcomplementarity.FlexRouteroptimizesforanswercoverage,maximizingtheprobabilitythatatleastoneselectedmodelyieldsacorrectresponse.Thisobjectivealignswithpracticalinferencepipelineswheremultiplecandidateoutputsaregeneratedandadownstreamverifieroruserselectsthefinalone.Weformulateroutingasacoverage-orientedsubsetselectionproblemandmodeltheroutingpolicyusingDeterminantalPointProcesses(DPPs),whichnaturallycapturebothmodelcompetenceandredundancy.Todirectlyoptimizecoveragewithoutrequiringaground-truthtargetsubset,weintroduceatrainingobjectivebasedonmarginalizingoverfailuresets.Duringinference,weemployagreedystrategybasedonmarginallog-determinantgains,enablingtheroutertoadaptivelydeterminesubsetsizeswithoutapredefinedbudget.Extensiveexperimentsonthelarge-scaleRouterEvalbenchmarkdemonstratethatourproposedFlexRouterachieveshighercoveragewithlowerredundancyacrossbothin-domainandout-of-domaintasksthanstrongbaselineswhilemaintainingflexibleinferencecost.

View arXiv pageView PDFAdd to collection

Get this paper in your agent:

hf papers read 2609\.38585

Don’t have the latest CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash

Models citing this paper0

No model linking this paper

Cite arxiv.org/abs/2609.38585 in a model README.md to link it from this page.

Datasets citing this paper0

No dataset linking this paper

Cite arxiv.org/abs/2609.38585 in a dataset README.md to link it from this page.

Spaces citing this paper0

No Space linking this paper

Cite arxiv.org/abs/2609.38585 in a Space README.md to link it from this page.

Collections including this paper0

No Collection including this paper

Add this paper to acollectionto link it from this page.

Similar Articles

FlexRouter: Learning Complementary Model Sets for Flexible LLM Routing

arXiv cs.LG

FlexRouter 提出一种显式建模模型互补性的 LLM 路由框架,将路由建模为基于覆盖的子集选择问题,并用 Determinantal Point Processes 对模型能力与冗余进行联合建模,同时通过边缘化失败集合的目标函数直接优化答案覆盖率。在 RouterEval 基准上,该方法在域内与域外任务上以更低冗余实现了更高的覆盖率,且推理成本灵活可控。

Pretrain Once, Route Anywhere: Towards a Foundation Model for LLM Routing

Hugging Face Daily Papers

The paper introduces RouteFM, a foundation model for LLM routing that learns reusable routing capabilities via episodic pretraining across heterogeneous environments, allowing a frozen router to adapt to new domains, modalities, and candidate pools through behavioral context alone. RouteFM outperforms the strongest baseline by 2.23 quality points on MMR-Bench with only eight observations per candidate, supporting a 'pretrain once, route anywhere' paradigm.