ModeSwitch-LLM: A Lightweight Phase-Aware Controller for Cross-Mode LLM Inference on a Single GPU
Abstract
ModeSwitch-LLM improves single-GPU LLM inference efficiency by dynamically routing requests to optimal fixed modes based on workload features, achieving significant latency speedup and energy savings while maintaining accuracy.
ModeSwitch-LLM is a lightweight request-boundary controller for improving single-GPU large language model inference efficiency by routing each request to an appropriate fixed inference mode. Instead of relying on one static serving configuration, the system selects among FP16, quantized modes, speculative decoding, and hybrid modes such as GPTQ plus prefix caching and INT8 plus continuous batching using cheap workload-level features. We evaluate ModeSwitch-LLM on Meta-Llama-3.1-8B-Instruct served on a single NVIDIA A100 GPU. On deployment-style synthetic workloads, the online controller achieves a 2.10x mean latency speedup over FP16 and a 0.48x mean energy ratio, corresponding to 51.7% lower energy per token. On automatic benchmarks used as a quality gate, accuracy remains close to FP16 with a mean delta of +0.17 percentage points. We also evaluate lightweight learned routers, but find that they do not clearly outperform the rule-based controller because they add routing overhead and more often select modes that violate quality, energy, or memory constraints. These results show that simple request-aware routing can recover substantial efficiency from existing inference modes without retraining the model or changing its architecture.
Community
Thanks for indexing our paper!
Code: https://github.com/ModeSwitch-LLM/ModeSwitch-LLM
arXiv: https://arxiv.org/abs/2605.23057
Zenodo: https://doi.org/10.5281/zenodo.20371757
ModeSwitch-LLM is a request-boundary controller for routing single-GPU LLM inference requests across FP16, GPTQ 4-bit, INT8 quantization, speculative decoding, prefix caching, continuous batching, and hybrid modes.
Headline results: 2.10× mean latency speedup, 0.48× energy ratio, +0.17 percentage-point mean benchmark accuracy delta, GPU memory usage close to FP16, and ~0.0096 ms CPU routing overhead per request on Meta-Llama-3.1-8B-Instruct with vLLM on one A100.
Get this paper in your agent:
hf papers read 2605.23057 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper