找回密码
 立即注册
首页 业界区 业界 从零到一:利用金仓社区数据,LoRa微调与Spring AI 构建 ...

从零到一:利用金仓社区数据,LoRa微调与Spring AI 构建私有化千问模型

嗅叽 2025-6-2 23:16:48
上次我们在Coze平台上成功搭建了一个针对金仓问题的解决助手。这个智能体的核心工作流程相对简单:每次它通过HTTP接口调用插件,在金仓平台内部进行搜索,随后利用大模型的推理能力对查询结果进行分析,从而为用户提供问题解答。然而,问题也随之而来——金仓内部的搜索功能存在一定的局限性。搜索出来的博文资料之间的关联性并不强,这使得智能体在回答问题时,往往需要检索大量、冗长的资料,甚至有时需要检索上万字的内容才能找到有价值的信息,从而影响了整体的工作效率。
1.png

针对这一问题,我们的目标是优化和提高智能体回答问题的准确性与效率。通过深入分析,问题的本质其实就是如何更有效地利用已有资料进行推理和回答。尽管我们无法直接改变金仓社区的搜索机制,但我们可以采取一种间接的方式:提前对社区开放的文档或博文进行检索,并基于这些内容对已有的大模型进行微调。通过这种方法,大模型在处理用户问题时可以更高效地利用已知信息,进而提供更精准的答案,达到与直接优化搜索机制类似的效果。
好的,今天我们的主要目标是利用现有的资料对Qwen2.5-7B-Instruct开源模型进行Lora微调,旨在使其能够真正地在离线环境下回答用户的问题。考虑到社区中关于该模型的博文和文档资料非常丰富,我们今天将聚焦在一特定领域,我们会专注于数据库迁移方面的训练,确保模型能够在没有网络连接的情况下,依然具备有效的知识查询与处理能力。
什么是LoRa

智能体(AI)技术已经广泛应用并逐渐深入人们的日常生活,大家对它有了较为全面的了解。然而,当前大规模语言模型的微调技术尚未真正普及,其原因主要在于高门槛的问题。首先,微调大模型不仅需要强大的计算资源,像高性能的服务器、GPU等硬件设备,而且还需要解决诸如各种复杂的参数设置等技术难题,这使得很多企业和个人难以承受。
此外,真正从零开始训练一个大模型,所需的资源和成本也是常规公司难以负担的,因此目前广泛采用的策略是基于已有的预训练模型进行再训练(fine-tuning)。这种方法充分利用了现有大模型的基础架构,再加上企业内部的专有数据进行定制化调整,能够大大降低训练成本并提高模型的适应性和效率。一个典型的技术就是LoRa(Low-Rank Adaptation)技术,它通过优化参数空间,使得企业能够在不重新训练整个模型的情况下,针对特定任务进行快速微调。
这部分技术内容并不是我们目前关注的重点,因此在这里就不再详细展开。
环境准备

要想使用资料库数据进行微调,我们需要做好充分的准备。首先,必须准备好内部资料数据,确保数据的质量和多样性,能够覆盖所需的业务场景或问题。其次,还需要一台配备GPU的服务器,确保能够提供足够的计算能力来处理大量数据,并支持深度学习模型的训练和优化。
资料准备

我们主要致力于针对数据库迁移问题的解决方案进行训练,为了高效获取相关的资料和内容,我们直接访问金仓社区,进行检索并下载与数据库迁移相关的最新数据和文档,具体操作步骤如图所示:
2.png

数据下载

接下来,我们将直接下载相关的文档或文库,以便更好地了解和使用相关资源,具体操作如下图所示:

另外,还有一些资料来源,例如博客、论坛等在线资源。我们可以直接选择使用爬虫进行抓取,但为了避免对社区服务器造成过多负担,我们决定采取更为温和的方式,即将相关数据直接保存到本地,随后进行解析和信息提取。具体操作如图所示:

指令集构建

好的,有了这些基本数据之后,接下来我们需要将其解析成我们所需的数据集格式,以便进行后续的处理和分析。LLM(大语言模型)的微调,通常是指指令微调过程。所谓指令微调,指的是通过特定格式的微调数据来引导模型学习如何根据给定的指令进行响应。这些微调数据通常具有如下的结构和形式::
  1. [{
  2.     "instruction": "KES是什么?",
  3.     "input": "",
  4.     "output": "KES是人大金仓的简称。"
  5. }]
复制代码
其中,instruction 是用户指令,告知模型其需要完成的任务;input 是用户输入,是完成用户指令所必须的输入内容;output 是模型应该给出的输出。
既然我们已经获得了目标格式的内容,接下来需要做的就是对这些文档进行解析。为此,我们将采用Spring AI,它支持对多种文件格式的解析,包括但不限于PDF、HTML、Markdown、JSON、Docx等。当前,我们所处理的文档格式仅限于PDF和HTML,因此Spring AI完全能够满足我们的需求。
文档提取

首先需要创建一个新的 Spring Boot 项目,并在项目中引入所需的相关依赖。项目的基本结构安排如下所示:
5.png

接下来,我们将引入项目中所需的POM依赖,以确保相关的库和工具能够被正确加载和使用,具体的依赖项如下所示:
  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <project xmlns="http://maven.apache.org/POM/4.0.0"
  3.          xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  4.          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  5.     <modelVersion>4.0.0</modelVersion>
  6.     <parent>
  7.         <groupId>org.springframework.boot</groupId>
  8.         spring-boot-starter-parent</artifactId>
  9.         <version>3.2.0</version>
  10.         <relativePath/>
  11.     </parent>
  12.     02-example-RAG-backend</artifactId>
  13.     <properties>
  14.         <maven.compiler.source>17</maven.compiler.source>
  15.         <maven.compiler.target>17</maven.compiler.target>
  16.         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
  17.         <knife4j-version>4.1.0</knife4j-version>
  18.         <hutool-version>5.8.2</hutool-version>
  19.     </properties>
  20.     <dependencyManagement>
  21.         <dependencies>
  22.             <dependency>
  23.                 <groupId>org.springframework.ai</groupId>
  24.                 spring-ai-bom</artifactId>
  25.                 <version>1.0.0-M8</version>
  26.                 <type>pom</type>
  27.                 <scope>import</scope>
  28.             </dependency>
  29.         </dependencies>
  30.     </dependencyManagement>
  31.     <dependencies>
  32.         <dependency>
  33.             <groupId>org.springframework.boot</groupId>
  34.             spring-boot-starter-actuator</artifactId>
  35.         </dependency>
  36.         <dependency>
  37.             <groupId>org.springframework.boot</groupId>
  38.             spring-boot-starter-web</artifactId>
  39.         </dependency>
  40.         <dependency>
  41.             <groupId>com.github.xiaoymin</groupId>
  42.             knife4j-openapi3-jakarta-spring-boot-starter</artifactId>
  43.             <version>${knife4j-version}</version>
  44.         </dependency>
  45.         <dependency>
  46.             <groupId>org.projectlombok</groupId>
  47.             lombok</artifactId>
  48.             <optional>true</optional>
  49.         </dependency>
  50.         <dependency>
  51.             <groupId>org.springframework.ai</groupId>
  52.             spring-ai-tika-document-reader</artifactId>
  53.         </dependency>
  54.         <dependency>
  55.             <groupId>org.springframework.ai</groupId>
  56.             spring-ai-pdf-document-reader</artifactId>
  57.         </dependency>
  58.         <dependency>
  59.             <groupId>org.springframework.ai</groupId>
  60.             spring-ai-jsoup-document-reader</artifactId>
  61.         </dependency>
  62.         <dependency>
  63.             <groupId>cn.hutool</groupId>
  64.             hutool-all</artifactId>
  65.             <version>${hutool-version}</version>
  66.         </dependency>
  67.         <dependency>
  68.             <groupId>org.springframework.ai</groupId>
  69.             spring-ai-starter-model-openai</artifactId>
  70.         </dependency>
  71.     </dependencies>
  72.     <build>
  73.         <plugins>
  74.             <plugin>
  75.                 <groupId>org.springframework.boot</groupId>
  76.                 spring-boot-maven-plugin</artifactId>
  77.                 <configuration>
  78.                     <excludes>
  79.                         <exclude>
  80.                             <groupId>org.projectlombok</groupId>
  81.                             lombok</artifactId>
  82.                         </exclude>
  83.                     </excludes>
  84.                 </configuration>
  85.             </plugin>
  86.         </plugins>
  87.     </build>
  88.     <repositories>
  89.         <repository>
  90.             <id>spring-snapshots</id>
  91.             <name>Spring Snapshots</name>
  92.             <url>https://repo.spring.io/snapshot</url>
  93.             <releases>
  94.                 <enabled>false</enabled>
  95.             </releases>
  96.         </repository>
  97.         <repository>
  98.             <name>Central Portal Snapshots</name>
  99.             <id>central-portal-snapshots</id>
  100.             <url>https://central.sonatype.com/repository/maven-snapshots/</url>
  101.             <releases>
  102.                 <enabled>false</enabled>
  103.             </releases>
  104.             <snapshots>
  105.                 <enabled>true</enabled>
  106.             </snapshots>
  107.         </repository>
  108.     </repositories>
  109. </project>
复制代码
在这里,我们引入了当前 M8 版本的 Spring AI 以及 OpenAI 提供的相关模型接口。这是因为在完成文档提取后,我们需要将提取到的数据进行解析,转化为指令集格式。借助 AI我们能够显著减少传统手工数据清理所需的工作量。此外,我们还引入了与相关模型接口配套的秘钥配置信息,确保系统能够访问并使用这些服务。具体的配置如下所示:
  1. server.port=9166
  2. spring.ai.openai.base-url=
  3. spring.ai.openai.api-key=
  4. spring.ai.openai.chat.options.model=
复制代码
这里主要是配置OpenAI接口的相关信息。如果你没有国外的API访问权限,实际上国内的大部分大模型也能兼容这种接口配置,你可以直接填入相应的国内大模型接口,基本也能顺利调用。
接下来,我们将利用Spring AI的文档提取器对文档进行解析,具体的实现代码如下:
  1. public List<Document> getDocsFromHTML() throws IOException {
  2.     ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver(resourceLoader);
  3.     org.springframework.core.io.Resource[] resourceArray = resolver.getResources("classpath:/file/*.html");
  4.     List<Document> documentList = Arrays.stream(resourceArray)
  5.             .flatMap(resource -> {
  6.                 JsoupDocumentReader reader = new JsoupDocumentReader(resource,
  7.                         JsoupDocumentReaderConfig.builder()
  8.                                 .selector("p") // 可根据实际结构调整选择器
  9.                                 .charset("UTF-8")
  10.                                 .build());
  11.                 return reader.get().stream();
  12.             })
  13.             .collect(Collectors.toList());
  14.     return documentList;
  15. }
复制代码
这里主要解析如何从HTML文件中提取文本内容。由于下载的HTML文件通常包含了各种样式、脚本以及其他无关信息,因此我们仅关注其中的文本段落部分,即标签中的文字内容。除此之外,我们还讨论了如何从PDF文档中提取文本,具体实现方法如下所示:
  1. public List<Document> getDocsFromPDF() throws IOException {
  2.     ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver(resourceLoader);
  3.     org.springframework.core.io.Resource[] resourceArray = resolver.getResources("classpath:/file/*.{pdf,PDF,Pdf}");
  4.     List<Document> documentList = Arrays.stream(resourceArray)
  5.             .filter(resource -> resource.getFilename() != null &&
  6.                     resource.getFilename().toLowerCase().endsWith(".pdf"))
  7.             .flatMap(resource -> {
  8.                 //打印文件名字
  9.                 System.out.println("文件名字:" + resource.getFilename());
  10.                 try {
  11.                     ParagraphPdfDocumentReader pdfReader = new ParagraphPdfDocumentReader(resource,
  12.                             PdfDocumentReaderConfig.builder()
  13.                                     .withPageTopMargin(0)
  14.                                     .withPageExtractedTextFormatter(ExtractedTextFormatter.builder()
  15.                                             .withNumberOfTopTextLinesToDelete(0)
  16.                                             .build())
  17.                                     .withPagesPerDocument(1)
  18.                                     .build());
  19.                     return pdfReader.read().stream();
  20.                 }catch (Exception e){
  21.                     PagePdfDocumentReader pdfReader = new PagePdfDocumentReader(resource,
  22.                             PdfDocumentReaderConfig.builder()
  23.                                     .withPageTopMargin(0)
  24.                                     .withPageExtractedTextFormatter(ExtractedTextFormatter.builder()
  25.                                             .withNumberOfTopTextLinesToDelete(0)
  26.                                             .build())
  27.                                     .withPagesPerDocument(2)
  28.                                     .build());
  29.                     return pdfReader.read().stream();
  30.                 }
  31.             })
  32.             .collect(Collectors.toList());
  33.     return documentList;
  34. }
复制代码
在文本提取的过程中,我将其分为两种不同的格式进行处理。第一种是使用 ParagraphPdfDocumentReader 类,该类能够提取带有目录的PDF文件,并根据目录内容进行分批提取。另一种是处理没有目录的PDF文件,此时使用 PagePdfDocumentReader 类,该类会根据页数对PDF文件进行逐页文本提取。
无论是通过哪种方式提取文本,最终提取的内容都会被自动封装为适合向量数据库存储的 Document 格式。在此过程中,我们只关注文本内容,其他部分的内容将被忽略,确保AI能够高效地进行提取处理。具体代码实现如下所示:
  1. public List<InputData> aiByHTML() throws IOException {
  2.     List<Document> docsFromHTML = getDocsFromHTML();
  3.     TokenTextSplitter splitter = new TokenTextSplitter();
  4.     List<Document> documentList = splitter.apply(docsFromHTML);
  5. //        documentList = enricher.apply(documentList);
  6.     List<Future<List<InputData>>> futures = documentList.stream()
  7.             .map(doc -> executorService.submit(() -> processDocument(doc).toList()))
  8.             .collect(Collectors.toList());
  9.     List<InputData> results = new ArrayList<>();
  10.     for (Future<List<InputData>> future : futures) {
  11.         try {
  12.             results.addAll(future.get());
  13.         } catch (Exception e) {
  14.             log.error("Error processing document in thread: {}", e.getMessage());
  15.         }
  16.     }
  17.     return results;
  18. }
复制代码
由于HTML提取后的每篇博文或文档通常包含大量文字,因此我们需要使用 TokenTextSplitter 类对提取的文本进行切割。切割后的文本会被分成多个更小的块,以便更高效地处理。为了加快处理速度,本文还采用了线程池的方式,让AI能够并行地快速解析文档内容。具体的代码实现如下所示:
  1. private Stream<InputData> processDocument(Document document) {
  2.   String jsonStr = document.getText();
  3.   log.info("Processing document content: {}", jsonStr.length() > 100 ? jsonStr.substring(0, 100) : jsonStr);
  4.   try {
  5.       List<InputData> dataList = chatClient.prompt()
  6.               .system("""
  7.                   - Role: 数据清洗专家和信息提取工程师
  8.                   - Background: 用户需要从一系列包含KES人大金仓迁移内容的文档中提取关键信息,但文档中存在大量无关内容,需要高效地筛选出有价值的部分。
  9.                   - Profile: 你是一位资深的数据清洗专家,擅长从海量文本中提取关键信息,具备强大的文本分析能力和信息筛选能力,能够精准识别与KES人大金仓迁移相关的知识点、注意点和实践经验。
  10.                   - Skills: 你精通自然语言处理技术、文本挖掘方法和信息筛选策略,能够快速定位和提取文档中的关键内容,同时排除无关信息。
  11.                   - Goals: 从用户提供的文档中提取与KES人大金仓迁移相关的注意点、知识点和实践经验,必须去除无关内容,确保提取的信息准确、完整且有价值。
  12.                   - Constrains: 提取的信息应简洁明了,避免冗余,确保与KES人大金仓迁移直接相关,禁止包含无关内容。禁止出现本文、后台获取等字样。
  13.                   - OutputFormat: 以结构化的形式输出提取的关键信息,包括知识点、注意点和实践经验的分类整理。
  14.                   - Workflow:
  15.                     1. 仔细阅读用户提供的文档,理解文档的整体内容和结构。
  16.                     2. 确定与KES人大金仓迁移相关的关键词和主题,如“迁移注意点”“KES知识点”“迁移实践”等。
  17.                     3. 运用文本分析技术,必须从文档中提取与这些关键词和主题相关的内容,同时排除无关信息。
  18.                     4. 对提取的内容进行分类整理,形成结构化的输出。
  19.                     5. instruction为问题,input固定为"",output为问题答案,问题答案字数必须在200-400字范围之间且必须为Markdown格式。
  20.                   - Example:
  21.                       [{
  22.                           "instruction": "KES是什么?",
  23.                           "input": "",
  24.                           "output": "KES是人大金仓的简称。"
  25.                       }]
  26.                   """)
  27.               .user(jsonStr)
  28.               .options(ChatOptions.builder().temperature(0.8).build())
  29.               .call()
  30.               .entity(new ParameterizedTypeReference<List<InputData>>() {});
  31.       log.info("dataList size: {}", dataList != null ? dataList.size() : 0);
  32.       return dataList == null ? Stream.empty() : dataList.stream();
  33.   } catch (Exception e) {
  34.       log.error("Error processing document: {}", ExceptionUtil.stacktraceToString(e));
  35.       return Stream.empty();
  36.   }
  37. }
复制代码
我们通过简单的提示词来引导 AI 进行总结和归纳,生成一系列指令集。最后,利用结构化输出,将这些指令封装成具体的类,以便进一步使用和管理。具体的实现代码如下所示:
  1. public record InputData(String instruction, String input, String output) {}
复制代码
最后,我们将处理后的数组以JSON格式写入文件中,基本的流程就此完成。接下来,我们可以查看最终的提取效果,具体效果如图所示:
6.png

综合来看,数据集整体表现尚可,但提示词仍需要进一步优化。最终生成的数据集规模也不理想,只有几百条数据,基本上无法进行有效训练。为了展示整体效果,我多次运行相同的文档,最终凑足了2000多个数据样本。不过,这样的做法肯定存在一定问题。因此,如果时间允许,还是应该多下载一些文献或资料,以确保数据集的质量和规模足够支持更好的训练效果。
服务器准备

接下来,在解析完指令集后,我们需要进行大模型微调。首先,确保你有一台至少具备20GB GPU内存的服务器。如果没有,可以考虑在云端租赁一台。这里我使用的是AutoDL AI算力云,具体细节就不再赘述。
如果你没有微调的经验,我建议从零开始学习。我个人是在一个开源项目中积累经验的,开源项目地址如下:https://github.com/datawhalechina/self-llm
7.png

微调

接下来,我们需要租赁一台服务器,环境依赖可以直接使用社区提供的镜像。开源项目也提供了相关的镜像,大家可以根据自己的需求进行创建。最终的配置如下图所示:
8.png

接下来,我们将使用JupyterLab快速连接到服务器。如图所示:
9.png

我使用的是从Qwen2.5-7B拉取的镜像,该镜像已包含开源项目所需的所有文件和目录。如图所示:
10.png

接下来,我们只需将数据集上传至指定位置,并根据需要调整相关文件目录设置,具体操作步骤如下图所示:
11.png

代码只需简单修改数据集地址,由于数据集包含多个 JSON 文件,因此需要在加载所有文件后进行合并。以下是修改后的代码:
  1. import os
  2. import glob
  3. import pandas as pd
  4. from datasets import Dataset
  5. # 设置目录路径
  6. directory_path = 'kes'  # 替换为你的目录路径
  7. # 获取目录下所有的 JSON 文件
  8. json_files = glob.glob(os.path.join(directory_path, '*.json'))
  9. # 存储所有加载的 JSON 数据
  10. all_data = []
  11. # 遍历所有 JSON 文件并加载它们
  12. for json_file in json_files:
  13.     print(f"正在加载: {json_file}")
  14.     df = pd.read_json(json_file)  # 加载每个 JSON 文件为 DataFrame
  15.     all_data.append(df)
  16. # 合并所有的 DataFrame
  17. combined_df = pd.concat(all_data, ignore_index=True)
  18. # 将 DataFrame 转换为 Hugging Face Dataset
  19. ds = Dataset.from_pandas(combined_df)
复制代码
在对相关指令进行修改时,主要是将系统提示词更改为“金仓小助手”,具体修改内容如下所示:
  1. def process_func(example):
  2.     MAX_LENGTH = 384    # Llama分词器会将一个中文字切分为多个token,因此需要放开一些最大长度,保证数据的完整性
  3.     input_ids, attention_mask, labels = [], [], []
  4.     instruction = tokenizer(f"<|im_start|>system\n现在你是人大金仓的小助手<|im_end|>\n<|im_start|>user\n{example['instruction'] + example['input']}<|im_end|>\n<|im_start|>assistant\n", add_special_tokens=False)  # add_special_tokens 不在开头加 special_tokens
  5.     ## 省略重复代码
复制代码
接下来,将进入微调训练的过程,具体步骤如下图所示:
12.png

整个过程大约需要花费我们10分钟左右的时间,考虑到目前数据集的规模还相对较小,处理速度不会太慢。
效果演示

最后,训练完成后,我们将进行一个简单的测试,以评估Lora微调的效果。下面是相关的代码实现:
  1. from transformers import AutoModelForCausalLM, AutoTokenizer
  2. import torch
  3. from peft import PeftModel
  4. mode_path = '/root/autodl-tmp/qwen/Qwen2.5-7B-Instruct/'
  5. lora_path = './output/Qwen2.5_instruct_lora/checkpoint-450' # 这里改称你的 lora 输出对应 checkpoint 地址
  6. # 加载tokenizer
  7. tokenizer = AutoTokenizer.from_pretrained(mode_path, trust_remote_code=True)
  8. # 加载模型
  9. model = AutoModelForCausalLM.from_pretrained(mode_path, device_map="auto",torch_dtype=torch.bfloat16, trust_remote_code=True).eval()
  10. # 加载lora权重
  11. model = PeftModel.from_pretrained(model, model_id=lora_path)
  12. prompt = "KES数据迁移工具KDTS支持哪些源数据库的迁移?"
  13. inputs = tokenizer.apply_chat_template([{"role": "system", "content": "假设你是人大金仓的小助手。"},{"role": "user", "content": prompt}],
  14.                                        add_generation_prompt=True,
  15.                                        tokenize=True,
  16.                                        return_tensors="pt",
  17.                                        return_dict=True
  18.                                        ).to('cuda')
  19. gen_kwargs = {"max_length": 2500, "do_sample": True, "top_k": 1}
  20. with torch.no_grad():
  21.     outputs = model.generate(**inputs, **gen_kwargs)
  22.     outputs = outputs[:, inputs['input_ids'].shape[1]:]
  23.     print(tokenizer.decode(outputs[0], skip_special_tokens=True))
复制代码
目前,我们的Lora权重尚未与基础模型合并,而只是被加载到模型中。在此,我们将进行一个简单的测试,目的是查看加载后的效果和验证Lora权重的作用,输出如下所示:


  • 支持Oracle、MySQL、SQLServer、Gbase、PostgreSQL、DM、KingbaseES等数据库作为源端进行数据迁移。
  • 这些数据库版本均被纳入支持范围,确保迁移过程中的兼容性和稳定性。
如果你希望将可视化的WEB页面暴露出来进行访问,首先需要根据部署文件的要求操作。具体步骤是:创建一个名为 chatbot.py 的新文件,并在该文件中编写以下内容:
[code]from transformers import AutoTokenizer, AutoModelForCausalLMimport torchimport streamlit as stfrom peft import PeftModel# 在侧边栏中创建一个标题和一个链接with st.sidebar:    st.markdown("## Qwen2.5 LLM")    "[开源大模型食用指南 self-llm](https://github.com/datawhalechina/self-llm.git)"    # 创建一个滑块,用于选择最大长度,范围在 0 到 8192 之间,默认值为 512(Qwen2.5 支持 128K 上下文,并能生成最多 8K tokens)    max_length = st.slider("max_length", 0, 8192, 512, step=1)# 创建一个标题和一个副标题st.title("
来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!

相关推荐

您需要登录后才可以回帖 登录 | 立即注册