arXiv:2512.10794cs.CVcs.AI2025-12被引 65

空间结构比全局语义更重要,能显著提升生成模型训练速度

What matters for Representation Alignment: Global Information or Spatial Structure?

  • 用卷积和空间归一化强化目标表征的空间信息传递
  • iREPA方法在27种视觉编码器上均加快收敛速度
  • 适合关注生成模型训练效率的开发者和研究者

表示对齐(REPA)通过将强预训练视觉编码器的表示蒸馏到扩散模型的中间特征中来指导生成训练。本文探讨一个根本问题:目标表示中什么因素对生成更关键——其全局语义信息(如ImageNet-1K准确率)还是空间结构(即块令牌间的成对余弦相似性)?主流观点认为更强的全局语义性能带来更好生成效果。我们对27种不同视觉编码器及多种模型规模进行了大规模实证分析,结果令人意外:空间结构而非全局性能才是生成表现的关键驱动因素。为此,我们提出两个简单修改:用卷积层替代标准MLP投影层,并为外部表示引入空间归一化层。所提方法iREPA(<4行代码实现)在多种视觉编码器、模型规模及训练变体(如REPA、REPA-E、Meanflow、JiT等)下一致提升收敛速度。本工作启发重新思考表示对齐的基本机制及其在生成模型训练中的应用。代码与项目页见https://end2end-diffusion.github.io/irepa

原文摘要 · Abstract (English)

Representation alignment (REPA) guides generative training by distilling representations from a strong, pretrained vision encoder to intermediate diffusion features. We investigate a fundamental question: what aspect of the target representation matters for generation, its \textit{global} \revision{semantic} information (e.g., measured by ImageNet-1K accuracy) or its spatial structure (i.e. pairwise cosine similarity between patch tokens)? Prevalent wisdom holds that stronger global semantic performance leads to better generation as a target representation. To study this, we first perform a large-scale empirical analysis across 27 different vision encoders and different model scales. The results are surprising; spatial structure, rather than global performance, drives the generation performance of a target representation. To further study this, we introduce two straightforward modifications, which specifically accentuate the transfer of \emph{spatial} information. We replace the standard MLP projection layer in REPA with a simple convolution layer and introduce a spatial normalization layer for the external representation. Surprisingly, our simple method (implemented in $<$4 lines of code), termed iREPA, consistently improves convergence speed of REPA, across a diverse set of vision encoders, model sizes, and training variants (such as REPA, REPA-E, Meanflow, JiT etc). %, etc. Our work motivates revisiting the fundamental working mechanism of representational alignment and how it can be leveraged for improved training of generative models. The code and project page are available at https://end2end-diffusion.github.io/irepa

表示对齐扩散模型训练加速

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