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)의 효율적인 서빙에 대한 연구를 다룬다. 다양한 RAG 변형과 상이한 작업 부하 특성으로 인해 효율적인 RAG 서빙이 어려움을 겪고 있는 현실을 고려하여, 본 논문에서는 RAG 알고리즘의 폭넓은 범위를 포착하는 구조적 추상화인 RAGSchema를 제시하고, 다양한 RAG 작업 부하를 분석하여 성능 변동성을 밝힌다. 또한 이러한 변동성을 해결하고 다양한 성능 요구사항을 충족하기 위해 효율적인 RAG 서빙을 위한 시스템 최적화 프레임워크인 RAGO를 제안한다. 실험 결과, RAGO는 LLM 시스템 확장 기반 RAG 시스템에 비해 칩당 QPS를 최대 2배 향상시키고 첫 토큰 지연 시간을 55% 단축시키는 것으로 나타났다.

시사점, 한계점

시사점:
RAGSchema를 통해 다양한 RAG 알고리즘을 효율적으로 관리하고 최적화할 수 있는 기반을 마련하였다.
RAG 작업 부하의 성능 변동성을 분석하여 효율적인 서빙 전략 수립에 중요한 정보를 제공하였다.
RAGO를 통해 기존 RAG 시스템 대비 성능을 크게 향상시킬 수 있음을 실험적으로 증명하였다.
한계점:
RAGSchema가 모든 RAG 알고리즘을 완벽하게 포괄하는지는 추가적인 연구가 필요하다.
RAGO의 성능 향상 효과는 특정 하드웨어 및 작업 부하 환경에 국한될 수 있다.
다양한 RAG 알고리즘과 작업 부하에 대한 포괄적인 분석이 부족할 수 있다.
👍