English

RouteNLP: Closed-Loop LLM Routing with Conformal Cascading and Distillation Co-Optimization

Computation and Language 2026-04-28 v1 Machine Learning

Abstract

Serving diverse NLP workloads with large language models is costly: at one enterprise partner, inference costs exceeded $200K/month despite over 70% of queries being routine tasks well within the capability of smaller models. We present RouteNLP, a closed-loop framework that routes queries across a tiered model portfolio to minimize cost while satisfying per-task quality constraints. The framework integrates three components: a difficulty-aware router with shared task-conditioned representations trained on preference data and quality signals; confidence-calibrated cascading that uses conformal prediction for distribution-free threshold initialization; and a distillation-routing co-optimization loop that clusters escalation failures, applies targeted knowledge distillation to cheaper models, and automatically retrains the router, yielding over twice the cost improvement of untargeted distillation. In an 8-week pilot deployment processing ~5K queries/day at an enterprise customer-service division, RouteNLP reduced inference costs by 58% while maintaining 91% response acceptance and reducing p99 latency from 1,847 ms to 387 ms. On a six-task benchmark spanning finance, customer service, and legal domains, the framework achieves 40-85% cost reduction while retaining 96-100% quality on structured tasks and 96-98% on generation tasks, with human evaluation confirming that 74.5% of routed generation outputs match or exceed frontier-model quality.

Keywords

Cite

@article{arxiv.2604.23577,
  title  = {RouteNLP: Closed-Loop LLM Routing with Conformal Cascading and Distillation Co-Optimization},
  author = {Dongxin Guo and Jikun Wu and Siu Ming Yiu},
  journal= {arXiv preprint arXiv:2604.23577},
  year   = {2026}
}

Comments

Accepted at ACL 2026 Industry Track. 13 pages, 2 figures, 15 tables, 1 algorithm

R2 v1 2026-07-01T12:35:34.433Z