FoveatedKV hits 2x KV compression for Mac
FoveatedKV brings VR-inspired foveated rendering to LLM KV caches, achieving 2x memory reduction and up to 2.3x faster 7B inference on 8GB Apple Silicon devices. By partitioning the cache into importance-based tiers and utilizing custom Metal kernels, it maintains 0.995+ fidelity even at high compression ratios.
FoveatedKV is a masterclass in exploiting Apple Silicon's shared memory architecture and Metal performance for LLM optimization. It employs a mixed-precision strategy using fp16 for near tokens and fp8/INT4 for far tokens to preserve fidelity while slashing memory footprint. Custom fused Metal kernels perform dequantization in registers, bypassing global memory bottlenecks and maximizing throughput. A novel spike-driven promotion mechanism from NVMe allows the system to recover precision for suddenly-important tokens dynamically. This enables 7B models on 8GB machines where memory pressure usually prevents effective local inference. Finally, the use of precision asymmetry—fp8 for keys and INT4 for values—correctly identifies that keys have a greater impact on softmax stability than values.
DISCOVERED
19d ago
2026-03-24
PUBLISHED
19d ago
2026-03-23
RELEVANCE
AUTHOR
hybls