提出SAR方法解决分尺度自回归生成的训练偏差问题。
Rethinking Training Dynamics in Scale-wise Autoregressive Generation
- 引入错位尺度滚动机制,让模型提前适应自身预测
- 在ImageNet上10轮训练即降低5.2%FID,仅需5小时
- 适合希望提升已有自回归生成模型质量的研究者
自回归生成模型在媒体合成中表现日益强大,其中分尺度生成成为主流范式,即从粗到细逐步生成图像。然而,分尺度自回归模型存在暴露偏差问题,影响生成质量。我们识别出两大根源:(1) 训练与测试不一致,推理时模型依赖自身不完美的预测;(2) 各尺度学习难度失衡,部分尺度优化复杂度过高。通过深入分析训练动态,我们提出自回归精炼(SAR)方法,包含轻量级错位尺度滚动(SSR)机制,使模型提前接触自身中间预测以对齐训练-推理模式,并引入对比学生强迫损失(CSFL),为自生成上下文提供充分监督,确保训练稳定。实验表明,将SAR应用于预训练模型可在极低计算开销下持续提升生成质量。例如,在ImageNet 256上训练的FlexVAR-d16,仅用10轮(32xA100 GPU运行5小时)即实现5.2%的FID降低。因其高效、可扩展且有效,SAR有望成为视觉自回归生成的可靠后处理方案。
原文摘要 · Abstract (English)
Recent advances in autoregressive (AR) generative models have produced increasingly powerful systems for media synthesis. Among them, next-scale prediction has emerged as a popular paradigm, where models generate images in a coarse-to-fine manner. However, scale-wise AR models suffer from exposure bias, which undermines generation quality. We identify two primary causes of this issue: (1) train-test mismatch, where the model must rely on its own imperfect predictions during inference, and (2) imbalance in scale-wise learning difficulty, where certain scales exhibit disproportionately higher optimization complexity. Through a comprehensive analysis of training dynamics, we propose Self-Autoregressive Refinement (SAR) to address these limitations. SAR introduces a Stagger-Scale Rollout (SSR) mechanism that performs lightweight autoregressive rollouts to expose the model to its own intermediate predictions, thereby aligning train-test patterns, and a complementary Contrastive Student-Forcing Loss (CSFL) that provides adequate supervision for self-generated contexts to ensure stable training. Experimental results show that applying SAR to pretrained AR models consistently improves generation quality with minimal computational overhead. For instance, SAR yields a 5.2% FID reduction on FlexVAR-d16 trained on ImageNet 256 within 10 epochs (5 hours on 32xA100 GPUs). Given its efficiency, scalability, and effectiveness, we expect SAR to serve as a reliable post-training method for visual autoregressive generation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。