arXiv:2502.20313cs.CV2025-02NeurIPS被引 23

抛弃残差预测,用真值直接生成图像,让扩散模型更灵活高效。

FlexVAR: Flexible Visual Autoregressive Modeling without Residual Prediction

  • 不依赖残差,每步直接预测真实图像,训练更简单。
  • 256像素训练后可生成更高分辨率、不同比例的图像,支持多种图像修复任务。
  • 支持任意推理步数,速度快慢自由调节,零样本迁移表现超主流模型。

本文挑战视觉自回归建模中的残差预测范式,提出FlexVAR——一种无需残差预测的灵活视觉自回归图像生成新框架。该方法通过直接使用真实图像作为每一步的预测目标,使每个生成步骤都能独立产出合理图像。仅在≤256px低分辨率图像上训练,FlexVAR即可:(1) 生成多种分辨率和长宽比的图像,包括超过训练分辨率的输出;(2) 支持图像修复、内外补全、图像扩展等各类图像到图像任务;(3) 适应不同自回归步数,实现更快推理或更高画质。其1.0B模型在ImageNet 256×256基准上优于对应VAR模型。零样本迁移至13步生成时,FID达2.08,优于AiM/VAR(-0.25/-0.28)和LDM/DiT(-1.52/-0.19)。将1.0B模型零样本迁移至512×512基准,性能媲美完全监督训练的VAR 2.3B模型。

原文摘要 · Abstract (English)

This work challenges the residual prediction paradigm in visual autoregressive modeling and presents FlexVAR, a new Flexible Visual AutoRegressive image generation paradigm. FlexVAR facilitates autoregressive learning with ground-truth prediction, enabling each step to independently produce plausible images. This simple, intuitive approach swiftly learns visual distributions and makes the generation process more flexible and adaptable. Trained solely on low-resolution images ($\leq$ 256px), FlexVAR can: (1) Generate images of various resolutions and aspect ratios, even exceeding the resolution of the training images. (2) Support various image-to-image tasks, including image refinement, in/out-painting, and image expansion. (3) Adapt to various autoregressive steps, allowing for faster inference with fewer steps or enhancing image quality with more steps. Our 1.0B model outperforms its VAR counterpart on the ImageNet 256$\times$256 benchmark. Moreover, when zero-shot transfer the image generation process with 13 steps, the performance further improves to 2.08 FID, outperforming state-of-the-art autoregressive models AiM/VAR by 0.25/0.28 FID and popular diffusion models LDM/DiT by 1.52/0.19 FID, respectively. When transferring our 1.0B model to the ImageNet 512$\times$512 benchmark in a zero-shot manner, FlexVAR achieves competitive results compared to the VAR 2.3B model, which is a fully supervised model trained at 512$\times$512 resolution.

图像生成自回归灵活生成零样本迁移

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。