RAG Foundry: A Framework for Enhancing LLMs for Retrieval Augmented Generation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Fleischer, Daniel, Berchansky, Moshe, Wasserblat, Moshe, Izsak, Peter
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929449688104960
author Fleischer, Daniel
Berchansky, Moshe
Wasserblat, Moshe
Izsak, Peter
author_facet Fleischer, Daniel
Berchansky, Moshe
Wasserblat, Moshe
Izsak, Peter
contents Implementing Retrieval-Augmented Generation (RAG) systems is inherently complex, requiring deep understanding of data, use cases, and intricate design decisions. Additionally, evaluating these systems presents significant challenges, necessitating assessment of both retrieval accuracy and generative quality through a multi-faceted approach. We introduce RAG Foundry, an open-source framework for augmenting large language models for RAG use cases. RAG Foundry integrates data creation, training, inference and evaluation into a single workflow, facilitating the creation of data-augmented datasets for training and evaluating large language models in RAG settings. This integration enables rapid prototyping and experimentation with various RAG techniques, allowing users to easily generate datasets and train RAG models using internal or specialized knowledge sources. We demonstrate the framework effectiveness by augmenting and fine-tuning Llama-3 and Phi-3 models with diverse RAG configurations, showcasing consistent improvements across three knowledge-intensive datasets. Code is released as open-source in https://github.com/IntelLabs/RAGFoundry.
format Preprint
id arxiv_https___arxiv_org_abs_2408_02545
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle RAG Foundry: A Framework for Enhancing LLMs for Retrieval Augmented Generation
Fleischer, Daniel
Berchansky, Moshe
Wasserblat, Moshe
Izsak, Peter
Computation and Language
Artificial Intelligence
Information Retrieval
Machine Learning
Implementing Retrieval-Augmented Generation (RAG) systems is inherently complex, requiring deep understanding of data, use cases, and intricate design decisions. Additionally, evaluating these systems presents significant challenges, necessitating assessment of both retrieval accuracy and generative quality through a multi-faceted approach. We introduce RAG Foundry, an open-source framework for augmenting large language models for RAG use cases. RAG Foundry integrates data creation, training, inference and evaluation into a single workflow, facilitating the creation of data-augmented datasets for training and evaluating large language models in RAG settings. This integration enables rapid prototyping and experimentation with various RAG techniques, allowing users to easily generate datasets and train RAG models using internal or specialized knowledge sources. We demonstrate the framework effectiveness by augmenting and fine-tuning Llama-3 and Phi-3 models with diverse RAG configurations, showcasing consistent improvements across three knowledge-intensive datasets. Code is released as open-source in https://github.com/IntelLabs/RAGFoundry.
title RAG Foundry: A Framework for Enhancing LLMs for Retrieval Augmented Generation
topic Computation and Language
Artificial Intelligence
Information Retrieval
Machine Learning
url https://arxiv.org/abs/2408.02545