Keep pulling the thread on Tri Dao.
The SAW-INT4 method, which uses token-wise INT4 quantization with block-diagonal Hadamard rotation, consistently achieves the best accuracy-efficiency trade-off for 4-bit KV-cache quantization under practical serving constraints.
The SAW-INT4 method for KV-cache quantization recovers nearly all of the accuracy lost by naive INT4 quantization.
The fused rotation-quantization kernel for the SAW-INT4 method introduces zero measurable end-to-end overhead.
The SAW-INT4 KV-cache quantization method matches the throughput of plain INT4 quantization across all concurrency levels.
Many existing KV-cache compression methods are ineffective in deployment as they violate practical serving constraints like paged memory layouts, regular memory access, and fused attention execution.
More complex methods for KV-cache compression, such as vector quantization and Hessian-aware quantization, provide only marginal additional gains over SAW-INT4 once serving compatibility is considered.
A fused rotation-quantization kernel for the SAW-INT4 method was implemented to integrate directly into paged KV-cache layouts.