Hao Mark Chen
Author directory2026
AgentKV: Phase-Aware KV Eviction for Agentic LLMs
Taowen Tony Liu | Jeffrey T. H. Wong | Can Xiao | Bowen Yang | Hao Mark Chen | Yiren Zhao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Taowen Tony Liu | Jeffrey T. H. Wong | Can Xiao | Bowen Yang | Hao Mark Chen | Yiren Zhao
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Agentic serving can consume orders of magnitude more tokens than chatbot workloads, stressing both KV-cache capacity and decode-time bandwidth. Most KV-eviction methods score cached keys against representative queries drawn from the most recent tokens, assuming future attention resembles recent attention. We show that agentic generation violates this assumption: future queries form a mixture over think, act, tool, and others phases, and principal-angle analysis shows these components occupy measurably different query subspaces, so recency representatives systematically undervalue keys that upcoming phases will need. We propose AgentKV, which maintains a small query buffer per phase and scores cached keys against their union. We further implement AgentKV in a persistent multi-turn serving path that carries compressed KV state across turns and compacts retained KV pages online. Across two models, six task domains, and three KV budgets each, AgentKV improves task score by 5.5 points on average over R-KV and 5.3 over Tri-attention. Relative to upstream full-KV SGLang, AgentKV improves output-token throughput by up to 1.80×. Code: https://github.com/LiuTaowen-Tony/agentkv.
2025
Hardware-Aware Parallel Prompt Decoding for Memory-Efficient Acceleration of LLM Inference
Hao Mark Chen | Wayne Luk | Yiu Ka Fai Cedric | Rui Li | Konstantin Mishchenko | Stylianos Venieris | Hongxiang Fan
Findings of the Association for Computational Linguistics: EMNLP 2025
Hao Mark Chen | Wayne Luk | Yiu Ka Fai Cedric | Rui Li | Konstantin Mishchenko | Stylianos Venieris | Hongxiang Fan
Findings of the Association for Computational Linguistics: EMNLP 2025
The auto-regressive decoding of Large Language Models (LLMs) results in significant overheads in their hardware performance. While recent research has explored various speculative decoding techniques for multi-token generation, these methods introduce high memory costs from the additional weights and KV cache of separate draft models, limiting efficiency in edge and long-context scenarios. To overcome these limitations in edge-scale LLMs, we propose a novel parallel prompt decoding that requires only runtime memory overhead by employing a unified single model for both speculation and verification. Inspired by the human natural language generation process, PPD approximates outputs generated at future timesteps in parallel by using multiple prompt tokens. Furthermore, we present a hardware-aware two-stage tree pruning algorithm that adaptively optimizes this decoding scheme to fully leverage the computational capacities on different GPUs. Through extensive experiments across LLMs ranging from MobileLlama to Vicuna-13B on a wide range of benchmarks, our approach demonstrates up to 2.49 times speedup. Moreover, our parallel prompt decoding can serve as an orthogonal optimization for synergistic integration with existing speculative decoding, showing up to 1.22 times further speed improvement. To support future development, we have included our code implementation with this submission.