Few-shot Generation via Recalling Brain-Inspired Episodic-Semantic Memory
Zhibin Duan, Zhiyi Lv, Chaojie Wang, Bo Chen, Bo An, Mingyuan Zhou
Abstract
Aimed at adapting a generative model to a novel generation task with only a few given data samples, the capability of few-shot generation is crucial for many real-world applications with limited data, e.g. , artistic domains. Instead of training from scratch, recent works tend to leverage the prior knowledge stored in previous datasets, which is quite similar to the memory mechanism of human intelligence, but few of these works directly imitate the memory-recall mechanism that humans make good use of in accomplishing creative tasks, e.g. , painting and writing. Inspired by the memory mechanism of human brain, in this work, we carefully design a variational structured memory module (VSM), which can simultaneously store both episodic and semantic memories to assist existing generative models efficiently recall these memories during sample generation. Meanwhile, we introduce a bionic memory updating strategy for the conversion between episodic and semantic memories, which can also model the uncertainty during conversion. Then, we combine the developed VSM with various generative models under the Bayesian framework, and evaluate these memory-augmented generative models with few-shot generation tasks, demonstrating the effectiveness of our methods.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 0b50f1ae-4a29-4d7b-9c19-9102dc5e8b0eCited by top-tier papers1
Ask how each one uses itBuilds on11
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- Generating images with sparse representationsCharlie Nash, Jacob Menick, Sander Dieleman, Peter W. BattagliaICML 2021 · 291 citations
- Retrieval-Augmented Diffusion ModelsAndreas Blattmann, Robin Rombach, Kaan Oktay, Jonas Müller et al.NeurIPS 2022 · 239 citations
- D2C: Diffusion-Decoding Models for Few-Shot Conditional GenerationAbhishek Sinha, Jiaming Song, Chenlin Meng, Stefano ErmonNeurIPS 2021 · 149 citations
Related papers
- Learning to Learn Variational Semantic MemoryXiantong Zhen, Ying-Jun Du, Huan Xiong, Qiang Qiu et al.NeurIPS 2020 · 40 citations
- Domain Re-Modulation for Few-Shot Generative Domain AdaptationYi Wu, Ziqiang Li, Chaoyue Wang, Heliang Zheng et al.NeurIPS 2023 · 31 citations
- Remember the Difference: Cross-Domain Few-Shot Semantic Segmentation via Meta-Memory TransferWenjian Wang, Lijuan Duan, Yuxi Wang, Qing En et al.CVPR 2022 · 32 citations
- Hierarchical Variational Memory for Few-shot Learning Across DomainsYing-Jun Du, Xiantong Zhen, Ling Shao, Cees G. M. SnoekICLR 2022 · 24 citations
- Task-Free Continual Generation and Representation Learning via Dynamic Expansionable Memory ClusterFei Ye, Adrian G. BorsAAAI 2024 · 8 citations
