Skip to main content

Routing Strategies & Feature Modeling

Paper: CIKM Short Paper - Knowing When to Correct: Cost-Aware LLM Routing
Authors: Stergios Konstantinidis, Hayman Lotfy, Michalis Vlachos (University of Lausanne)
Venue: ACM International Conference on Information and Knowledge Management (CIKM) - Short Paper
GitHub: https://github.com/Stergios-Konstantinidis/CIKM_public
Overleaf: https://git.overleaf.com/6a1d78e803cbdf7def32a839


Cost-Aware Routing Strategies & Models

1. Feature Engineering (54 Features)

The routing engine extracts 54 computationally inexpensive features prior to invoking any LLM:

  1. OCR Confidence Metrics: Mean confidence, minimum token confidence, standard deviation, count of low-confidence tokens (<80%, <50%).
  2. Lexical & Linguistic Features: Out-of-vocabulary (OOV) ratio against historical lexicon, archaic character frequency (e.g. ſ, œ, ligature anomalies).
  3. Statistical Text Metrics: Punctuation density, digit-to-letter ratios, average word length, uppercase token anomalies.
  4. Layout Context: Bounding box coordinates, line height variance, bounding box density.

2. Evaluated Router Models

The router was benchmarked across multiple learning paradigms:

  • Lasso Regression (L1): Sparse feature selection, highly interpretable, $<0.1$1s inference time on single CPU core.
  • Ridge Regression (L2): Smooth shrinkage across correlated confidence features.
  • Support Vector Machines (SVM): Non-linear RBF kernel separating degradation clusters.
  • Multi-Layer Perceptron (NN): 2-layer feedforward network predicting expected CER reduction.
  • ConfBERT: Fine-tuned lightweight language model scoring token perplexity and error likelihood.

3. Results Summary

  • Selective Efficiency: At $\tau = 0$0, the Lasso router skips 40.2% of segments, reducing API costs by 38.5% while degrading final CER by only 0.20 percentage points (3.12% vs 2.92%).
  • Oracle Tracking: The regression router achieves $91.4%$4\% of the theoretical maximum gain achievable by an omniscient oracle.

Empirical Routing Curves & Threshold Analysis

Regression-Based Routing Performance

[![

cikm_regression_routing.png

Figure: cikm_regression_routing.png](https://wiki.stergios.ch/uploads/images/gallery/2026-09/scaled-1680-/cikm-regression-routing.png)](https://wiki.stergios.ch/uploads/images/gallery/2026-09/cikm-regression-routing.png)png

Figure 1: Cost vs accuracy trade-off curves for regression routing across prompt regimes.

Threshold Tuning & Boundary Optimization

[![

cikm_threshold_comparison.png

Figure: cikm_threshold_comparison.png](https://wiki.stergios.ch/uploads/images/gallery/2026-09/scaled-1680-/cikm-threshold-comparison.png)](https://wiki.stergios.ch/uploads/images/gallery/2026-09/cikm-threshold-comparison.png)png

Figure 2: Optimal confidence threshold cutoffs minimizing inference cost while maximizing CER delta.