Skip to main content

平台定位与职责​

模型开发训练平台是面向AI大模型全生命周期的开发、训练、微调的统一管理平台。平台提供从数据准备到模型产出的端到端能力,与下游推理服务平台形成完整的MLOps闭环。

核心职责:

  • 提供AI模型开发、训练、微调的全流程支撑能力
  • 管理和调度异构算力资源(GPU/NPU等)
  • 实现数据、代码、模型等资产的统一管理
  • 支持多租户资源隔离与配额管理
  • 输出标准化模型产物供推理服务使用

核心功能特性​

数据管理​

数据集管理​

功能项功能描述
数据集创建支持从本地上传、对象存储导入、数据库导入等多种方式创建数据集
数据集版本管理对数据集进行版本控制,支持版本回溯和对比
数据集预览支持文本、图片、音频、视频等多模态数据的在线预览
数据集分割支持按比例自动分割训练集、验证集、测试集
数据集共享支持跨项目、跨租户的数据集共享与权限控制

数据标注​

功能项功能描述
标注任务管理创建、分配、跟踪标注任务,支持多人协作标注
多模态标注工具提供文本分类、NER、图像分类、目标检测、语义分割等标注工具
智能预标注利用已有模型进行预标注,提升标注效率
标注质检支持抽样审核、交叉验证等质检机制,确保标注质量
标注数据导出支持导出为COCO、VOC、JSON等主流格式

数据处理​

功能项功能描述
数据清洗去重、去噪、格式标准化等数据清洗能力
数据增强支持图像旋转、裁剪、文本回译等数据增强策略
特征工程提供特征提取、特征选择、特征转换等能力
数据血缘追踪数据从原始到处理后的完整链路

在线开发(Notebook)​

开发环境管理​

功能项功能描述
Notebook实例创建支持创建JupyterLab、VS Code Server等在线开发环境
镜像管理提供预置镜像(含主流深度学习框架),支持自定义镜像
资源规格选择支持选择CPU/GPU/NPU等不同算力规格
环境持久化开发环境数据持久化存储,支持停止后恢复
SSH远程连接支持通过SSH连接到Notebook实例进行开发

开发辅助能力​

功能项功能描述
代码版本管理集成Git,支持代码提交、分支管理、代码对比
数据集挂载将数据集直接挂载到开发环境,方便数据访问
模型调试支持在Notebook中进行模型调试和小规模验证
依赖管理支持pip/conda等包管理工具,管理Python依赖
终端访问提供Web Terminal,支持命令行操作

模型训练​

训练任务管理​

功能项功能描述
单机训练支持单机单卡/多卡训练任务提交
分布式训练支持多机多卡分布式训练,支持PyTorch DDP、Horovod、DeepSpeed等框架
训练任务模板提供常用训练任务模板,降低使用门槛
任务生命周期管理支持任务创建、启动、暂停、恢复、停止、删除等操作
任务队列管理基于Volcano实现任务队列管理,支持优先级调度

训练配置​

功能项功能描述
超参数配置支持通过界面或配置文件设置训练超参数
资源配置配置CPU、内存、GPU/NPU等计算资源
环境变量配置支持设置训练任务所需的环境变量
挂载配置配置数据集、代码、模型输出等存储挂载
网络配置配置分布式训练的网络模式(如RDMA/InfiniBand)

训练监控​

功能项功能描述
训练日志实时查看训练日志输出
训练指标可视化集成TensorBoard,可视化loss、accuracy等训练指标
资源监控监控GPU利用率、显存使用、网络带宽等资源指标
训练进度展示训练进度、预估剩余时间
告警通知训练异常、完成等事件的告警通知

训练优化​

功能项功能描述
自动超参调优支持Grid Search、Random Search、Bayesian Optimization等调参策略
早停机制支持基于验证集指标的早停策略
断点续训支持从Checkpoint恢复训练,避免资源浪费
弹性训练支持训练过程中动态调整Worker数量

模型微调(Fine-tuning)​

预训练模型管理​

功能项功能描述
模型仓库提供主流开源预训练模型(LLaMA、Qwen、ChatGLM等)
模型导入支持从HuggingFace、ModelScope等平台导入模型
模型版本管理对预训练模型进行版本管理
模型元数据记录模型架构、参数量、适用场景等元信息

微调能力​

功能项功能描述
全参数微调支持对模型全部参数进行微调
参数高效微调支持LoRA、QLoRA、P-Tuning、Prefix-Tuning等PEFT方法
指令微调支持基于指令数据集的SFT(Supervised Fine-Tuning)
RLHF支持基于人类反馈的强化学习微调
多任务微调支持同时在多个任务上进行微调

微调配置​

功能项功能描述
微调数据集配置配置用于微调的数据集,支持多种数据格式
LoRA参数配置配置LoRA的rank、alpha、target modules等参数
量化配置配置INT8/INT4量化参数,降低显存占用
学习率策略配置学习率调度策略(Cosine、Linear等)

模型管理​

模型注册​

功能项功能描述
模型注册将训练/微调产出的模型注册到模型仓库
模型版本管理对模型进行版本控制,支持版本对比和回滚
模型元数据管理记录模型的训练参数、数据集、指标等元信息
模型标签管理支持为模型添加标签,便于检索和分类

模型评估​

功能项功能描述
离线评估在测试集上进行模型离线评估
评估指标支持Accuracy、F1、BLEU、ROUGE等多种评估指标
模型对比支持多个模型版本的指标对比
评估报告生成详细的模型评估报告

模型转换与优化​

功能项功能描述
格式转换支持PyTorch、TensorFlow、ONNX等格式互转
模型量化支持INT8/INT4/FP16等量化,减小模型体积
模型剪枝支持结构化/非结构化剪枝,压缩模型
推理优化支持TensorRT、vLLM等推理优化

模型导出​

功能项功能描述
模型文件导出导出模型权重文件(.pt、.safetensors等)
推理镜像构建基于模型自动构建推理服务镜像
模型卡片生成生成标准化的Model Card,描述模型信息

镜像管理​

镜像仓库​

功能项功能描述
公共镜像提供预置的训练/推理基础镜像
私有镜像支持用户上传和管理私有镜像
镜像版本管理对镜像进行版本管理和标签管理
镜像扫描对镜像进行安全漏洞扫描

镜像构建​

功能项功能描述
Dockerfile构建支持基于Dockerfile构建镜像
在线构建提供Web界面进行镜像在线构建
构建缓存支持构建缓存,加速镜像构建
自动构建支持代码提交触发自动镜像构建

资源管理​

算力资源管理​

功能项功能描述
资源池管理管理GPU/NPU等异构算力资源池
资源配额基于租户/项目设置资源配额
资源监控实时监控资源使用情况和利用率
资源调度基于Volcano实现高级调度策略(Gang调度、优先级抢占等)

存储资源管理​

功能项功能描述
存储卷管理管理PV/PVC,支持多种存储后端
共享存储提供NFS/CephFS等共享存储,支持多任务共享数据
对象存储集成集成S3/MinIO等对象存储,存储大规模数据集
存储配额设置租户/项目的存储配额

实验管理​

实验跟踪​

功能项功能描述
实验记录自动记录每次训练的参数、指标、产出
实验对比支持多个实验的参数和指标对比
实验复现基于实验记录一键复现训练
实验分组支持对实验进行分组管理

流水线(Pipeline)​

功能项功能描述
流水线编排支持可视化编排数据处理→训练→评估→部署流水线
流水线模板提供常用流水线模板
定时触发支持定时触发流水线执行
流水线监控监控流水线执行状态和各阶段耗时

多租户与权限​

功能项功能描述
租户管理支持多租户隔离,每个租户独立的资源空间
项目管理租户下支持多项目管理
用户管理用户账号管理,支持LDAP/OAuth集成
角色权限基于RBAC的细粒度权限控制
操作审计记录用户操作日志,支持审计追溯

平台工作流与物料流转​

整体工作流​

输入物料​

物料类型物料描述来源存储方式
原始数据未经处理的原始数据(文本、图片、音视频等)业务系统、公开数据集、爬虫采集对象存储(S3/MinIO)
标注数据经过人工或自动标注的数据集数据标注模块产出对象存储 + 元数据库
训练数据集格式化后用于训练的数据集数据处理模块产出共享存储(NFS/CephFS)
训练代码模型训练脚本和配置文件开发人员编写、Git仓库Git仓库 + 共享存储
预训练模型开源或自研的预训练模型权重HuggingFace、ModelScope、自研模型仓库(共享存储)
基础镜像包含深度学习框架的容器镜像镜像仓库Harbor/Docker Registry
配置文件训练超参数、环境配置等开发人员编写Git仓库 + ConfigMap

输出物料​

物料类型物料描述产出环节存储方式下游使用
模型权重文件训练/微调后的模型参数文件(.pt、.safetensors、.bin等)模型训练/微调模型仓库(共享存储)推理服务加载
LoRA权重参数高效微调产出的增量权重模型微调模型仓库与基座模型合并后推理
模型配置模型架构配置文件(config.json等)模型训练/微调模型仓库推理服务加载
Tokenizer文件分词器配置和词表文件模型训练/微调模型仓库推理服务加载
训练镜像包含训练代码和依赖的容器镜像镜像构建镜像仓库训练任务使用
推理镜像包含模型和推理框架的容器镜像镜像构建镜像仓库推理服务部署
训练日志训练过程的日志和指标数据模型训练日志系统问题排查、实验分析
评估报告模型评估指标和分析报告模型评估元数据库模型选型决策
Model Card模型说明文档模型导出模型仓库模型使用说明

物料流转详细流程​

数据流转​

详细说明:

  1. 原始数据导入:从业务系统或公开数据源导入原始数据到对象存储
  2. 数据清洗:去重、去噪、格式标准化,产出清洗后数据
  3. 数据标注:人工或智能预标注,产出标注数据
  4. 数据增强:根据需要进行数据增强,扩充数据集
  5. 数据集发布:将处理后的数据发布为可用数据集,记录版本

代码流转​

详细说明:

  1. 本地开发:开发人员在本地编写训练代码
  2. Git提交:代码提交到Git仓库进行版本管理
  3. Notebook调试:在Notebook环境中进行代码调试和小规模验证
  4. 训练脚本:调试完成后,整理为可提交的训练脚本
  5. 训练任务:将代码打包或挂载到训练任务中执行

模型流转​

详细说明:

  1. 预训练模型:从模型仓库获取预训练模型
  2. 模型微调:使用业务数据进行微调,产出微调权重
  3. 模型评估:在测试集上评估模型效果,产出评估报告
  4. 模型注册:将合格模型注册到模型仓库,记录版本
  5. 模型导出:导出模型文件或构建推理镜像

镜像流转​

详细说明:

  1. 基础镜像:选择包含深度学习框架的基础镜像
  2. 依赖安装:安装训练/推理所需的Python包和系统依赖
  3. 代码打包:将训练代码或推理代码打包到镜像
  4. 镜像构建:构建Docker镜像
  5. 镜像推送:推送到镜像仓库供后续使用

与推理服务平台的对接​

对接方式​

对接方式描述适用场景
模型文件对接将模型权重文件存储到共享存储,推理服务直接加载模型更新频繁、需要热更新
推理镜像对接将模型打包到镜像中,推理服务部署镜像模型稳定、需要版本化部署
模型仓库对接通过统一的模型仓库API进行模型交付需要模型元数据和版本管理

对接流程​

交付物清单​

交付物格式用途
模型权重.pt / .safetensors / .bin推理服务加载模型参数
模型配置config.json推理服务构建模型架构
Tokenizertokenizer.json / vocab.txt推理服务进行文本预处理
推理镜像Docker Image推理服务容器化部署
Model CardREADME.md模型使用说明和限制
推理配置inference_config.yaml推理服务参数配置