arXiv:2412.08175cs.CVcs.LG2024-12被引 11

提出新方法缓解流模型自生成数据训练中的性能退化问题。

Analyzing and Mitigating Model Collapse in Rectified Flow Models

  • 理论证明自生成数据训练会导致模型退化,首次分析去噪自编码器中的崩溃现象。
  • 实验证明修正流模型在每轮迭代中均会退化,且真实数据可有效防止崩溃。
  • 提出融合真实数据的增强型修正流方法,显著提升生成质量与采样效率。

随着深度生成模型的普及,合成数据的使用日益普遍,尤其在修正流模型中,其Refloew方法通过迭代使用自生成数据来简化数据流并提升采样效率。然而,近期研究指出反复使用自生成样本会导致模型崩溃(MC),性能随训练迭代下降。现有工作多聚焦于经验观察或回归/最大似然目标,缺乏对Refloew方法的严谨理论分析。本文首次系统分析去噪自编码器(DAE)在递归训练下的性能退化,并证明该现象可延伸至扩散模型与修正流模型。实验表明,修正流模型在每轮重流步骤中均存在性能下降风险。我们进一步证明,引入真实数据可有效阻止模型崩溃,支持了当前结合真实数据缓解崩溃的趋势。基于此,提出一种新型真实数据增强修正流(RA Reflow)及其系列改进版本,通过反向流机制无缝整合真实数据。在标准图像基准上的实证评估显示,RA Reflow能有效抑制模型崩溃,即使减少采样步数也能保持高质量生成。

原文摘要 · Abstract (English)

Training with synthetic data is becoming increasingly inevitable as synthetic content proliferates across the web, driven by the remarkable performance of recent deep generative models. This reliance on synthetic data can also be intentional, as seen in Rectified Flow models, whose Reflow method iteratively uses self-generated data to straighten the flow and improve sampling efficiency. However, recent studies have shown that repeatedly training on self-generated samples can lead to model collapse (MC), where performance degrades over time. Despite this, most recent work on MC either focuses on empirical observations or analyzes regression problems and maximum likelihood objectives, leaving a rigorous theoretical analysis of reflow methods unexplored. In this paper, we aim to fill this gap by providing both theoretical analysis and practical solutions for addressing MC in diffusion/flow models. We begin by studying Denoising Autoencoders and prove performance degradation when DAEs are iteratively trained on their own outputs. To the best of our knowledge, we are the first to rigorously analyze model collapse in DAEs and, by extension, in diffusion models and Rectified Flow. Our analysis and experiments demonstrate that rectified flow also suffers from MC, leading to potential performance degradation in each reflow step. Additionally, we prove that incorporating real data can prevent MC during recursive DAE training, supporting the recent trend of using real data as an effective approach for mitigating MC. Building on these insights, we propose a novel Real-data Augmented Reflow and a series of improved variants, which seamlessly integrate real data into Reflow training by leveraging reverse flow. Empirical evaluations on standard image benchmarks confirm that RA Reflow effectively mitigates model collapse, preserving high-quality sample generation even with fewer sampling steps.

流模型模型崩溃生成模型数据增强

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