paper-with-me

홈 › Papers

Context Awareness Gate For Retrieval Augmented Generation

2024-11-25 · Mohammad Hassan Heydari, Arshia Hemmat, Erfan Naman, Afsaneh Fatemi

Retrieval Augmented Generation (RAG) has emerged as a widely adopted approach to mitigate the limitations of large language models (LLMs) in answering domain-specific questions. Previous research has predominantly focused on improving the accuracy and quality of retrieved data chunks to enhance the overall performance of the generation pipeline. However, despite ongoing advancements, the critical issue of retrieving irrelevant information -- which can impair the ability of the model to utilize its internal knowledge effectively -- has received minimal attention. In this work, we investigate the impact of retrieving irrelevant information in open-domain question answering, highlighting its significant detrimental effect on the quality of LLM outputs. To address this challenge, we propose the Context Awareness Gate (CAG) architecture, a novel mechanism that dynamically adjusts the LLMs' input prompt based on whether the user query necessitates external context retrieval. Additionally, we introduce the Vector Candidates method, a core mathematical component of CAG that is statistical, LLM-independent, and highly scalable. We further examine the distributions of relationships between contexts and questions, presenting a statistical analysis of these distributions. This analysis can be leveraged to enhance the context retrieval process in Retrieval Augmented Generation (RAG) systems.

📄 PDF Abstract BibTeX arXiv:2411.16133

Code (1)

heydaari/CAG 공식 구현 jax

Tasks

Open-Domain Question AnsweringQuestion AnsweringRAGRetrievalRetrieval-augmented Generation

Similar Papers 제목 키워드 기반

No Free Lunch: Retrieval-Augmented Generation Undermines Fairness in LLMs, Even for Vigilant Users

2024-10-10 · Mengxuan Hu, Hongyi Wu, Zihan Guan, Ronghang Zhu 외

Retrieval-Augmented Generation (RAG) is widely adopted for its effectiveness and cost-efficiency in mitigating hallucinations and enhancing the domain-specific generation capabilities of large language models (LLMs). How…

FairnessRAGRetrievalRetrieval-augmented Generation

PEAR: Position-Embedding-Agnostic Attention Re-weighting Enhances Retrieval-Augmented Generation with Zero Inference Overhead

2024-09-29 · Tao Tan, Yining Qian, Ang Lv, Hongzhan Lin 외

Large language models (LLMs) enhanced with retrieval-augmented generation (RAG) have introduced a new paradigm for web search. However, the limited context awareness of LLMs degrades their performance on RAG tasks. Exist…

PositionRAGRetrievalRetrieval-augmented Generation

HASH-RAG: Bridging Deep Hashing with Retriever for Efficient, Fine Retrieval and Augmented Generation

2025-05-22 · Jinyu Guo, Xunlei Chen, Qiyang Xia, Zhaokun Wang 외

Retrieval-Augmented Generation (RAG) encounters efficiency challenges when scaling to massive knowledge bases while preserving contextual relevance. We propose Hash-RAG, a framework that integrates deep hashing technique…

ChunkingDeep HashingPrompt EngineeringRAG+3

RPO: Retrieval Preference Optimization for Robust Retrieval-Augmented Generation

2025-01-23 · Shi-Qi Yan, Zhen-Hua Ling

While Retrieval-Augmented Generation (RAG) has exhibited promise in utilizing external knowledge, its generation process heavily depends on the quality and accuracy of the retrieved context. Large language models (LLMs) …

MemorizationRAGResponse GenerationRetrieval+1

AttentionRetriever: Attention Layers are Secretly Long Document Retrievers

2026-02-12 · David Jiahao Fu, Lam Thanh Do, Jiayu Li, Kevin Chen-Chuan Chang arxiv

Retrieval augmented generation (RAG) has been widely adopted to help Large Language Models (LLMs) to process tasks involving long documents. However, existing retrieval models are not designed for long document retrieval…