In-context Autoencoder for Context Compression in a Large Language Model

AI-generated keywords: In-context Autoencoder LoRA Pretraining Fine-tuning LLM

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.
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Tao Ge, Jing Hu, Xun Wang, Si-Qing Chen, Furu Wei

Work in progress

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.

Submitted to arXiv on 13 Jul. 2023

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.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2307.06945v1

This paper's license doesn't allow us to build upon its content and the summarizing process is here made with the paper's metadata rather than the 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 which compresses a long context into a limited number of memory slots, and a fixed decoder that is the target LLM capable of conditioning on these memory slots for various purposes. To train the ICAE, the authors first pretrain it using both autoencoding and language modeling objectives on extensive text data. This enables the model to generate memory slots that accurately represent the original context. Then 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 demonstrate that the proposed pretraining and fine-tuning paradigm effectively 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 promising results highlight significant implications of the ICAE in addressing long context problems and reducing computation and memory overheads in LLM inference. The authors plan to release their code and data soon. Overall, this research contributes to advancing context management in large language models, suggesting further research efforts in this area.
Created on 17 Jul. 2023

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

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 representation

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while 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.