LiteVAR: Compressing Visual Autoregressive Modelling with Efficient Attention and Quantization
About
Visual Autoregressive (VAR) has emerged as a promising approach in image generation, offering competitive potential and performance comparable to diffusion-based models. However, current AR-based visual generation models require substantial computational resources, limiting their applicability on resource-constrained devices. To address this issue, we conducted analysis and identified significant redundancy in three dimensions of the VAR model: (1) the attention map, (2) the attention outputs when using classifier free guidance, and (3) the data precision. Correspondingly, we proposed efficient attention mechanism and low-bit quantization method to enhance the efficiency of VAR models while maintaining performance. With negligible performance lost (less than 0.056 FID increase), we could achieve 85.2% reduction in attention computation, 50% reduction in overall memory and 1.5x latency reduction. To ensure deployment feasibility, we developed efficient training-free compression techniques and analyze the deployment feasibility and efficiency gain of each technique.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Text-to-Image Generation | GenEval | GenEval Score72 | 459 | |
| Text-to-Image Generation | HPS v2.1 | Overall Score32.03 | 153 | |
| Semantic segmentation | COCO | mIoU66.9 | 119 | |
| Text-to-Image Generation | ImageReward | ImageReward Score0.912 | 119 | |
| Class-conditional Image Generation | ImageNet 2009 (val) | Inception Score (IS)261.8 | 52 | |
| Panoptic Segmentation | COCO | PQ57.4 | 46 | |
| Instance Segmentation | COCO | AP0.482 | 21 | |
| Class-conditional Editing | VAR-d30 | CLIP Score24.58 | 9 | |
| Image Outpainting | VAR-d30 | LPIPS0.2143 | 9 | |
| Image Inpainting | VAR-d30 | LPIPS0.2861 | 9 |