Shift-and-Sum Quantization for Visual Autoregressive Models
About
Post-training quantization (PTQ) enables efficient deployment of deep networks using a small set of data. Its application to visual autoregressive models (VAR), however, remains relatively unexplored. We identify two key challenges for applying PTQ to VAR: (i) large reconstruction errors in attention-value products, especially at coarse scales where high attention scores occur more frequently; and (ii) a discrepancy between the sampling frequencies of codebook entries and their predicted probabilities due to limited calibration data. To address these challenges, we propose a PTQ framework tailored for VAR. First, we introduce a shift-and-sum quantization method that reduces reconstruction errors by aggregating quantized results from symmetrically shifted duplicates of value tokens. Second, we present a resampling strategy for calibration data that aligns sampling frequencies of codebook entries with their predicted probabilities. Experiments on class-conditional image generation, inpainting, outpainting, and class-conditional editing show consistent improvements across VAR architectures, establishing a new state of the art in PTQ for VAR.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Text-to-Image Generation | GenEval | GenEval Score72.5 | 459 | |
| Text-to-Image Generation | HPS v2.1 | Overall Score32.08 | 153 | |
| Semantic segmentation | COCO | mIoU67 | 119 | |
| Text-to-Image Generation | ImageReward | ImageReward Score0.929 | 119 | |
| Class-conditional Image Generation | ImageNet 2009 (val) | Inception Score (IS)269.9 | 52 | |
| Panoptic Segmentation | COCO | PQ57.8 | 46 | |
| Instance Segmentation | COCO | AP0.486 | 21 | |
| Class-conditional Editing | VAR-d30 | CLIP Score24.65 | 9 | |
| Image Inpainting | VAR-d30 | LPIPS0.2835 | 9 | |
| Image Outpainting | VAR-d30 | LPIPS0.2134 | 9 |