AI & ML
KVGuard: A KV Cache Safety Checker for LLM Serving Systems
arXiv cs.CR0 views
arXiv:2611.06138v1 Announce Type: new Abstract: Modern LLM serving systems manage the KV cache through PagedAttention, a custom GPU allocator that partitions a single CUDA-allocated region into fixed-size physical blocks and maps each request to those blocks through a CPU-built block table. This block-based design exposes a previously unstudied class of memory safety bugs in the KV cache, including double-free, use-after-free, out-of-bounds access, uninitialized read, and permission violation, which silently corrupt outputs or break request isolation without ever crashing the system. Existing memory sanitizers cannot detect them, because they reason about CUDA-level addresses rather than KV cache blocks, requests, and block tables, and their redzone- and quarantine-based designs do not fit the scarce GPU memory budget of LLM serving. This paper presents KVGuard, the first systematic memory-safety abstraction for KV cache management in LLM serving systems. KVGuard maintains a compact shadow block for each physical KV block, recording per-request permission and per-entry initialization, and uses this metadata to perform five safety checks across the PagedAttention workflow, spanning CPU allocation, block table construction and translation, and GPU-side block table lookup and KV entry access. Each check fires at the boundary that introduced the violation, enabling early detection during development rather than reporting downstream output corruption many requests later. A prototype of KVGuard on vLLM and SGLang covers all 29 historical real-world KV cache safety bugs collected from the two systems' patch histories. KVGuard further found 8 previously unknown safety violations while reporting no false positives. It adds 3.0% and 7.4% geomean TPOT overhead on Qwen 2.5 14B on vLLM and SGLang and at most 0.11% of the KV cache in memory overhead, two orders of magnitude below NVIDIA Compute Sanitizer on the same workloads.
Source:https://arxiv.org/abs/2611.06138PreviousSenTrojan: Revealing Analog Hardware Trojan Attacks on SensorsNextNetwork Intrusion Forensic Detection System with Explainable Artificial Intelligence and Digital Chain of Custody
Related posts
How Chiron Changes AI Incident Investigation on ORCA-bench
0Hacker News New
Ask HN: How do small teams pragmatically manage Dependabot fatigue?
0Hacker News New
How WeRateDogs Is Fighting A.I. Slop, One Perfect Puppy at a Time
0Hacker News New
Piston – small causal model for time series that can refuse
0Hacker News New
Novel AI framework runs on optical hardware
0Hacker News New
Nexus-Q: the first commits were a threat model and architecture, not code
0Hacker News New