CrAM: Credibility-Aware Attention Modification in LLMs for Combating Misinformation in RAG

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Deng, Boyi, Wang, Wenjie, Zhu, Fengbin, Wang, Qifan, Feng, Fuli
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916527536603136
author Deng, Boyi
Wang, Wenjie
Zhu, Fengbin
Wang, Qifan
Feng, Fuli
author_facet Deng, Boyi
Wang, Wenjie
Zhu, Fengbin
Wang, Qifan
Feng, Fuli
contents Retrieval-Augmented Generation (RAG) can alleviate hallucinations of Large Language Models (LLMs) by referencing external documents. However, the misinformation in external documents may mislead LLMs' generation. To address this issue, we explore the task of "credibility-aware RAG", in which LLMs automatically adjust the influence of retrieved documents based on their credibility scores to counteract misinformation. To this end, we introduce a plug-and-play method named $\textbf{Cr}$edibility-aware $\textbf{A}$ttention $\textbf{M}$odification (CrAM). CrAM identifies influential attention heads in LLMs and adjusts their attention weights based on the credibility of the documents, thereby reducing the impact of low-credibility documents. Experiments on Natual Questions and TriviaQA using Llama2-13B, Llama3-8B, and Qwen1.5-7B show that CrAM improves the RAG performance of LLMs against misinformation pollution by over 20%, even surpassing supervised fine-tuning methods.
format Preprint
id arxiv_https___arxiv_org_abs_2406_11497
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle CrAM: Credibility-Aware Attention Modification in LLMs for Combating Misinformation in RAG
Deng, Boyi
Wang, Wenjie
Zhu, Fengbin
Wang, Qifan
Feng, Fuli
Computation and Language
Retrieval-Augmented Generation (RAG) can alleviate hallucinations of Large Language Models (LLMs) by referencing external documents. However, the misinformation in external documents may mislead LLMs' generation. To address this issue, we explore the task of "credibility-aware RAG", in which LLMs automatically adjust the influence of retrieved documents based on their credibility scores to counteract misinformation. To this end, we introduce a plug-and-play method named $\textbf{Cr}$edibility-aware $\textbf{A}$ttention $\textbf{M}$odification (CrAM). CrAM identifies influential attention heads in LLMs and adjusts their attention weights based on the credibility of the documents, thereby reducing the impact of low-credibility documents. Experiments on Natual Questions and TriviaQA using Llama2-13B, Llama3-8B, and Qwen1.5-7B show that CrAM improves the RAG performance of LLMs against misinformation pollution by over 20%, even surpassing supervised fine-tuning methods.
title CrAM: Credibility-Aware Attention Modification in LLMs for Combating Misinformation in RAG
topic Computation and Language
url https://arxiv.org/abs/2406.11497