Saved in:
Bibliographic Details
Main Authors: Chung, Isaac, Vo, Phat, Kizilkale, Arman C., Reite, Aaron
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2409.17648
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910654562041856
author Chung, Isaac
Vo, Phat
Kizilkale, Arman C.
Reite, Aaron
author_facet Chung, Isaac
Vo, Phat
Kizilkale, Arman C.
Reite, Aaron
contents Retrieval Augmented Generation (RAG) is a common method for integrating external knowledge into pretrained Large Language Models (LLMs) to enhance accuracy and relevancy in question answering (QA) tasks. However, prompt engineering and resource efficiency remain significant bottlenecks in developing optimal and robust RAG solutions for real-world QA applications. Recent studies have shown success in using fine tuning to address these problems; in particular, Retrieval Augmented Fine Tuning (RAFT) applied to smaller 7B models has demonstrated superior performance compared to RAG setups with much larger models such as GPT-3.5. The combination of RAFT with parameter-efficient fine tuning (PEFT) techniques, such as Low-Rank Adaptation (LoRA), promises an even more efficient solution, yet remains an unexplored area. In this work, we combine RAFT with LoRA to reduce fine tuning and storage requirements and gain faster inference times while maintaining comparable RAG performance. This results in a more compute-efficient RAFT, or CRAFT, which is particularly useful for knowledge-intensive QA tasks in resource-constrained environments where internet access may be restricted and hardware resources limited.
format Preprint
id arxiv_https___arxiv_org_abs_2409_17648
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient In-Domain Question Answering for Resource-Constrained Environments
Chung, Isaac
Vo, Phat
Kizilkale, Arman C.
Reite, Aaron
Computation and Language
Retrieval Augmented Generation (RAG) is a common method for integrating external knowledge into pretrained Large Language Models (LLMs) to enhance accuracy and relevancy in question answering (QA) tasks. However, prompt engineering and resource efficiency remain significant bottlenecks in developing optimal and robust RAG solutions for real-world QA applications. Recent studies have shown success in using fine tuning to address these problems; in particular, Retrieval Augmented Fine Tuning (RAFT) applied to smaller 7B models has demonstrated superior performance compared to RAG setups with much larger models such as GPT-3.5. The combination of RAFT with parameter-efficient fine tuning (PEFT) techniques, such as Low-Rank Adaptation (LoRA), promises an even more efficient solution, yet remains an unexplored area. In this work, we combine RAFT with LoRA to reduce fine tuning and storage requirements and gain faster inference times while maintaining comparable RAG performance. This results in a more compute-efficient RAFT, or CRAFT, which is particularly useful for knowledge-intensive QA tasks in resource-constrained environments where internet access may be restricted and hardware resources limited.
title Efficient In-Domain Question Answering for Resource-Constrained Environments
topic Computation and Language
url https://arxiv.org/abs/2409.17648