Generative Pseudo-Inverse Memory
Kha Pham, Hung Le, Man Ngo, Truyen Tran, Bao Ho, Svetha Venkatesh
Abstract
We propose Generative Pseudo-Inverse Memory (GPM), a class of deep generative memory models that are fast to write in and read out. Memory operations are recast as seeking robust solutions of linear systems, which naturally lead to the use of matrix pseudo-inverses. The pseudo-inverses are iteratively approximated, with practical computation complexity of almost O(1). We prove theoretically and verify empirically that our model can retrieve exactly what have been written to the memory under mild conditions. A key capability of GPM is iterative reading, during which the attractor dynamics towards fixed points are enabled, allowing the model to iteratively improve sample quality in denoising and generating. More impressively, GPM can store a large amount of data while maintaining key abilities of accurate retrieving of stored patterns, denoising of corrupted data and generating novel samples. Empirically we demonstrate the efficiency and versatility of GPM on a comprehensive suite of experiments involving binarized MNIST, binarized Omniglot, FashionMNIST, CIFAR10 & CIFAR100 and CelebA.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers3
- Larimar: Large Language Models with Episodic Memory ControlPayel Das, Subhajit Chaudhury, Elliot Nelson, Igor Melnyk et al.ICML 2024 · 37 citations
- BayesPCN: A Continually Learnable Predictive Coding Associative MemoryJinsoo Yoo, Frank WoodNeurIPS 2022 · 16 citations
- Balancing Context Length and Mixing Times for Reinforcement Learning at ScaleMatthew Riemer, Khimya Khetarpal, Janarthanan Rajendran, Sarath ChandarNeurIPS 2024 · 7 citations
Builds on3
- Hopfield Networks is All You NeedHubert Ramsauer, Bernhard Schäfl, Johannes Lehner, Philipp Seidl et al.ICLR 2021 · 620 citations
- Attention Approximates Sparse Distributed MemoryTrenton Bricken, Cengiz PehlevanNeurIPS 2021 · 44 citations
- Kanerva++: Extending the Kanerva Machine With Differentiable, Locally Block Allocated Latent MemoryJason Ramapuram, Yan Wu, Alexandros KalousisICLR 2021 · 4 citations
Related papers
- Optimization for Amortized Inverse ProblemsTianci Liu, Tong Yang, Quan Zhang, Qi LeiICML 2023 · 7 citations
- PLUGIn: A simple algorithm for inverting generative models with recovery guaranteesBabhru Joshi, Xiaowei Li, Yaniv Plan, Özgür YilmazNeurIPS 2021 · 7 citations
- Towards Authentic Face Restoration with Iterative Diffusion Models and BeyondYang Zhao, Tingbo Hou, Yu-Chuan Su, Xuhui Jia et al.ICCV 2023 · 30 citations
- Denoising Diffusion Implicit ModelsJiaming Song, Chenlin Meng, Stefano ErmonICLR 2021 · 11,743 citations
- Unified Gradient-Based Machine Unlearning with Remain Geometry EnhancementZhehao Huang, Xinwen Cheng, JingHao Zheng, Haoran Wang et al.NeurIPS 2024 · 57 citations
