arXiv:2410.03755cs.LGcs.CV2024-10ICLR被引 11

将自监督模型JEPA用于生成任务,实现高效图像与视频生成。

Denoising with a Joint-Embedding Predictive Architecture

  • 将JEPA重解释为广义的自回归预测,支持数据生成。
  • 引入扩散损失和流匹配损失,实现连续空间建模。
  • 在ImageNet上优于所有已有生成模型,适合图像、视频等连续数据。

联合嵌入预测架构(JEPAs)在自监督表征学习中表现优异,但在生成建模中应用仍较少。本文提出一种新型生成模型D-JEPA,首次将JEPA引入生成建模。通过将JEPA视为掩码图像建模,将其重新解释为广义的下一个词元预测策略,从而实现自回归式数据生成。同时,引入扩散损失以建模每个词元的概率分布,使生成过程在连续空间中进行。此外,还采用流匹配损失作为替代方案,提升模型灵活性。实验表明,随着计算量增加,D-JEPA在更少训练轮次下持续获得更低的FID分数,展现出良好可扩展性。其基础、大和超大模型在ImageNet条件生成基准上均超越所有先前生成模型。除图像生成外,D-JEPA也适用于视频、音频等连续数据建模。

原文摘要 · Abstract (English)

Joint-embedding predictive architectures (JEPAs) have shown substantial promise in self-supervised representation learning, yet their application in generative modeling remains underexplored. Conversely, diffusion models have demonstrated significant efficacy in modeling arbitrary probability distributions. In this paper, we introduce Denoising with a Joint-Embedding Predictive Architecture (D-JEPA), pioneering the integration of JEPA within generative modeling. By recognizing JEPA as a form of masked image modeling, we reinterpret it as a generalized next-token prediction strategy, facilitating data generation in an auto-regressive manner. Furthermore, we incorporate diffusion loss to model the per-token probability distribution, enabling data generation in a continuous space. We also adapt flow matching loss as an alternative to diffusion loss, thereby enhancing the flexibility of D-JEPA. Empirically, with increased GFLOPs, D-JEPA consistently achieves lower FID scores with fewer training epochs, indicating its good scalability. Our base, large, and huge models outperform all previous generative models across all scales on ImageNet conditional generation benchmarks. Beyond image generation, D-JEPA is well-suited for other continuous data modeling, including video and audio.

生成模型自监督扩散模型连续建模

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