Context Embeddings for Efficient Answer Generation in RAG

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Rau, David, Wang, Shuai, Déjean, Hervé, Clinchant, Stéphane
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913566776360960
author Rau, David
Wang, Shuai
Déjean, Hervé
Clinchant, Stéphane
author_facet Rau, David
Wang, Shuai
Déjean, Hervé
Clinchant, Stéphane
contents Retrieval-Augmented Generation (RAG) allows overcoming the limited knowledge of LLMs by extending the input with external information. As a consequence, the contextual inputs to the model become much longer which slows down decoding time directly translating to the time a user has to wait for an answer. We address this challenge by presenting COCOM, an effective context compression method, reducing long contexts to only a handful of Context Embeddings speeding up the generation time by a large margin. Our method allows for different compression rates trading off decoding time for answer quality. Compared to earlier methods, COCOM allows for handling multiple contexts more effectively, significantly reducing decoding time for long inputs. Our method demonstrates a speed-up of up to 5.69 $\times$ while achieving higher performance compared to existing efficient context compression methods.
format Preprint
id arxiv_https___arxiv_org_abs_2407_09252
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Context Embeddings for Efficient Answer Generation in RAG
Rau, David
Wang, Shuai
Déjean, Hervé
Clinchant, Stéphane
Computation and Language
Information Retrieval
Retrieval-Augmented Generation (RAG) allows overcoming the limited knowledge of LLMs by extending the input with external information. As a consequence, the contextual inputs to the model become much longer which slows down decoding time directly translating to the time a user has to wait for an answer. We address this challenge by presenting COCOM, an effective context compression method, reducing long contexts to only a handful of Context Embeddings speeding up the generation time by a large margin. Our method allows for different compression rates trading off decoding time for answer quality. Compared to earlier methods, COCOM allows for handling multiple contexts more effectively, significantly reducing decoding time for long inputs. Our method demonstrates a speed-up of up to 5.69 $\times$ while achieving higher performance compared to existing efficient context compression methods.
title Context Embeddings for Efficient Answer Generation in RAG
topic Computation and Language
Information Retrieval
url https://arxiv.org/abs/2407.09252