Mass-Editing Memory in a Transformer
Authors: Kevin Meng, Arnab Sen Sharma, Alex Andonian, Yonatan Belinkov, David Bau
AI-generated Key Points
- There is a growing interest in updating large language models with new memories.
- Most existing methods are limited to updating single associations.
- MEMIT is a method for directly updating a language model with many memories.
- MEMIT can scale up to thousands of associations for GPT-J (6B) and GPT-NeoX (20B).
- The method involves encoding each memory as a sequence of tokens and then concatenating them into one long sequence that is fed into the transformer model along with the original input text.
- The researchers conducted experiments on several benchmark datasets and found that their method significantly outperformed previous approaches in terms of accuracy and efficiency.
- They released their code and data at https://memit.baulab.info for others to use and build upon.
- MEMIT represents an important step forward in the development of language models capable of incorporating multiple memories simultaneously.
Abstract: Recent work has shown exciting promise in updating large language models with new memories, so as to replace obsolete information or add specialized knowledge. However, this line of work is predominantly limited to updating single associations. We develop MEMIT, a method for directly updating a language model with many memories, demonstrating experimentally that it can scale up to thousands of associations for GPT-J (6B) and GPT-NeoX (20B), exceeding prior work by orders of magnitude. Our code and data are at https://memit.baulab.info.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.