用视觉自回归模型实现快速主体驱动生成
Fine-Tuning Visual Autoregressive Models for Subject-Driven Generation
- 分层选择性微调降低计算开销
- 加权跨尺度微调提升主体一致性
- 适合需要快速生成的实用场景
近期文本到图像生成模型进展推动了主体驱动生成的应用,即通过少量样本微调预训练模型以捕捉主体语义。尽管基于扩散的模型生成质量高,但其大量去噪步骤带来显著计算开销,限制实际应用。视觉自回归(VAR)模型通过预测下一尺度的标记而非空间邻近项,具备更快推理速度,更适合部署。本文提出首个基于VAR的主体驱动生成方法。然而,直接微调VAR会导致计算复杂度高、语言漂移和多样性下降。为此,我们引入选择性层微调以降低复杂度,以及先验蒸馏缓解语言漂移。此外,我们发现早期阶段对主体生成影响更大,后期仅合成细节。基于此,提出分尺度加权微调,优先优化粗分辨率以聚焦主体信息而非局部细节。大量实验表明,本方法在多个指标上显著优于扩散基线,并验证了其实用性。
原文摘要 · Abstract (English)
Recent advances in text-to-image generative models have enabled numerous practical applications, including subject-driven generation, which fine-tunes pretrained models to capture subject semantics from only a few examples. While diffusion-based models produce high-quality images, their extensive denoising steps result in significant computational overhead, limiting real-world applicability. Visual autoregressive (VAR) models, which predict next-scale tokens rather than spatially adjacent ones, offer significantly faster inference suitable for practical deployment. In this paper, we propose the first VAR-based approach for subject-driven generation. However, naive fine-tuning VAR leads to computational overhead, language drift, and reduced diversity. To address these challenges, we introduce selective layer tuning to reduce complexity and prior distillation to mitigate language drift. Additionally, we found that the early stages have a greater influence on the generation of subject than the latter stages, which merely synthesize minor details. Based on this finding, we propose scale-wise weighted tuning, which prioritizes coarser resolutions for promoting the model to focus on the subject-relevant information instead of local details. Extensive experiments validate that our method significantly outperforms diffusion-based baselines across various metrics and demonstrates its practical usage.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。