arXiv:2410.15805cs.AI2024-10EMNLP被引 15

面向企业运维的可微调RAG框架,提升私有数据问答性能

RAG4ITOps: A Supervised Fine-Tunable and Comprehensive RAG Framework for IT Operations and Maintenance

  • 采用对比学习与检索增强微调,优化嵌入模型和大模型
  • 在云服务领域数据上,问答准确率显著优于现有方法
  • 适合需要数据安全与持续迭代的企业级运维系统

随着企业运维问答系统需求不断增长,亟需一种高效且可监督微调的框架,以保障数据安全、支持私有化部署和持续升级。尽管大语言模型显著提升了开放域问答性能,但如何高效处理企业专属语料并构建领域专用问答系统,在工业应用中仍研究不足。本文提出一个基于检索增强生成(RAG)的通用综合框架——RAG4ITOps,涵盖两个阶段:(1) 模型微调与数据向量化;(2) 在线问答系统服务。第一阶段采用对比学习结合两种负采样策略微调嵌入模型,并设计指令模板,通过检索增强微调方法对大模型进行优化;第二阶段构建高效问答服务流程。我们从云计算领域收集企业专属语料,大量实验表明,该方法在两类问答任务上均优于现有方法。实验还验证了RAG4ITOps在真实企业级场景中的可行性。

原文摘要 · Abstract (English)

With the ever-increasing demands on Question Answering (QA) systems for IT operations and maintenance, an efficient and supervised fine-tunable framework is necessary to ensure the data security, private deployment and continuous upgrading. Although Large Language Models (LLMs) have notably improved the open-domain QA's performance, how to efficiently handle enterprise-exclusive corpora and build domain-specific QA systems are still less-studied for industrial applications. In this paper, we propose a general and comprehensive framework based on Retrieval Augmented Generation (RAG) and facilitate the whole business process of establishing QA systems for IT operations and maintenance. In accordance with the prevailing RAG method, our proposed framework, named with RAG4ITOps, composes of two major stages: (1) Models Fine-tuning \& Data Vectorization, and (2) Online QA System Process. At the Stage 1, we leverage a contrastive learning method with two negative sampling strategies to fine-tune the embedding model, and design the instruction templates to fine-tune the LLM with a Retrieval Augmented Fine-Tuning method. At the Stage 2, an efficient process of QA system is built for serving. We collect enterprise-exclusive corpora from the domain of cloud computing, and the extensive experiments show that our method achieves superior results than counterparts on two kinds of QA tasks. Our experiment also provide a case for applying the RAG4ITOps to real-world enterprise-level applications.

RAG运维问答微调企业应用

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。