Daily Arxiv

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

RAGO: Systematic Performance Optimization for Retrieval-Augmented Generation Serving

Created by
  • Haebom
Category
Empty

저자

Wenqi Jiang, Suvinay Subramanian, Cat Graves, Gustavo Alonso, Amir Yazdanbakhsh, Vidushi Dadu

개요

본 논문은 대규모 언어 모델(LLM)과 외부 지식 데이터베이스 검색을 결합한 검색 증강 생성(RAG)의 효율적인 서빙을 위한 시스템 최적화 프레임워크인 RAGO를 제시합니다. RAG의 다양한 변형과 상이한 작업 부하 특성으로 인해 효율적인 RAG 서빙이 어려운 점을 해결하기 위해, RAG 알고리즘의 폭넓은 범위를 포착하는 구조적 추상화인 RAGSchema를 도입하고, 다양한 RAG 작업 부하를 분석하여 성능 변동성을 밝힙니다. RAGO는 이러한 변동성을 해결하고 다양한 성능 요구 사항을 충족하도록 설계되었습니다. 평가 결과, RAGO는 LLM 시스템 확장 기반 RAG 시스템에 비해 칩당 QPS를 최대 2배 향상시키고 첫 토큰 대기 시간을 55% 단축하는 것으로 나타났습니다.

시사점, 한계점

시사점:
RAGSchema를 통한 RAG 알고리즘의 체계적인 분류 및 성능 최적화 기반 마련.
다양한 RAG 작업 부하의 성능 변동성 분석을 통해 효율적인 서빙 전략 수립 가능성 제시.
RAGO를 통한 RAG 서빙 성능 향상 (최대 2배 QPS 증가, 첫 토큰 대기 시간 55% 감소).
한계점:
RAGSchema의 포괄성 및 적용 가능한 RAG 알고리즘의 범위에 대한 추가적인 검증 필요.
RAGO의 성능 향상 효과는 특정 하드웨어 및 작업 부하 환경에 의존적일 수 있음.
다양한 RAG 변형에 대한 일반화 가능성 및 확장성에 대한 추가 연구 필요.
👍