Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Rui Xie, Tianchen Zhao, Zhihang Yuan, Rui Wan, Wenxi Gao, Zhenhua Zhu, Xuefei Ning, Yu Wang• 2024

Related benchmarks

TaskDatasetResultRank
Text-to-Image GenerationGenEval
GenEval Score72
459
Text-to-Image GenerationHPS v2.1
Overall Score32.03
153
Semantic segmentationCOCO
mIoU66.9
119
Text-to-Image GenerationImageReward
ImageReward Score0.912
119
Class-conditional Image GenerationImageNet 2009 (val)
Inception Score (IS)261.8
52
Panoptic SegmentationCOCO
PQ57.4
46
Instance SegmentationCOCO
AP0.482
21
Class-conditional EditingVAR-d30
CLIP Score24.58
9
Image OutpaintingVAR-d30
LPIPS0.2143
9
Image InpaintingVAR-d30
LPIPS0.2861
9
Showing 10 of 10 rows

Other info

Follow for update