RAGPerf可全面评测检索增强生成系统性能,支持灵活配置与多场景模拟。
RAGPerf: An End-to-End Benchmarking Framework for Retrieval-Augmented Generation Systems
- 将RAG流程拆解为嵌入、索引、检索等模块化组件,支持独立参数调节
- 在真实数据集上实测显示端到端吞吐量高,内存占用低,性能开销可忽略
- 适合研究者和工程师优化RAG系统,尤其关注性能与准确率平衡的场景
我们提出并实现了RAGPerf框架,用于对基于检索增强生成(RAG)的AI系统进行端到端基准测试,以刻画RAG流水线的系统行为。为支持细致分析与细粒度性能评估,RAGPerf将RAG工作流解耦为多个模块化组件:嵌入、索引、检索、重排序和生成。用户可灵活配置各组件核心参数,观察其对端到端查询性能与质量的影响。RAGPerf内置工作负载生成器,支持多种数据类型(如文本、PDF、代码、音频),不同检索与更新比例,以及多样化的查询分布,模拟真实应用场景。同时支持多种嵌入模型、主流向量数据库(LanceDB、Milvus、Qdrant、Chroma、Elasticsearch)及不同LLM进行内容生成。框架自动采集性能指标(如端到端查询吞吐量、主机/GPU内存占用、CPU/GPU利用率)和准确率指标(如上下文召回率、查询准确率、事实一致性)。通过一系列实验验证了RAGPerf的能力,并开源代码至GitHub。评估表明,RAGPerf自身引入的性能开销可忽略不计。
原文摘要 · Abstract (English)
We present the design and implementation of a RAG-based AI system benchmarking (RAGPerf) framework for characterizing the system behaviors of RAG pipelines. To facilitate detailed profiling and fine-grained performance analysis, RAGPerf decouples the RAG workflow into several modular components - embedding, indexing, retrieval, reranking, and generation. RAGPerf offers the flexibility for users to configure the core parameters of each component and examine their impact on the end-to-end query performance and quality. RAGPerf has a workload generator to model real-world scenarios by supporting diverse datasets (e.g., text, pdf, code, and audio), different retrieval and update ratios, and query distributions. RAGPerf also supports different embedding models, major vector databases such as LanceDB, Milvus, Qdrant, Chroma, and Elasticsearch, as well as different LLMs for content generation. It automates the collection of performance metrics (i.e., end-to-end query throughput, host/GPU memory footprint, and CPU/GPU utilization) and accuracy metrics (i.e., context recall, query accuracy, and factual consistency). We demonstrate the capabilities of RAGPerf through a comprehensive set of experiments and open source its codebase at GitHub. Our evaluation shows that RAGPerf incurs negligible performance overhead.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。