In-context Autoencoder for Context Compression in a Large Language Model
Authors: Tao Ge, Jing Hu, Xun Wang, Si-Qing Chen, Furu Wei
AI-generated Key Points
⚠The license of the paper does not allow us to build upon its content and the key points are generated using the paper metadata rather than the full article.
- The paper presents the In-Context Autoencoder (ICAE), a novel approach for context compression in large language models (LLMs)
- The ICAE consists of two modules: a learnable encoder adapted from an LLM using LoRA and a fixed decoder that is the target LLM
- The authors pretrain the ICAE using autoencoding and language modeling objectives on extensive text data
- They fine-tune the pretrained ICAE using a small amount of instructive data to enhance its interaction with different prompts and produce desirable responses
- Experimental results show that the proposed pretraining and fine-tuning paradigm enables the ICAE to produce memory slots with 4 times context compression
- These compressed memory slots can be well conditioned on by the target LLM to respond to various prompts
- The ICAE has significant implications in addressing long context problems and reducing computation and memory overheads in LLM inference
- The authors plan to release their code and data soon, contributing to further research efforts in advancing context management in large language models.
Abstract: We propose the In-context Autoencoder (ICAE) for context compression in a large language model (LLM). The ICAE has two modules: a learnable encoder adapted with LoRA from an LLM for compressing a long context into a limited number of memory slots, and a fixed decoder which is the target LLM that can condition on the memory slots for various purposes. We first pretrain the ICAE using both autoencoding and language modeling objectives on massive text data, enabling it to generate memory slots that accurately and comprehensively represent the original context. Then, we fine-tune the pretrained ICAE on a small amount of instruct data to enhance its interaction with various prompts for producing desirable responses. Our experimental results demonstrate that the ICAE learned with our proposed pretraining and fine-tuning paradigm can effectively produce memory slots with $4\times$ context compression, which can be well conditioned on by the target LLM to respond to various prompts. The promising results demonstrate significant implications of the ICAE for its novel approach to the long context problem and its potential to reduce computation and memory overheads for LLM inference in practice, suggesting further research effort in context management for an LLM. Our code and data will be released shortly.
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)
By clicking on the button above, our algorithm compares the title and abstract of this paper with all the papers of the same arXiv category that we have summarized, and finds the closest ones. You can rerun it from time to time: the results improve 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.