做科研平台落地这几年,最明显的一个感受是:数据越来越像一个“后勤部门”,而模型推理越来越像一个“生产车间”。但很多团队恰恰把这两块拆得太开——管数据的人不关心推理,做推理的人不碰数据,最后项目推进一半就卡在“数据传不过来”“结果复现不了”这种最基础的问题上。
今天想聊的这套一站式科研分析平台,就是围绕“从数据到推理”这条完整链路搭的一整套方案。它不是一个具体软件,而是一套把数据采集、治理、标注、训练、推理、结果回流全部串起来的工程方法论。适合正在搭建内部分析平台、或者手上同时有好几个科研项目要跑的人参考。我把这几年踩过的坑、验证过好用的配置、以及每次排查问题时的思路都整理进了下文,尽量做到你照着做也能复现。
1. 整体设计思路:为什么非得“从数据到推理”一条链路打通
1.1 科研场景里最常见的断点
我最早接手这类平台时,团队里同时跑着四五个项目:有的是做目标检测的,有的是做大模型问答的,还有的是拿历史数据做统计分析。表面上看各干各的,真正跑起来才发现,所有项目都卡在同一批问题上。
第一个断点叫“数据二次加工”。同一个数据集,A项目洗成了CSV,B项目转成了JSON,C项目又嫌格式不够要重新写脚本。三个人三套代码,谁也不敢动别人的数据集,因为一改就可能影响别人的实验。第二个断点是“推理黑盒”。模型训练完之后通常丢给一个Python脚本,输入一张图或一段文本,输出一个结果就完了。但用到实际业务里,用户要的不是“跑通”,而是结果稳定、可追溯、能批量处理。第三个断点是“反馈不回传”。推理结果里发现的Bad Case,没有渠道回流到数据集侧,标注人员不知道,数据治理的人也不知道,下一轮训练依然带着同样的错误。
所以我在设计这个平台时,第一条原则就是:不要把数据链路和推理链路当成两个系统做,而是要当成一条流水线。每一段输出都要有明确的格式、版本和落库位置,下游可以无缝消费上游的产出。
1.2 平台模块划分与主流程
整个平台我按功能拆成了五个层次,每一层只跟相邻层打交道,避免上下游耦合过深。
- 数据接入层:负责从爬虫、公开数据集、数据库、API、传感器设备等渠道拿原始数据。
- 数据治理与存储层:负责清洗、去重、格式统一、备份恢复,并把数据落到合适的存储里。
- 数据标注与数据集建设层:负责把“可用数据”变成“模型可训练的数据”,包括标注任务管理、格式转换、数据增强。
- 训练与推理层:负责模型训练、推理服务部署、性能调优、结果保存。
- 评估与回流层:负责推理结果的分析、可视化,以及把Bad Case反馈回数据侧。
这个分层最关键的一点是“每一层输入输出都必须有明确标准”。比如数据接入层出来的,统一是带schema的原始数据文件或者数据库表;数据治理层出来的,一定是经过校验、符合质量规则的数据版本;推理层出来的,是结构化的推理结果,包含模型版本、推理时间、置信度、原始输入ID等字段。
标准定好之后,不仅人跟人之间协作顺畅,连后续接自动化流水线都省了很多事。我现在跑新项目的时候,只要数据格式满足平台要求,基本几个小时内就能把一条完整的“采集—清洗—训练—推理—回流”跑通。
1.3 技术选型的原则:在“够用”和“先进”之间找平衡
做这类平台最容易犯的毛病是过度设计。早期我也迷恋过K8s、微服务、湖仓一体这些概念,结果发现团队维护成本极高,光搭环境就花了两周。后来我换了一种思路:按项目实际规模,选最顺手且留有扩展余地的方案。
比如推理框架,我是从ONNX Runtime起步的,后面跑熟了再上TensorRT。数据存储也是,结构化数据就用MySQL,文件类数据用对象存储,向量检索等有需求了再引入专门的向量库。消息队列这类组件,一开始甚至只用Redis的Stream就够了。
这套“够用就好”的原则不是说不要先进技术,而是让平台先跑起来,再根据瓶颈逐步替换。数据量大了换并行计算框架,推理时延要求高了换推理引擎,前端可视化需求复杂了再上BI工具。每一步都有明确收益预期,而不是为了技术而技术。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 数据链路:没有可靠的数据,推理就是空中楼阁
2.1 多源数据接入:爬虫、公开数据集、数据库一网打尽
做科研项目,数据来源普遍很杂。我这边接触过的就有公开基准数据集、网页爬虫抓取的数据、业务数据库导出的表格、问卷系统生成的记录,还有硬件设备通过Socket上报的16进制数据流。
以爬虫为例,很多做地理分析的人需要批量下载百度POI数据。我自己试过两条路:一条是直接用八爪鱼这类可视化采集工具,配置好翻页规则和字段映射就能跑,适合不写代码的人;另一条是用Python写爬虫脚本,用requests带UA和Cookie池去请求接口,再解析返回的JSON。两条路各有优劣,八爪鱼上手快但字段抽取灵活度低,自写脚本灵活但需要处理反爬、断线重试、增量更新这些问题。我的建议是:一次性采集用可视化工具,长期维护的数据采集任务必须自建脚本,把采集结果规范化成统一格式落到数据库里。
硬件设备的数据接入也经常遇到。比如有些传感器设备会用Socket协议以16进制报文格式推送数据,你这边要写一个接收服务,做帧解析、校验、转码,再把有效字段拆出来入库存档。这类数据通常还带着时间戳和点位编号,数据字典必须提前定义好,不然存进去之后根本没法查。
数据接入层最容易被忽略的是“血缘管理”。每一份数据到底是从哪个来源爬的、哪个系统导出的、采集时间是什么时候,这些元信息如果不记录,后面数据出问题只能两眼一抹黑。我现在会在每个表或者每个数据文件边上维护一个元数据清单,至少包含数据来源、采集时间、采集脚本版本、字段含义、更新频率这几个字段。
2.2 数据治理:从“有数据”到“有可用数据”需要过五关
数据不治理直接喂给模型,基本等于自杀。我见过太多项目因为数据集里有大量重复样本和错误标注,导致模型训练出来精度虚高或者根本收敛不了。
治理流程我一般分四步走。第一步是去重,尤其是爬虫数据,同一个POI可能被重复抓取几十次,做数据清洗时要按业务主键做唯一性约束。第二步是缺失值处理,数值型字段要不要填默认值、填0还是填均值,字符串字段要不要保留为空,都要根据下游用途来定。第三步是异常值过滤,比如某些数值突然冒出几百倍于正常范围的顶点,这种多半是采集设备故障,需要用分位数或标准差方法识别并剔除。第四步是格式统一,时间字段全部转成统一时区,坐标字段统一成WGS84标准,文本编码统一转成UTF-8。
另外还有一个容易被忽视的点:如何把关系数据库里的结构化数据加工成大模型能读懂的数据。这个问题在RAG和知识库类项目里特别常见。我的做法是,先把数据库表导出成文本描述,按“表名—字段含义—记录内容”的模板组织成上下文块,再根据问题类型做切分和索引。比如“用户ID为123的订单记录”可以加工成“user_123在2024年5月下的订单包括:A商品2件、B商品1件,总金额XXX元”这种自然语言段落。加工完之后一定要保留一个“原始结构化数据→大模型输入文本”的映射关系文件,方便出问题时回溯。
数据治理的终极目标,是让下游拿到数据时不需要再问“这个字段是什么意思”“这个空值代表什么”。这些信息应该在数据字典里写清楚,而不是靠口头沟通。
2.3 数据备份与恢复:不要等磁盘满了才开始想备份方案
数据备份与恢复这个话题,每次我提起来都觉得自己是在“爹味发言”,但真出问题的时候,没有备份的方案基本就是灾难现场。
做过一次惨痛教训:有一个跑了几周的爬虫任务,数据都存放在一台工作机的本地磁盘上,磁盘故障后所有原始数据直接蒸发。后来重建整个数据集花了一个多月,比当初采集的时间还长。
从那以后,我给自己定了几条备份铁律。第一,所有原始采集数据必须入对象存储或独立备份盘,不允许只存在个人电脑或工作机的本地磁盘里。第二,遵循“3-2-1”原则,至少3份副本、2种不同存储介质、至少1份异地或者异机存储。第三,每天做增量备份,每周做一次全量快照,备份任务要有日志监控,备份失败要能报警。第四,至少每个季度做一次恢复演练,不演练你就永远不知道备份文件能不能真正恢复。
具体到技术工具,可以结合自身环境来选。Linux服务器上我用过restic、restic的自动任务,或者直接用系统自带的rsync做增量同步,再配合定期快照。个人Mac上如果发现系统数据占用过大,也要学会清理和备份——比如把“其他”类型的大文件找出来,把Xcode派生数据、Docker镜像、旧备份清掉,这些操作对维护科研环境的整洁度非常重要。
2.4 数据存储与数据字典:关系和文件要分得清
存储选型我目前用的是比较朴素的组合:结构化数据放MySQL,非结构化原始文件放对象存储或文件系统,中间结果用Parquet或JSON归档。如果需要给大模型做向量检索,再单独引入向量数据库,但不会为了“显得高级”一上来就全上。
MySQL虽然“老派”,但在科研平台里它依然是最稳定的元数据和结构化数据底座。数据字典、任务状态、标注记录、推理结果索引,这些用MySQL存非常合适。我日常管理库表时经常用到的命令其实就那么几条:show tables、desc table_name、show create table、select ... where ... limit,以及用information_schema查表结构。别小看这些简单命令,很多复杂的元数据管理问题靠它们就能解决大半。
数据结构设计这条我也想多说一句。很多搞算法的人写代码时不太在意表结构,一张表里塞了几十个字段,字段名随便起。等到平台要接入多项目时,这种表就成了灾难。我现在的做法是,在设计数据表之前先出数据字典,明确每个字段的名称、类型、是否可空、取值约束,并且字段命名统一用小写下划线风格,避免混用驼峰和大小写。这样不管是人查数还是写代码,都能少踩很多坑。
3. 数据集建设与标注:模型上限的一半在这里决定
3.1 标注流程与工具链:规范比工具更重要
很多人觉得数据标注就是拿个工具在图片上画框,或者给文本打个标签,其实没那么简单。一个靠谱的标注流程至少要包含四个环节:标注规范制定、标注工具配置、标注执行、质量抽检。
标注规范这一步最容易被跳过。我见过一个项目,负责人扔给标注人员一批图片和一套开源标注工具,说“你们开始标吧”。结果一周后收回来的数据,有的框只框了目标的一半,有的把遮挡目标标成了不同类别,有的漏标了边缘样本。追问下来才发现,大家连“框要贴近目标边缘”这种基本标准都没对齐。所以我现在做任何标注任务之前,一定会先写一份标注文档,至少包含各类别判断标准、边界情况处理方式、框选精度要求、常见错误示例,并且拿一小批数据做试标和评审,评审通过后才开始批量标注。
工具选型上,轻量任务我用LabelImg或者Label Studio,团队协作类任务推荐用CVAT,因为支持多人分配任务和review流程。至于Label Studio,它更适合做文本和多模态类标注,灵活性比CVAT高不少。
数据标注另一个重要环节是数据安全管理。科研数据很多有保密要求,标注平台要设置权限,标注人员只能看到分配给自己的批次,不能看全量数据。这个在平台设计时就要考虑进去,不能等出事了再补救。
3.2 数据集结构与格式:COCO还是YOLO?别混着用
做过视觉项目的人对COCO2017数据集肯定不陌生,但真正吃透它目录结构的人其实不多。COCO2017的格式核心是annotations目录下的JSON文件,里面包含info、licenses、images、annotations、categories五个部分。images记录每张图片的id、文件名、宽高;annotations记录每个标注框的id、所属图片id、类别id、bbox坐标和分割多边形;categories记录类别id和类别名。
你在训练自己的数据集时不一定非要原封不动地按照COCO格式来,但一定要理解这种分离式结构的好处:图片文件和标注文件分开,训练集、验证集、测试集只需要在JSON里通过定义不同的文件列表来区分,不需要实际复制图片。这个设计在数据集版本迭代时特别省事。
如果你用YOLOv8训练自己的数据集,标准的做法是整理成images和labels两个目录,每个样本对应一个txt标注文件,格式是“class_id x_center y_center width height”,坐标值归一化到0-1之间。训练前按比例划分train和val,并把路径写到data.yaml里。YOLO格式直观好用,但要注意它做了坐标归一化,转回绝对坐标时要按图片宽高反向计算,很多人在这里吃过亏。
格式转换是数据准备中最繁琐的一步。我的建议是写一个统一的转换脚本,把不同来源的数据统一转成目标格式,并且在脚本里输出转换日志和校验结果,确保没有样本漏转换、没有坐标越界。转换完再做一次可视化抽样检查,把标注框画在图上人工过一眼,这一步能发现大量坐标问题。
3.3 数据增强与样本平衡:别让模型“背过”训练集
数据增强是提升模型泛化能力的有效手段,但也容易用过头。传统的图像增强方法有随机翻转、随机裁剪、旋转、缩放、颜色抖动、亮度调整、加噪声等,这些操作能在不改变标注语义的前提下扩充样本量。更现代一点的比如Mosaic和MixUp,能在早期训练阶段显著提升模型的鲁棒性。
但我提醒几点。第一,增强操作不能破坏数据本身的物理约束。比如做X光安检物品检测,你把图像旋转了90度,原本的“行李箱”朝向变了,形状还合理,但有些类别比如“液体瓶”上下颠倒是没问题的,不过如果你做的是车牌识别,垂直翻转直接会让文本不可读。第二,增强会改变图片的底层信息分布,如果训练时用增强,推理和测试时也必须统一使用一致的预处理流程,否则性能会打折扣。第三,同一个样本的多个增强版本不要同时出现在训练集和验证集中,不然会产生数据泄漏,让验证结果虚高。
对于类别不均衡问题,除了简单的过采样和欠采样,还可以通过Copy-Paste或者合成样本的方式补充稀有小类。我自己的经验是:优先把真实采集的难例和边界样本纳入训练集,其次才考虑用增强扩充。增强是锦上添花,不能代替真实数据的多样性。
4. 推理链路:从模型文件到可用的业务结果
4.1 推理框架学习路线:别一上来就啃TensorRT
推理框架这块,网上资料很多但也很杂,新手很容易迷失。我整理一条自己验证过比较顺畅的学习路线,按这条路线走,大概两到三周能上手。
第一步是模型导出。训练好的模型要先转成中间表示格式,最常见的是ONNX。PyTorch模型用torch.onnx.export导出,注意要设置input_names、output_names、dynamic_axes以及opset_version,否则后续转换会遇到一堆兼容性问题。
第二步是ONNX Runtime。这个框架部署简单,CPU和GPU都能跑,适合作为第一个上线的推理后端。它的API很直观,加载ONNX模型、创建session、构造输入、跑推理、取输出。用ONNX Runtime跑通一次完整流程,你就理解了推理程序的基本结构。
第三步是TensorRT。在英伟达GPU上,TensorRT通过层融合、精度校准、内核自动调优,能把推理性能提升不少。但它的学习曲线陡峭,坑也多,比如不同GPU架构需要重新构建engine,输入输出张量维度不匹配会直接跑不起来。我的建议是:先用TensorRT将ONNX模型转成engine文件,再通过trtexec工具验证性能,确认没问题之后再接入自己的服务代码。
第四步是工程化部署,把推理逻辑封装成HTTP服务或者gRPC服务,加上请求日志、结果持久化、异常处理。到了这一步,推理就从“脚本”变成了“服务”。
4.2 推理性能调优的几个关键参数:GPU占用、批大小、内存拷贝
推理性能调优是平台上线后最常做的工作。很多人一看到GPU利用率低就着急调模型,其实瓶颈往往不在模型本身。
最典型的例子是“服务器内存和推理卡之间的影响”。我排查过不少推理性能问题,最后发现很多是CPU端预处理和内存拷贝瓶颈导致的。GPU要处理的数据必须先从内存搬到显存,如果你的数据加载、前处理(比如图像解码、缩放、归一化)都在CPU上串行执行,GPU只能干等。解决方法是:用数据加载器或者流水线机制,让CPU在GPU算当前batch的时候,提前准备下一个batch;同时开启pin memory,减少Host到Device的拷贝耗时。
Batch Size也是关键参数。检测类模型推理时,Batch Size从1调到4,GPU利用率可能从30%升到70%,但调到16之后性能提升就放缓了,显存占用反而暴增。所以Batch Size要按“显存余量—吞吐量—时延要求”三者权衡。我在跑自动驾驶感知模型(比如Apollo中的CenterPoint)推理时,会先用nvidia-smi监控当前显存占用,再反向推算能设多大Batch。同样的道理,做YOLOv11推理时如果想保存结果,也建议一次性处理一批图片再统一保存,比单张循环处理要快得多。
精度选择也不能忽略。FP32、FP16、INT8的推理速度差距可能有好几倍,但精度也会逐渐下降。我一般默认用FP16,只有在精度明显下降时才退回FP32。INT8需要标定,精度损失更难控制,除非是极端的时延敏感场景,否则不推荐一上来就上。
如果你是用Ollama部署大模型,感觉推理生成速度慢,通常要从这几方面排查:模型量化版本(Q4_K_M通常比FP16快很多)、上下文长度设置(上下文越长,占用显存越多,也会拖慢生成速度)、并发请求数(默认并发低时单请求等待时间会增加)、是否开启GPU加速(Ollama默认会用GPU,但驱动没装好时会退回CPU,这个一定要检查)。
4.3 推理任务的编排、结果保存与多轮对话
一个完整的推理链路,不只是把模型跑一遍,还要考虑任务怎么编排、结果怎么保存、异常怎么重试。
推理任务可以简单分成批处理任务和在线服务。批处理适合离线刷全量数据,比如给历史数据集统一打推理标签;在线服务适合交互式场景,比如用户上传一张图片立刻得到检测结果。我这边是把两种模式都做了,底层共用同一个推理模块,只是入口和任务管理不同。
结果保存我强烈建议做成结构化格式。以目标检测为例,每一条推理结果至少包含:结果唯一ID、输入数据ID、模型版本、推理时间、检测框坐标、置信度、类别ID。这样后续做评估分析时,可以按模型版本做回归对比,也可以按类别统计Bad Case。保存到数据库后再提供一个查询接口,前端展示就非常方便。
YOLOv11这类模型如果要保存推理结果,除了保存标注后的图片,最好也把检测框和置信度存成JSON或者数据库记录。图片带标注图可以给人看,结构化记录是给程序用的,两者都留,后续做统计才顺手。
如果你要做一个支持多轮对话的推理应用,比如用Dify搭建Chatflow,需要确认它是否支持多轮对话推理。答案是肯定的,但要注意对话历史的管理。每轮对话要把用户输入、模型输出、检索到的上下文拼装好塞给大模型,超出上下文窗口时要对历史消息做截断或摘要。我之前遇到过一个现象,对话过几轮之后模型开始“失忆”,回答得前言不搭后语,排查下来就是对话历史没有做降级处理,直接把所有历史一股脑塞满上下文导致的。后来改成前几轮保留完整、更早的轮次做摘要压缩,问题就消失了。
4.4 大规模推理任务调度:幂等、重试、资源隔离
当推理任务量变大之后,调度就成了新问题。比如要给一个10万条记录的数据库批量生成推理结果,单线程跑可能要跑一整天,这时候就需要任务拆分和并行。
我的做法是维护一个任务表,每条任务记录包含输入数据范围、任务状态(PENDING、RUNNING、SUCCESS、FAILED)、执行节点、重试次数、开始结束时间。多个Worker进程或机器从任务表里拉取任务,处理完更新状态。这里有几个细节很重要。第一,任务必须幂等,同一个任务被两个Worker同时拉到,也不能产生重复结果,处理方式是在执行前给任务加分布式锁或唯一索引。第二,失败要自动重试,但要限制最大重试次数,避免死循环。第三,不同项目或者不同优先级的任务要尽量做资源隔离,不能让一个跑批任务把GPU占满、把在线推理服务拖垮。
推理结果生成完之后,最好自动触发一轮评估脚本,比如计算mAP、准确率、召回率,或者抽取错误样本。评估通过后才算任务完成,评估不通过就标记为“需人工检查”。这个机制能很大程度避免“模型跑了但结果根本没法用”的尴尬。
5. 常见问题与排查技巧实录
5.1 数据侧的几个典型故障
先分享几个我在数据侧遇到过的真实问题,这些问题的排查思路比问题本身更有参考价值。
第一个是“Mac系统数据占用过大”。这个现象其实很普遍,因为macOS会把缓存、日志、临时文件、App支持文件都算进“系统数据”里。排查时我先用终端du命令定位大目录,重点看~/Library/Caches、~/Library/Containers、Docker.raw、Xcode派生数据这些位置。清理之后如果还嫌空间不够,可以备份旧数据到移动硬盘或对象存储,再删本机副本。这一步的核心思路是:先确认哪些是可再生的缓存,哪些是不可再生的重要数据,千万别一键清理把重要实验数据也删了。
第二个是“无法读取USBPerf注册表项下的First Counter值”。这个错误一般出现在Windows服务器上,是性能计数器注册表损坏导致的。原因是某些软件卸载或系统更新会破坏PerfLib键值。修复方法是用lodctr /r命令重建性能计数器。如果不行,可以手动从其他相同系统版本的机器导出对应注册表项导入。这个坑在监控平台部署时很常见,因为监控组件往往需要读取系统性能计数器。
第三个是“MySQL数据恢复”。数据表被误删或者被错误更新,是每个运维都会遇到的噩梦。如果开启了binlog,可以通过binlog回放到误操作前的时间点;如果做的是全量备份加binlog增量,基本上可以恢复到任意时间点。所以我一直强调,数据库备份不只是“每天导出一份SQL”,而是要开启binlog并定期做全量备份。在这个基础上,配合定时恢复演练,才能确保“恢复”这件事不是纸上谈兵。
5.2 推理侧的几个典型故障
推理侧的问题通常集中在GPU占用异常、推理速度慢、显存溢出这几个方向。
GPU占用异常低,最常见的原因是CPU瓶颈。比如你在YOLO推理之前对每张图片做了复杂的前处理,而前处理没有并行化,GPU就只能在那里等着。解决办法是开多线程加载和预处理数据,或者用DataLoader的异步加载机制。另外,如果模型本身很小而输入图像分辨率很低,GPU计算任务太少,占用率也上不去,这时候减小Batch意义不大,反而要调大Batch或者把多个小请求合并。
GPU显存溢出(OOM)也很典型。常见原因是输入Batch过大、模型精度选了FP32、或者存在显存泄漏。排查时先用nvidia-smi观察显存占用是否随请求次数逐步增长,如果逐步增长大概率是代码里有Tensor或内存没有释放。另外,TensorRT的engine文件在多个服务实例并发加载时也会占用大量显存,如果部署了多个副本,要合理预估总显存需求。
大模型推理慢的问题我在前面说过,重点排查点包括:量化版本、上下文长度、并发参数、GPU是否真的在加速。用Ollama的时候,还可以通过ollama ps命令查看当前模型在CPU还是GPU上运行。如果你发现模型显示在CPU上,赶紧检查CUDA、ROCm版本和驱动,把GPU加速重新配好,推理速度能快一个数量级都不夸张。
5.3 问题速查表
下面的表格是筛选频率最高的几类问题,我把现象、可能原因和解决方向整理在一起,方便你参考。
| 问题现象 | 可能原因 | 排查与解决方向 |
|---|---|---|
| 系统数据占用过大 | 缓存、日志、旧备份堆积 | 用du定位大目录,分清理缓存和备份重要数据两步走 |
| MySQL数据误删 | 缺少备份或binlog未开启 | 建立全量+binlog备份机制,定期演练恢复流程 |
| GPU利用率低 | CPU预处理或内存拷贝成为瓶颈 | 使用异步数据加载、pin memory,调整Batch Size |
| 推理OOM | Batch过大或精度设置过高 | 降低Batch,改用FP16,检查显存泄漏 |
| 大模型生成极慢 | 模型未走GPU、量化级别高损失?并发不够 | 检查ollama ps,确认GPU加速;调整量化与上下文长度 |
| 性能计数器读取失败 | 注册表PerfLib损坏 | 用lodctr /r重建,或从同版本机器导入注册表键 |
| 标注结果混乱 | 缺少标注规范或质量抽检 | 先出规范和试标样例,再批量执行,标注过程加抽检 |
5.4 一些不想写进标题里的琐碎经验
有些经验单独拎出来讲可能显得不够“高深”,但它们在真实项目里很救命。
比如每个跑批脚本必须打印日志,日志里要带时间戳、数据ID、处理结果和耗时。没有日志的脚本,一旦跑挂根本无从排查。我见过有人用print("start")这种极简日志,跑了半天崩了,根本不知道崩在哪一批数据。现在我这边强制要求:脚本首尾各打一条日志,循环里每处理一千条打一条进度日志,异常捕获时打完整堆栈并记录当前数据ID。
比如推理服务上线前要做压测,不要只用一个样本测通就部署。压测至少要看三类指标:单请求时延、吞吐量(每秒处理多少请求)、失败率。用简单的ab或者wrk就能测,不需要复杂工具。压测时观察GPU占用和显存,如果并行已到极限但时延仍然不达标,再考虑换更小模型或者做模型剪枝。
比如数据版本和模型版本要同时记录。推理结果的每个字段如果不知道是哪个模型版本产出的,这个结果基本就没有分析价值。所以无论是数据表还是备份目录,都要带版本号或时间戳。
还有一些特殊情况:当某些推理结果涉及非常规的数学推导,比如符号推理里的QBF问题,或者用公式驱动的扩散模型等,我建议在代码里保留和论文一致的公式注释,并且把中间结果落盘。这样一旦结果异常,能逐个环节对比,看到底是公式实现错、初始化错,还是数据输入错。这类问题最难查,因为表面上看程序都在跑,但输出整体不可信。提前把中间结果存下来,排查效率能高很多。
6. 一点个人心得
搭建这样一套“从数据到推理”的一站式平台,最难的部分从来不是某个算法模型跑不通,而是把每个环节的边界和输入输出定义清楚。
数据侧的备份策略、治理规则如果没定好,后面推理出来的结果基本不可信;推理侧如果不把结果回写、不记录模型版本和数据版本,任何实验结果都无法复现。我自己的习惯是,接手一个新项目时先问三件事:数据从哪来、格式是什么、备份恢复怎么做。这三件事搞不清楚,模型精度再高都不敢投入使用。
这个平台也不会是一次性建完就一劳永逸的。随着后续接入更多数据集、更大参数规模的模型,存储层和推理层还需要反复调整。但“数据—训练—推理—反馈”这条流水线的思想不会变,只要坚持按这个框架迭代,平台会越用越顺手。希望我踩过的这些坑,能帮你少走一点弯路。
