大模型时代,我训练了一款小模型(4MB)

5 小时 5 分钟前
 Sanko

前情提要:开发了一款轻量养车记账软件,欢迎油车电车高端人士体验( 7 天前发的上篇)

这 7 天更新了什么

一句话:把「 AI 帮你记」背后的模型,从手机上跑一个 2.33GB 的大模型,换成了自己训练的 4.5MB 小模型 lingxi (灵犀) 0.1.0,纯 Dart 推理、无任何原生依赖。

为什么不用大模型

上篇评论区有人担心「端侧 AI 会让人变傻」,但其实更尴尬的是技术账:让一个通用 LLM 做槽位抽取,等于让它承担本该由规则承担的活——中文数字换算、元→分乘 100 、相对日期推算、句中多数字消歧、备注改写。

于是引入两类最难防的错:金额量级错误(×100 出错,7 元记成 700 元,用户会漏过去)和备注幻觉(编造原文不存在的文本,污染账本)。而且抽取任务真正承载的决策信息只有约 3 bit ,LLM 是按语言建模约 45 bit 造的,纯粹浪费容量。

所以我换了个思路:候选枚举 → 模型选择 → 规则解引用

"呃,昨天在壳牌那儿,95 的加了 42 升,367 块 8"
① 枚举   把所有片段编号,并顺手把计算全做完:
         N1=42 升(42.0)  N2=367 块 8(=36780 分)  D0=昨天(=设备当天-1)
② 标记   ⟨N1⟩42 升 ⟨N2⟩367 块 8
③ 模型   一次前向:指出 N2 是金额、N1 是升数、D0 是日期、类别=加油
④ 解引用 按编号取回算好的值 → 草稿

模型从头到尾没做过一次算术、没做过一次日期换算、没写过一个原文不存在的字。中文数字、小数、心算、自我改口、多数字消歧、相对日期这六类难题,从"生成/计算"坍缩成了"选择",容量需求直接掉两个数量级,且结构上不可能幻觉——它只能指认候选。

lingxi 0.1.0 本身

参数量 4.42M ( 4 层 × 256 宽编码器,RoPE + SwiGLU + RMSNorm ,全网络无 bias )
产物 slot.ptw 4.49MB( int8 逐行量化;加载反量化为 fp32 ,常驻内存约 18MB )
运行时 纯 Dart,无原生库、无新依赖;常驻 isolate ,Android/iOS/macOS/Windows 通吃
推理时延 p50 ~18ms / p90 ~48ms ( M4 Pro, JIT 实测; AOT 更快,300ms 预算内)
训练 发布版 3000 步,CPU 上约 4 分钟( 242s )训完

评测(这是我最想吹也最想交代的部分):

一个反直觉的结论来自容量扫参:精度和参数量几乎无关——630K 参数金标准已 100%,14.5M 无任何可辨识提升,瓶颈是数据与词汇覆盖而不是模型大小。当务之急是把评测集做大,而不是把模型做大。

训练数据怎么来的?模板族合成 + 用 qwen3:14b 当"教师"只借它的词汇多样性(一次问 30 种说法:「给车洗了个澡」「掏了过路费」「补点电」……),模式引擎再乘上槽位空间生成十万级样本;评测诚信纪律是生成的说法与评测集任何文本重合一律丢弃。整个训练链路、契约、测试向量都在 pitstop-ml/ 工作区,权重发布到 HuggingFace ,有防回归门禁。

这周踩的坑(省略一万字版)

  1. 真实语料只有 55 条时,模型和 0.75 vs 1.00 的差别都是自欺,先把评测集做大再谈调模型;
  2. 全角句号曾被归一化成小数点,导致分句器刻意躲着句号走——把它从映射表放出来当句子边界后,「昨天加油 300 。今天洗车 30 。」自然就对了;
  3. 纯 Dart 没有 int8 SIMD ,int8 只用来(省下载),计算仍走 fp32——下载体积和计算精度就此解耦。

App 还是那款不上应用商店的养车记账(本地单机 / 在线共享双模式,多车、固定费用、统计都有),现在多了个 4.5MB 的 AI 帮你记。下载:https://pitstops.top

模型仓库:https://huggingface.co/XiaoDaZhen/pitstop-lingxi-0.1.0( slot.ptw ,4.49MB )

有任何想问的尽管问,我很乐意展开聊。

1126 次点击
所在节点    分享创造
12 条回复
little_cup
4 小时 54 分钟前
对这个话题很感兴趣,最近做端侧的太少了。我的通知滤盒做了端侧推理,简单的文本分类,但特点是需要支持端侧继续学习来做个性化。
最近用 fable 5.1 搞了 40 来个小时,搞出了一个 8M 的双输出模型(固定权重+动态学习)。

但是……很想吐槽,为什么要用 AI 来写这篇文章,请换位思考你作为读者读的下去吗?
Sanko
4 小时 39 分钟前
@little_cup #1 我文笔太差了抱歉!关于动态学习这里我是做了客户端的主动点赞点踩入口,以及端侧模型识别结果如果与最终提交不一致有 diff 也会静默上报给服务端
loading
3 小时 36 分钟前
4.5MB 的模型感觉挺有意思的。
heyjany
3 小时 33 分钟前
@Sanko 也许你自己写更好。

这种没衔接突然跳跃、莫名其妙的修饰语,读起来太难受了
SeaBreathe
3 小时 15 分钟前
专注于养车记账吗?直接拓展成记账 App 怎么样
Sanko
2 小时 51 分钟前
@SeaBreathe 暂不考虑通用记账,类似可选择太多了。这个本来是我个人场景使用的 app ,奈何 vibe coding 越蹬越爽,就做成了现在这个程度
Sanko
2 小时 49 分钟前
@heyjany 下次尽量自己写了
0xo
2 小时 49 分钟前
你的场景下能 beat needle2 吗?
yuxizhe
2 小时 44 分钟前
你这需求不算训模型呀
Sanko
1 小时 19 分钟前
@0xo 包能的,我这个是中文输入(🐶
Sanko
1 小时 18 分钟前
@yuxizhe AI 识别一句话当中的记账分类,金额,备注信息,充电度数/加油升数 等,算啊
imqiyue
15 分钟前
有点腻害的啊,膜拜下~

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/1240647

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX