← 返回论文检索
ICML 2026PosterAccept (regular)

PACE: Proactive Agent-Level Admission Control for Efficient Agentic Batch Inference

Qiaoling Chen, Zhisheng Ye, Tian Tang, Peng Sun, Boyu Tian, Guoteng Wang, Shenggui Li, Zhenhua Han, Yonggang Wen, Tianwei Zhang

Nanyang Technological University · Independent Researcher · Shanghai Qiji Zhifeng Co., Ltd · Shanghai Qiji Zhifeng Co., Ltd. · Shanghai Artificial Intelligence Laboratory · Microsoft · Nanyang Technological University, Singapore

PDF 由论文原始站点提供,PaperCompass 不保存论文文件。

摘要

Batch inference for agentic workloads stresses the GPU key–value (KV) cache in a sustained and cumulative manner, often causing severe throughput degradation well before memory capacity is exhausted. We identify this phenomenon as middle-phase thrashing, a previously under-characterized pathology in which cache efficiency collapses as long-lived agents accumulate state over time. We argue that mitigating this pathology requires moving beyond reactive, request-level cache management to proactive, agent-level admission control. Drawing inspiration from congestion control in distributed systems, we view the KV cache as a shared resource whose efficient utilization depends on feedback-driven regulation. Based on this insight, we present PACE, a lightweight control layer that regulates agent admission to bound aggregate cache pressure while preserving execution continuity. PACE adapts a cache-aware control algorithm to dynamically adjust the number of active agents using runtime cache signals. Across large models and real-world agent workloads, PACE prevents middle-phase thrashing and improves batch inference throughput by up to 4.09× on Qwen3-32B and 1.90× on DeepSeek-V3, while remaining compatible with existing LLM serving systems.