Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

AttentionRAG: Attention-Guided Context Pruning in Retrieval-Augmented Generation

Created by
  • Haebom
Category
Empty

저자

Yixiong Fang, Tianran Sun, Yuling Shi, Xiaodong Gu

개요

본 논문은 Retrieval Augmented Generation (RAG) 시스템의 효율성을 향상시키기 위한 새로운 문맥 자르기 방법인 AttentionRAG를 제안합니다. 기존 방법인 LLMLingua는 문맥 인식이 부족하고 압축률 조절이 어려워 과도한 정보 손실 또는 불충분한 자르기 문제를 야기하는 반면, AttentionRAG는 어텐션 집중 메커니즘을 통해 RAG 질의를 다음 토큰 예측 패러다임으로 재구성하여 질의의 의미적 초점을 단일 토큰으로 분리함으로써 질의와 검색된 문맥 간의 정확하고 효율적인 어텐션 계산을 가능하게 합니다. LongBench와 Babilong 벤치마크를 이용한 실험 결과, AttentionRAG는 최대 6.3배의 문맥 압축률을 달성하면서 주요 지표에서 LLMLingua보다 약 10% 향상된 성능을 보였습니다.

시사점, 한계점

시사점:
RAG 시스템의 효율성을 크게 향상시킬 수 있는 새로운 문맥 자르기 방법 제시.
기존 방법보다 높은 압축률과 향상된 성능 달성.
어텐션 메커니즘을 활용한 정교하고 효율적인 문맥 자르기 가능.
한계점:
제안된 방법의 일반화 성능에 대한 추가적인 연구 필요.
다양한 RAG 시스템 및 데이터셋에 대한 실험 결과 확대 필요.
특정 토큰에 대한 집중으로 인한 의미적 정보 손실 가능성.
👍