用对比记忆库加速生成模型训练,无需外部编码器且收敛更快。
Repulsor: Accelerating Generative Modeling with a Contrastive Memory Bank
- 引入动态更新的记忆库,自动存储大量负样本。
- 40万步内达图像网256上2.40的FID,性能领先。
- 插件式设计,推理无额外开销,适合高效训练场景。
去噪生成模型(如扩散模型、流匹配)在视觉合成中占主导地位,但存在训练成本高和表征学习效率低的问题。尽管通过辅助对齐注入判别性表征有效,但仍受限于依赖外部预训练编码器带来的开销与领域偏移。一种基于分散的策略可缓解此依赖,即强化批次内潜在表示间的强分离。为评估负样本数量对生成建模的影响,我们提出{ name},一个无需外部编码器的即插即用训练框架。该方法集成记忆库机制,在训练迭代中维护一个大规模、动态更新的负样本队列。这使负样本数量脱离小批量大小限制,以极低计算开销提供丰富高质量负样本,支持对比目标。同时采用低维投影头,进一步降低内存与带宽开销。{ name}具备三大优势:(1) 自包含,消除对预训练视觉基础模型及其前向传播开销的依赖;(2) 推理阶段不引入额外参数或计算成本;(3) 显著加快收敛速度,更高效实现优异生成质量。在ImageNet-256上,{ name}仅用40万步即达到2.40的最优FID,显著优于同类方法。
原文摘要 · Abstract (English)
The dominance of denoising generative models (e.g., diffusion, flow-matching) in visual synthesis is tempered by their substantial training costs and inefficiencies in representation learning. While injecting discriminative representations via auxiliary alignment has proven effective, this approach still faces key limitations: the reliance on external, pre-trained encoders introduces overhead and domain shift. A dispersed-based strategy that encourages strong separation among in-batch latent representations alleviates this specific dependency. To assess the effect of the number of negative samples in generative modeling, we propose {\mname}, a plug-and-play training framework that requires no external encoders. Our method integrates a memory bank mechanism that maintains a large, dynamically updated queue of negative samples across training iterations. This decouples the number of negatives from the mini-batch size, providing abundant and high-quality negatives for a contrastive objective without a multiplicative increase in computational cost. A low-dimensional projection head is used to further minimize memory and bandwidth overhead. {\mname} offers three principal advantages: (1) it is self-contained, eliminating dependency on pretrained vision foundation models and their associated forward-pass overhead; (2) it introduces no additional parameters or computational cost during inference; and (3) it enables substantially faster convergence, achieving superior generative quality more efficiently. On ImageNet-256, {\mname} achieves a state-of-the-art FID of \textbf{2.40} within 400k steps, significantly outperforming comparable methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。