mlsanityData to 7 Oct 2026

HACK++: Towards More Effective Head-Aware Key-Value Compression for Efficient Visual Autoregressive Modeling

arXiv 2606.08302 ↗First posted 6 Jun 2026cs.CVcited by 1

Not on the map. Why?

Abstract (arXiv)

Visual Autoregressive (VAR) models adopt a next-scale prediction paradigm, offering high-quality generation with substantially fewer decoding steps. However, existing VAR models suffer from significant attention complexity and severe memory overhead due to the accumulation of key-value (KV) caches across scales. In this paper, we tackle this challenge by introducing KV cache compression into the next-scale paradigm. We begin with an in-depth analysis of VAR attention and observe that attention heads can be stably divided into two functionally distinct categories: Contextual Heads focus on maintaining semantic consistency, while Structural Heads preserve spatial coherence. Their functional divergence makes existing one-size-fits-all compression methods perform poorly on VAR models. We further find that the two head types differ markedly in their reliance on historical scales, and that this reliance shifts across layers and generation steps, arguing for an adaptive cache budget allocation. To address these challenges, we propose HACK++, a training-free Head-Aware key-value Compression frameworK for VAR models. From a one-time offline calibration, HACK++ classifies head types and derives head-specific priors. At inference, it decouples attention from cache compression under independent budgets, bounding the current-scale attention cost while compressing the accumulated cache far more aggressively, via pattern-specific strategies and a reliance-aware budget allocation. Extensive experiments on multiple VAR models across text-to-image, class-conditional, and unified understanding-and-generation tasks validate the effectiveness and generalizability of HACK++. For example, on Infinity-2B/8B, HACK++ maintains near-lossless generation with only a 30% attention budget and a 10% cache budget, and remains robust even under a 1% cache budget.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 61 of its 68 references that are in the map. First posted on arXiv 6 Jun 2026.

Builds on

61 of 68 references.

References · 61

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2023GPT-4 Technical Reportcited by 18,132
  3. 2024The Llama 3 Herd of Modelscited by 14,192
  4. 2025Qwen3 Technical Reportcited by 8,195
  5. 2022Scalable Diffusion Models with Transformerscited by 5,931
  6. 2017Neural Discrete Representation Learningcited by 4,650
  7. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
  8. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
  9. 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
  10. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
  11. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  12. 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
38 more
  1. 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
  2. 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
  3. 2024Chameleon: Mixed-Modal Early-Fusion Foundation Modelscited by 904
  4. 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 848
  5. 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 847
  6. 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Modelscited by 805
  7. 2025Janus-Pro: Unified Multimodal Understanding and Generation with Data and Model Scalingcited by 787
  8. 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 760
  9. 2024Autoregressive Image Generation without Vector Quantizationcited by 644
  10. 2024KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cachecited by 604
  11. 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignmentcited by 455
  12. 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 447
  13. 2023Scissorhands: Exploiting the Persistence of Importance Hypothesis for LLM KV Cache Compression at Test Timecited by 428
  14. 2024PixArt-Sigma: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generationcited by 326
  15. 2024Infinity: Scaling Bitwise AutoRegressive Modeling for High-Resolution Image Synthesiscited by 297
  16. 2024DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Headscited by 283
  17. 2024Playground v2.5: Three Insights towards Enhancing Aesthetic Quality in Text-to-Image Generationcited by 249
  18. 2024GEAR: An Efficient KV Cache Compression Recipe for Near-Lossless Generative Inference of LLMcited by 155
  19. 2024HART: Efficient Visual Generation with Hybrid Autoregressive Transformercited by 128
  20. 2024MiniCache: KV Cache Compression in Depth Dimension for Large Language Modelscited by 122
  21. 2024Transformers are Multi-State RNNscited by 120
  22. 2024LOOK-M: Look-Once Optimization in KV Cache for Efficient Multimodal Long-Context Inferencecited by 93
  23. 2024ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identificationcited by 86
  24. 2024WKVQuant: Quantizing Weight and Key/Value Cache for Large Language Models Gains Morecited by 69
  25. 2025CAKE: Cascading and Adaptive KV Cache Eviction with Layer Preferencescited by 59
  26. 2024ControlVAR: Exploring Controllable Visual Autoregressive Modelingcited by 58
  27. 2025OneCAT: Decoder-Only Auto-Regressive Model for Unified Understanding and Generationcited by 50
  28. 2024MARS: Mixture of Auto-Regressive Models for Fine-grained Text-to-image Synthesiscited by 44
  29. 2025VARGPT: Unified Understanding and Generation in a Visual Autoregressive Multimodal Large Language Modelcited by 44
  30. 2024VL-Cache: Sparsity and Modality-Aware KV Cache Compression for Vision-Language Model Inference Accelerationcited by 41
  31. 2024SAR3D: Autoregressive 3D Object Generation and Understanding via Multi-scale 3D VQVAEcited by 38
  32. 2025Visual Autoregressive Modeling for Image Super-Resolutioncited by 36
  33. 2025FastVAR: Linear Visual Autoregressive Modeling via Cached Token Pruningcited by 34
  34. 2024Switti: Designing Scale-Wise Transformers for Text-to-Image Synthesiscited by 33
  35. 2024On the Efficacy of Eviction Policy for Key-Value Constrained Generative Language Model Inferencecited by 32
  36. 2024Collaborative Decoding Makes Visual Auto-Regressive Modeling Efficientcited by 32
  37. 2025MEDA: Dynamic KV Cache Allocation for Efficient Multimodal Long-Context Inferencecited by 31
  38. 2024M-VAR: Decoupled Scale-wise Autoregressive Modeling for High-Quality Image Generationcited by 28

50 most cited of 61 All 61 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Head-Aware Key-Value Compression for Efficient Autoregressive Image Generation
  2. 2025Visual Autoregressive Transformers Must Use Omega(n2 d) Memory
  3. 2026HeatKV: Head-tuned KV-cache Compression for Visual Autoregressive Modeling
  4. 2025SSD: Spatial-Semantic Head Decoupling for Efficient Autoregressive Image Generation
  5. 2025AMS-KV: Adaptive KV Caching in Multi-Scale Visual Autoregressive Transformers
  6. 2026Logit Refiner: Improving Visual Autoregressive Models via Intra-Scale Dependency Modeling
  7. 2026Adaptive Visual Autoregressive Acceleration via Dual-Linkage Entropy Analysis
  8. 2025Scale-Wise VAR is Secretly Discrete Diffusion