为化学领域设计首个RAG评估基准,提升大模型知识推理能力
Benchmarking Retrieval-Augmented Generation for Chemistry
- 构建化学专用检索增强生成框架,集成文献与数据库等多源知识
- 相比直接推理,平均性能提升17.4%,验证RAG在化学任务中的有效性
- 提供可扩展工具包,适合化学信息检索与AI辅助科研人员使用
检索增强生成(RAG)作为提升大语言模型外部知识能力的有力框架,在科学领域尤其适用于需要专业动态信息的场景。然而,由于缺乏高质量、领域特定的语料库和系统化评估基准,其在化学领域的应用仍不充分。本文提出ChemRAG-Bench,一个全面的基准,用于系统评估RAG在多样化化学任务中的表现。配套的化学语料库整合了科学文献、PubChem数据库、PubMed摘要、教科书及维基百科条目等异构知识源。同时,我们发布ChemRAG-Toolkit,一个模块化且可扩展的RAG工具包,支持五种检索算法和八种大语言模型。利用该工具包,我们证明了RAG显著提升性能——相比直接推理方法,平均相对提升达17.4%。进一步分析了检索器架构、语料选择及检索段落数量的影响,提出了指导未来研究与部署的实用建议。代码与数据可在https://chemrag.github.io获取。
原文摘要 · Abstract (English)
Retrieval-augmented generation (RAG) has emerged as a powerful framework for enhancing large language models (LLMs) with external knowledge, particularly in scientific domains that demand specialized and dynamic information. Despite its promise, the application of RAG in the chemistry domain remains underexplored, primarily due to the lack of high-quality, domain-specific corpora and well-curated evaluation benchmarks. In this work, we introduce ChemRAG-Bench, a comprehensive benchmark designed to systematically assess the effectiveness of RAG across a diverse set of chemistry-related tasks. The accompanying chemistry corpus integrates heterogeneous knowledge sources, including scientific literature, the PubChem database, PubMed abstracts, textbooks, and Wikipedia entries. In addition, we present ChemRAG-Toolkit, a modular and extensible RAG toolkit that supports five retrieval algorithms and eight LLMs. Using ChemRAG-Toolkit, we demonstrate that RAG yields a substantial performance gain -- achieving an average relative improvement of 17.4% over direct inference methods. We further conduct in-depth analyses on retriever architectures, corpus selection, and the number of retrieved passages, culminating in practical recommendations to guide future research and deployment of RAG systems in the chemistry domain. The code and data is available at https://chemrag.github.io.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。