Latent Compaction with Test-Time Training
Novel compression technique for long-context models that preserves performance while reducing memory and computation costs.
AI Summary
Researchers introduce Latent Compaction, a test-time method that compresses KV caches 32× while maintaining 96.7% accuracy on retrieval tasks.
Excerpt
Long contexts enlarge key--value (KV) caches, increasing memory use and attention costs during generation. Compressing these caches must preserve information for unseen questions. We introduce **Latent Compaction**, which learns a compact document-specific cache through test-time training. We initialize cache entries using reconstruction importance, teacher attention, and key-space diversity. Training first fits attention outputs and log-normalizers to those of the full cache, with reconstructio
