用独立记忆模块提升生成模型效率,训练快50倍,效果更好。
GMem: A Modular Approach for Ultra-Efficient Generative Models
- 将记忆从网络中分离,用静态内存存储数据语义信息。
- 在ImageNet上训练速度提升50倍,28轮内达FID=7.66,仅需4小时。
- 无分类器引导下达FID=1.53,优于同类方法,适合高效生成任务。
近期研究表明,深度生成扩散模型的去噪过程会隐式学习并记忆数据分布中的语义信息。这些发现表明,捕捉更复杂的数据分布需要更大的神经网络,导致计算需求大幅增加,成为训练与推理的主要瓶颈。为此,我们提出GMem:一种超高效的生成模型模块化方法。GMem将记忆容量与模型解耦,将其作为独立的、不可变的记忆集合,以保留数据中的关键语义信息。结果显著:GMem提升了训练与采样效率,并增强了生成多样性。该设计一方面降低了对网络记忆复杂数据分布的依赖,从而提升训练与采样效率。在256×256分辨率的ImageNet上,GMem相比SiT实现50倍训练加速,在不到28个周期(约4小时)内达到FID=7.66,而SiT需1400个周期。无需分类器引导时,GMem在160个周期内达最优性能FID=1.53,仅需约20小时训练,优于需800周期和约95小时才能达到FID=2.17的LightningDiT。
原文摘要 · Abstract (English)
Recent studies indicate that the denoising process in deep generative diffusion models implicitly learns and memorizes semantic information from the data distribution. These findings suggest that capturing more complex data distributions requires larger neural networks, leading to a substantial increase in computational demands, which in turn become the primary bottleneck in both training and inference of diffusion models. To this end, we introduce GMem: A Modular Approach for Ultra-Efficient Generative Models. Our approach GMem decouples the memory capacity from model and implements it as a separate, immutable memory set that preserves the essential semantic information in the data. The results are significant: GMem enhances both training, sampling efficiency, and diversity generation. This design on one hand reduces the reliance on network for memorize complex data distribution and thus enhancing both training and sampling efficiency. On ImageNet at $256 \times 256$ resolution, GMem achieves a $50\times$ training speedup compared to SiT, reaching FID $=7.66$ in fewer than $28$ epochs ($\sim 4$ hours training time), while SiT requires $1400$ epochs. Without classifier-free guidance, GMem achieves state-of-the-art (SoTA) performance FID $=1.53$ in $160$ epochs with only $\sim 20$ hours of training, outperforming LightningDiT which requires $800$ epochs and $\sim 95$ hours to attain FID $=2.17$.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。