about
Generative Modeling with Explicit Memory (arxiv.org)
2 points by PaulHoule on Jan 2, 2025 | hide | past | pdf | discuss on HN

In plain words: Instead of growing the network to memorize the training data's contents, this stores that knowledge in a separate, fixed memory the model can look up. On ImageNet at 256×256, it trained 50 times faster than a standard diffusion network to reach similar image quality.

Abstract · GMem: A Modular Approach for Ultra-Efficient Generative Models

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$.

Yi Tang, Peng Sun, Zhenglin Cheng, Tao Lin
arXiv:2412.08781 · cs.CV, cs.LG · submitted Dec 11, 2024 · updated Feb 11, 2025
abstract · pdf · html · 9 pages, 5 figures, 3 tables

add comment on HN