我开发了一个面向大模型训练的数据集构建平台:DatasetLoom

在做 LLM 微调的过程中,你是否也遇到过这些问题?

  • 想做 SFT,但没有高质量的 instruction 数据;
  • 想做 DPO,却找不到可靠的"偏好对";
  • 人工标注成本高、周期长、一致性差;
  • 自动生成的数据看似流畅,实则"一本正经地胡说八道"。

我们越来越清楚地意识到:模型的能力上限,取决于训练数据的质量下限

但现实是,大多数团队还在用"人工+Prompt+Excel"的方式生产数据------效率低、难追溯、难协作。

于是,我做了 DatasetLoom ------ 一个面向 大模型训练 的智能数据集构建平台。


DatasetLoom:让训练数据生产变得专业、可控、可追溯

DatasetLoom 的目标不是"全自动生成数据",而是提供一个端到端、可验证、支持团队协作的数据构建闭环。

整个流程如下:

  1. 上传文档 → 2. 智能分块 → 3. 自定义 Prompt 生成问题/回答 → 4. AI 多维度评分 → 5. 人工审核 + 溯源验证 → 6. 导出为 SFT/DPO 数据集

核心功能

文档智能分块

支持上传 PDF、Word、Markdown、TXT 等文本文件,系统会自动按段落、标题或语义进行切分,避免上下文断裂,确保每一块内容都具备独立语义。

你可以根据文档类型配置不同的分块策略以及数据清洗规则


自定义 Prompt 生成内容

每个数据生成环节都设计了专属的 Prompt 模板,支持完全自定义,确保生成内容符合你的任务需求。

同时支持多个大模型生成结果,便于后续对比评估。

数据集管理:灵活切换,按需使用

在完成问题生成与 AI 评分后,所有数据会统一归集到 QA 数据集管理界面,支持三种展示模式,满足不同微调任务的需求:

  1. 全部数据
    展示每一个问题及其所有生成的回答(来自不同模型或不同 Prompt 的结果),适合用于数据审查、模型对比和历史追溯。

2. 用于 SFT(监督微调)

仅展示每个问题的"主答案"(可手动或自动选定最优回答),形成标准的 instruction → response 格式,可直接导出为 SFT 训练语料。

3. 用于 DPO(偏好对齐)

展示已标注偏好的问答对,每条记录包含同一个问题下的 chosen(优选回答)rejected(劣选回答),支持人工复核与 AI 辅助标注,确保偏好数据高质量、可解释。

可以基于同一份原始文档,高效产出多种类型的训练数据,真正实现"一套数据,多任务复用"。


AI 评分机制 + 溯源验证

让每一条生成的数据都可评估、可追溯。系统内置多维度 AI 评分体系,由大模型自动评估输出质量:

  • 事实准确性:是否与原文一致
  • 逻辑完备性:推理是否合理
  • 表达质量:语言是否流畅
  • 安全合规:是否包含敏感信息
  • 综合得分

生成的回答都会标注其来源段落,点击即可查看原始上下文,真正做到"有据可查"。这一机制极大提升了数据审核效率,尤其适合团队协作场景。


最终产出:训练语料导出

所有经过生成、评分、审核的数据,都可以一键导出为:

  • ✅ JSON / CSV(本地保存)
  • ✅ HuggingFace Dataset 格式(可直接上传至 HuggingFace Hub)
  • ✅ 支持Llama Factory微调框架

真正实现从"原始文档"到"可用语料/数据集"的无缝闭环。

技术架构:

层级 技术
前端 Next.js + React 18 + Tailwind CSS
后端 NestJS + TypeScript + RESTful API
ORM Prisma(支持 SQLite/MySQL/PostgreSQL/SQL Server)
向量数据库 Qdrant(用于 RAG 检索)
构建系统 Turborepo + pnpm

使用场景

DatasetLoom 适用于以下典型场景:

  • 构建 SFT 指令微调数据集
    从文档中生成 instruction-input-output 三元组
  • 生成 DPO 偏好对(chosen / rejected)
    多模型输出对比 + AI 评分,自动筛选偏好样本
  • 垂直领域知识库构建
    医疗、法律、金融等专业文档的结构化处理
  • 多模型输出质量评估
    对比 GPT-4、Qwen、LLaMA 等模型在同一任务上的表现

快速启动

bash 复制代码
git clone https://github.com/599yongyang/DatasetLoom.git
cd DatasetLoom
pnpm install
pnpm run dev

也支持 Docker 一键部署,生产环境开箱即用:

bash 复制代码
docker compose up -d --build

欢迎试用

如果你也在为高质量训练数据发愁,DatasetLoom 或许能帮上忙。

  • ⭐ 如果你觉得这个项目有价值,请给它一颗 Star
  • 📬 欢迎提交 Issue 或 PR,一起让它变得更强大

GitHub: github.com/599yongyang...


相关推荐
GitCode官方5 小时前
openPangu-2.0-Pro 模型及技术报告正式开源上线 AtomGit AI
人工智能·华为·开源·harmonyos·atomgit
Kevinten106 小时前
OpenOctopus:领域原生 AI 家庭助手系统,Summon 功能将实体转化为智能 Agent
人工智能·ai·开源·开发工具·智能体
看昭奚恤哭7 小时前
基于 RuoYi-Vue-Pro 定制了一个后台管理系统 magic-admin , 开源出来!
前端·vue.js·开源
To_OC16 小时前
从 0 到 1:Milvus + 大模型打造私人记忆知识库
人工智能·node.js·llm
孙启超16 小时前
【AI应用开发】 RAG篇(四):Prompt 工程与进阶技术
人工智能·llm·embedding·rag·向量化·chunking·文档切分
冬奇Lab18 小时前
每日一个开源项目(第171篇):Harness Handbook - 给 AI Agent 的 Harness 代码库生成一本可导航的行为手册
人工智能·开源·资讯
周末程序猿18 小时前
图解 120 个大语言模型(LLM)核心概念(91-120)
llm·agent
仙宇觉尘20 小时前
一款基于 .NET 开源美观、功能丰富的串口调试工具
开源·.net
iMountTai21 小时前
DeepSeek-V4-Flash 接入 Codex(Windows 版)
开源
众人皆醒我独醉21 小时前
TGI:HuggingFace 的官方推理服务——不止 PagedAttention,更懂模型生态
面试·llm·ai编程