Conformal LLM Routing with Distribution-Free Safety Guarantees

Iqtedar Uddin, André Bauer


Abstract
LLM routing directs queries to a cheaper model when it suffices and to an expensive model otherwise, reducing inference cost. Existing input-based routers optimize cost-performance trade-offs but provide no formal bound on how often the cheaper model fails among routed queries. We adapt a proactive conformal gate framework to LLM routing. A logistic regression gate trained on text embeddings predicts per-query safety, and Clopper-Pearson conformal calibration selects a routing threshold that guarantees the violation rate among routed queries stays below 𝛼 (the violation tolerance) with probability at least 1 - 𝛿 (the confidence level). On two benchmarks covering math reasoning (GSM8K) and multi-domain knowledge (MMLU), routing between Mixtral-8x7B and GPT-4 (a 24.5× cost difference), our method maintains the target 𝛼 within the 𝛿 tolerance across a sweep from 0.05 to 0.50, while a validation-tuned baseline crosses the violation boundary on GSM8K. A feasibility analysis across all 10 RouterBench models reveals that routability is jointly model- and task-dependent. To our knowledge, this is the first input-based LLM router with distribution-free safety guarantees.
Anthology ID:
2026.acl-srw.70
Volume:
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 4: Student Research Workshop)
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Santosh T.Y.S.S., Juan Diego Rodriguez, Ona de Gibert
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
791–799
Language:
URL:
https://preview.aclanthology.org/ingestion-form-platform/2026.acl-srw.70/
DOI:
Bibkey:
Cite (ACL):
Iqtedar Uddin and André Bauer. 2026. Conformal LLM Routing with Distribution-Free Safety Guarantees. In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 4: Student Research Workshop), pages 791–799, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
Conformal LLM Routing with Distribution-Free Safety Guarantees (Uddin & Bauer, ACL 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingestion-form-platform/2026.acl-srw.70.pdf