Towards Knowledge Checking in Retrieval-augmented Generation: A Representation Perspective

TL;DR

This work aims to provide a systematic study on knowledge checking in RAG systems, conducting a comprehensive analysis of LLM representation behaviors and demonstrating the significance of using representations in knowledge checking.

Abstract

Retrieval-Augmented Generation (RAG) systems have shown promise in enhancing the performance of Large Language Models (LLMs). However, these systems face challenges in effectively integrating external knowledge with the LLM's internal knowledge, often leading to issues with misleading or unhelpful information. This work aims to provide a systematic study on knowledge checking in RAG systems. We conduct a comprehensive analysis of LLM representation behaviors and demonstrate the significance of using representations in knowledge checking. Motivated by the findings, we further develop representation-based classifiers for knowledge filtering. We show substantial improvements in RAG performance, even when dealing with noisy knowledge databases. Our study provides new insights into leveraging LLM representations for enhancing the reliability and effectiveness of RAG systems.

Figures (8)

• Figure 1: ROC curve of probability-based methods
• Figure 2: Visualization on PCA space
• Figure 3: Visualization of contrastive scores
• Figure 4: Filtering results
• Figure 5: Accuracy on different layers