Work

Deputy: Accelerating Large Language Model Inference with Dynamic Low-Rank Substitution

Year: 2026
Type: conference-paper
Abstract: While the massive scale of modern LLMs enables remarkable performance, their static, input-agnostic computational graph incurs substantial resource wastage and high latency during inference.Existing dynamic schemes, such as early-exit and layer-drop reduce FLOPs but break batch processing or introduce KV-cache inconsistency.We propose Deputy, a dynamic low-rank substitution framework that employs a lightweight decision module at each layer to dynamically determine the execution branch for different tokens: Attention layers choose between full and low-rank computation to mitigate the KV cache issue, while FFN layers additionally support skipping to further reduce computation.We fine-tune the LLM with LoRA and then derive an additional low-rank matrix C via a least-squares fit BC ≈ W pre , where B is the shared LoRA matrix, so that only one extra low-rank matrix is introduced, effectively reducing memory overhead.Moreover, a hybrid KV cache strategy stores KV values generated by the low-rank branch, achieving a 38% reduction in cache storage.Experiments on Llama models demonstrate that Deputy reduces computation by approximately 40% compared to the original dense model while outperforming existing baseline methods.The code is available at https://github.com/yuhua-zhou/Deputy.
Language: English
Open Access status: hybrid
OpenAlex ID: W7166844486
FWCI:
Cites:
Cited by:
Related to: 10