Saved in:
Bibliographic Details
Main Authors: Wang, Yu, Gao, Yifan, Chen, Xiusi, Jiang, Haoming, Li, Shiyang, Yang, Jingfeng, Yin, Qingyu, Li, Zheng, Li, Xian, Yin, Bing, Shang, Jingbo, McAuley, Julian
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2402.04624
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911887723069440
author Wang, Yu
Gao, Yifan
Chen, Xiusi
Jiang, Haoming
Li, Shiyang
Yang, Jingfeng
Yin, Qingyu
Li, Zheng
Li, Xian
Yin, Bing
Shang, Jingbo
McAuley, Julian
author_facet Wang, Yu
Gao, Yifan
Chen, Xiusi
Jiang, Haoming
Li, Shiyang
Yang, Jingfeng
Yin, Qingyu
Li, Zheng
Li, Xian
Yin, Bing
Shang, Jingbo
McAuley, Julian
contents Existing Large Language Models (LLMs) usually remain static after deployment, which might make it hard to inject new knowledge into the model. We aim to build models containing a considerable portion of self-updatable parameters, enabling the model to integrate new knowledge effectively and efficiently. To this end, we introduce MEMORYLLM, a model that comprises a transformer and a fixed-size memory pool within the latent space of the transformer. MEMORYLLM can self-update with text knowledge and memorize the knowledge injected earlier. Our evaluations demonstrate the ability of MEMORYLLM to effectively incorporate new knowledge, as evidenced by its performance on model editing benchmarks. Meanwhile, the model exhibits long-term information retention capacity, which is validated through our custom-designed evaluations and long-context benchmarks. MEMORYLLM also shows operational integrity without any sign of performance degradation even after nearly a million memory updates. Our code and model are open-sourced at https://github.com/wangyu-ustc/MemoryLLM.
format Preprint
id arxiv_https___arxiv_org_abs_2402_04624
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MEMORYLLM: Towards Self-Updatable Large Language Models
Wang, Yu
Gao, Yifan
Chen, Xiusi
Jiang, Haoming
Li, Shiyang
Yang, Jingfeng
Yin, Qingyu
Li, Zheng
Li, Xian
Yin, Bing
Shang, Jingbo
McAuley, Julian
Computation and Language
Existing Large Language Models (LLMs) usually remain static after deployment, which might make it hard to inject new knowledge into the model. We aim to build models containing a considerable portion of self-updatable parameters, enabling the model to integrate new knowledge effectively and efficiently. To this end, we introduce MEMORYLLM, a model that comprises a transformer and a fixed-size memory pool within the latent space of the transformer. MEMORYLLM can self-update with text knowledge and memorize the knowledge injected earlier. Our evaluations demonstrate the ability of MEMORYLLM to effectively incorporate new knowledge, as evidenced by its performance on model editing benchmarks. Meanwhile, the model exhibits long-term information retention capacity, which is validated through our custom-designed evaluations and long-context benchmarks. MEMORYLLM also shows operational integrity without any sign of performance degradation even after nearly a million memory updates. Our code and model are open-sourced at https://github.com/wangyu-ustc/MemoryLLM.
title MEMORYLLM: Towards Self-Updatable Large Language Models
topic Computation and Language
url https://arxiv.org/abs/2402.04624