大数跨境

4B 小模型先动笔,卡壳再摇大模型:PyroDash 如何把 API 账单写进强化学习

4B 小模型先动笔,卡壳再摇大模型:PyroDash 如何把 API 账单写进强化学习 AI大模型智能体前沿
2026-10-19
3
导读:端侧跑 4B 小模型,搞不定再调一次云端大模型 API。看看把真实 Prefill 账单写进强化学习奖励是怎么训出来的。

导读大小模型协同,难点不只在选谁答题,还在何时交接。火思动力(PyroMind Dynamics)的 PyroDash 让 4B 小模型先生成,再用控制符把已有草稿至多一次交给冻结大模型续写。它通过三阶段训练,将输入、输出分别计价的规则纳入强化学习奖励。在论文的数学基准与价格模型下,省钱档降本 96.4%,准确率低于纯大模型 3.13 个百分点;提分档则以更高的大模型参与度换来 6.36 个百分点的提升。两组结果揭示了这套分工的收益,也划出了适用边界。

本文 4364 字,阅读约 11 分钟

为什么要把路由决策放进生成过程

小模型先写,控制符触发一次单向交接

三阶段训练:从认识控制符,到学会怎样交接

降本 96.4% 与提分 6.36 个百分点,各自付出了什么

从数学推理到编程 Agent,还要核对哪些边界

为什么要把路由决策放进生成过程

把大小模型搭在一起省钱,工程上很容易想到让大模型当包工头、小模型当打工人:主控大模型先读题、拆任务,再把简单的局部工作分给小模型。

这能分摊工作,却有一笔固定开销:大模型已经读过完整输入。商用 API 通常分别收取输入和输出 Token(模型处理文本的基本单元)的费用;输入侧主要对应预填充(Prefill),输出侧对应逐步解码(Decoding)。即使任务本可由小模型独立完成,只要先经过大模型规划,就已产生一次大模型调用成本。

另一条路线是请求级路由。例如 RouteLLM 在生成前,判断整道题该交给小模型还是大模型。FrugalGPT 则采用级联:先调用较便宜的模型,评估答案后再决定是否升级。两者决策时机不同,但处理的单位都是一次请求或一份答案。

对于多步推理,困难可能到中途才暴露:前面只是读题、换元、列式,后面才进入复杂推导。请求前的路由无法利用尚未生成的中间步骤;答案级级联虽能事后升级,也没有直接提供在某个 Token 位置接续同一段推理的机制。

把决策移进生成过程,可以利用这些中间信息,但也会带来协调成本。CITER 用外部路由器逐 Token 选模型;RelayLLM、GlimpRouter 等方法允许大模型多次介入。对每次调用都需重新提交上下文的无状态 API,反复切换可能增加输入计费;实际金额还受缓存规则影响。

投机解码优化的则是生成速度:小模型先起草,大模型批量验证草稿,大模型仍参与整个生成过程。它与“简单题只由小模型完成,必要时才调用大模型”的目标不同。

沿着决策时机和大模型参与方式,可以看出几条路线的区别:

协同方式 决策时机 大模型如何参与 主要成本约束
大模型主控派活 先读题、再拆任务 规划与汇总 简单请求也先产生大模型费用
请求级路由,如 RouteLLM 生成前 接手整道题 无法利用中途生成的推理信息
答案级级联,如 FrugalGPT 前一份答案评估后 升级调用 可能重复生成答案
生成期多次协同 按步或按 Token 多次介入 可能重复提交增长的上下文
PyroDash 单向交接 小模型生成过程中 至多一次续写 交接草稿仍需计入大模型输入

Agent 路由不只是选模型:还要分清治理和 GPU 调度

火思动力于 2026 年 7 月公开的 PyroDash,把路由变成小模型自己生成的一个动作:默认先写;需要帮助时,发出控制符并交出已有草稿。

小模型先写,控制符触发一次单向交接

PyroDash 由三个部件组成:可训练的小模型 M_s、参数保持冻结的大模型 M_l,以及执行交接的协同引擎(Collaborate Engine,CE)。论文实验分别使用 Qwen3.5-4B 和 GLM-5.2-FP8。这里的“冻结”指训练过程中不更新大模型参数,续写仍然需要计算。

PyroDash 协同推理架构:小模型流式生成,控制符触发上下文打包与单向交接 | 来源:PyroDash 论文 v1,Figure 1

顺着图中的数据流,一次请求只走两条路径之一:

1. 小模型独立完成:原始问题 q 连同协同提示 P_s 交给小模型。若生成结束前没有出现交接控制符,CE 直接返回小模型输出,不调用大模型。

2. 中途交接给大模型:小模型生成 <|llm_offload|>(论文记作 τ_off),CE 检测到这个停止序列后终止小模型解码,剥离控制符,把已有草稿 C_s、原始问题 q 与续写提示 P_l 一起交给大模型。大模型续写到最终答案,CE 将两段输出拼接返回,控制权不再交回小模型。

决策由小模型作出,CE 负责执行。所谓 Token 级自路由,是小模型可以在生成某个 Token 时触发交接,并非大小模型必须逐 Token 轮换。

这套协议只传文本,不要求访问大模型内部的输出概率,也不要求重训大模型,因此适合探索“本地小模型 + 云端 API”的组合。不过,接口能接通与效果能迁移是两回事:换后端时仍需检查续写提示、上下文格式和训练策略是否匹配。论文实验也没有直接验证手机或 Mac 上的端云部署体验。

交接的价值,在于大模型可以利用小模型已经完成的工作。官方项目页展示了一条对数方程组求解轨迹:

官方案例 AIME-2024 #744 的交接轨迹:蓝色小模型完成换元后发出紫色控制符,橙色大模型接续求解 | 来源:PyroDash 官方项目主页

小模型先设 a = log_2(x)、b = log_2(y)、c = log_2(z),把对数关系化为三元一次方程组,并写出目标式 |4a + 3b + 2c|。交接后,大模型沿着这份草稿求解,得到最终答案 m + n = 33。

论文还给出柠檬树盈利案例:小模型先算出年净收入 7 × $1.50 - $3.00 = $7.50,大模型再通过 7.50n > 90 区分第 12 年回本与第 13 年开始盈利。简单的花朵计数题则由小模型独立算完。这些轨迹说明两条路径如何运行,但不能单靠几个成功案例证明每次交接都发生在最佳位置。

三阶段训练:从认识控制符,到学会怎样交接

架构跑通后,核心难点在于:小模型词表里原本没有 <|llm_offload|>,怎样让它既学会这个动作,又学会权衡继续作答与调用大模型?

PyroDash 只训练小模型,大模型参数全程保持冻结。三阶段训练依次解决动作表示、行为示范和策略优化。

PyroDash 三阶段渐进式训练流程:从相对难度数据构造,到控制符词嵌入学习、冷启动 SFT,再到基于成本基线的 GRPO 强化对齐 | 来源:PyroDash 论文 v1,Figure 2

训练数据来自团队构建的 EasyHard-24k(24,061 条样本),按“模型相对难度”划分:直接用基础版 Qwen3.5-4B 跑每道题,答对的进 easy 子集;答错的 hard 题不直接用小模型的错误轨迹微调,而是由大模型重构出正确简洁的思维链(CoT),并在其中动态插入 1 到 4 个候选 <|llm_offload|>。这些是训练用的合成交接位置,后续还要通过奖励调整;推理时仍在首次触发后停止小模型。数据随后扩展为不含协同提示且无控制符的语料 A(保住独立解题能力),以及含协同提示 P_s 的语料 B(其中 easy 样本仍不含控制符,防止模型养成“有提示就甩锅”的习惯)。

基于这份数据,三阶段训练依次推进:

1. 第一阶段:学会表示控制符。用句号、换行符、结束符等自然断点的向量均值加噪声,初始化新控制符的输入、输出嵌入。在语料 B 上训练嵌入层、输出头和临时 LoRA 适配器;LoRA 是只训练少量附加参数的微调方法。训练后丢弃临时适配器,只保留新控制符对应的参数行,尽量保住基座能力。

2. 第二阶段:用示范建立初始行为。冻结词嵌入,用 LoRA 在语料 A+B 上做监督微调(SFT),让模型同时学习独立作答与开启协同提示后的条件性交接。

3. 第三阶段:按最终答案和成本调整策略。在 DAPO-Math 上使用群组相对策略优化(GRPO):对同一道题采样 8 条轨迹,有的独立完成,有的在不同位置交接;发生交接时,训练环境实际调用冻结大模型续写。系统根据最终答案是否正确、整条轨迹的成本,为这 8 次尝试打分,再比较组内相对表现。

关键在于第三阶段怎样算账。论文先让纯大模型回答同一道题,得到基线成本 Cost_base;再计算协同轨迹的成本 Cost_act。归一化开销与总奖励分别为:

R_eff = Cost_act / Cost_base

R_total = R_acc - λ · R_eff

其中 R_acc 是根据标准答案核验得到的 0/1 正确性奖励,λ 是训练时采用的成本惩罚权重。GRPO 不需要额外的价值评估网络,但这里的答案得分仍由核验程序给出,并非模型自己判断谁答得好。

对于发生交接的轨迹,论文式(5)使用如下紧凑成本表达:

Cost_act = (C_s + C_pre) · T_s + C_dec · T_l,dec

T_s 是交接前的小模型输出量,T_l,dec 是交接后的大模型输出量;C_s 是小模型统一 Token 单价,C_pre、C_dec 分别是大模型输入、输出单价。它突出了一笔容易漏算的钱:交接前的草稿既消耗小模型计算,又会作为大模型输入再次计费。

这不是适用于所有路径的完整账单公式。小模型独立完成时没有大模型输入费;发生交接时,原始问题和提示词也属于输入。论文将上式作为训练奖励的紧凑记账形式,评测则按小模型输入、小模型输出、大模型输入、大模型输出四项分别统计和折价。因此,“训练纳入 API 计费结构”与“实测商用 API 账单”需要分清。

同一道题的 8 条轨迹按总奖励计算组内优势,训练会提高较高奖励动作的概率、压低较低奖励动作的概率,梯度只更新小模型。更高成本会降低奖励,但是否获得负优势仍取决于组内比较,并不是开销一超过纯大模型就必然受罚。

由此,小模型学习的不只是发出控制符,还包括是否交接、交接前写多少,以及怎样让联合输出既正确又划算。

降本 96.4% 与提分 6.36 个百分点,各自付出了什么

论文最醒目的两组结果是降本 96.4%,以及准确率超过纯大模型 6.36 个百分点。它们来自不同成本权重下训练出的策略,不能合并为一个档位的成绩。λ 也不是部署后随手一拧就能改变同一模型行为的推理参数。

评估覆盖 GSM8K、Minerva、OlympiadBench、AIME25、AIME24 五项数学基准。AIME 使用每题 32 次采样的平均准确率(avg@32),其余使用贪心解码的单次准确率(pass@1),下表“平均准确率”为五项基准结果的均值。

费用均为论文按 Token 用量估算的测试集总成本,不是单题价格或实际付款账单。论文使用的每百万 Token 输入/输出单价分别为:Qwen3.5-4B 0.05/0.08 美元,GLM-5.2-FP8 0.90/2.86 美元。这是实验价格口径,不代表当前任一服务商的报价。

五项数学推理基准上的总成本(美元)与平均准确率(%)成本—准确率分布:橙色星号展示了 PyroDash 在 λ=0.6、0.2、0.1 到 0.05 下的表现,与 RouteLLM、GlimpRouter 及独立模型对照 | 来源:PyroMind Dynamics 官方 GitHub 仓库(对应论文 Figure 3)

把论文表 1、表 2 的主要结果压缩到同一组指标中:

方案 平均准确率 (%) 大模型输出占比 (%) 每题大模型调用 总成本 (美元)
纯小模型 28.36 0.00 0.000 2.26
小模型 + SFT 46.25 0.00 0.000 1.32
纯大模型 57.68 100.00 1.000 49.36
RouteLLM 52.74 77.37 0.808 44.62
GlimpRouter 54.20 75.11 1.200 31.61
PyroDash,λ=0.6 54.55 1.90 0.012 1.78
PyroDash,λ=0.1 55.29 8.19 0.058 4.71
PyroDash,λ=0.05 64.04 95.34 0.975 39.29

“大模型输出占比”只统计解码 Token,不包含输入,所以 1.90% 不能直接读作大模型费用占比。RouteLLM 数值采用论文表 2 的配置。

省钱档 λ=0.6 的重点是少调用:至多一次交接的约束下,平均每题调用 0.012 次,对应约 98.8% 的请求由小模型独立完成。大模型输出从 1708 万降到 23 万 Token,总成本从 49.36 降至 1.78 美元,降本 96.4%,同时平均准确率下降 3.13 个百分点。它的准确率仍高于两条路由基线,但不能据此说与纯大模型完全等效。

提分档 λ=0.05 几乎反过来:约 97.5% 的请求发生交接,大模型承担 95.34% 的输出 Token。联合准确率达到 64.04%,比纯大模型高 6.36 个百分点,同时成本降低 20.4%。AIME24 上,准确率从 46.56% 提升到 63.75%。这个档位主要展示了协作的质量收益,绝大部分生成工作仍由大模型完成。

官方轨迹显示,前置换元、列式可以为大模型提供有用上下文;但这些案例还不足以把整体提升全部归因于“化简降低了大模型出错率”。SFT 小模型独立作答的 46.25%,与完整协同策略的 64.04%,也包含推理时是否调用大模型的差别,不能把全部差值都算成 GRPO 对小模型自身能力的提升。

居中的 λ=0.1 以 4.71 美元取得 55.29% 的准确率,说明训练可以得到不同折中点。工程上应先明确业务可接受的正确率,再比较相应策略的成本,而不是只挑最低账单。

从数学推理到编程 Agent,还要核对哪些边界

7 月公开数学基准模型(官方开源了 Lambda-0.6 与 Lambda-0.05 权重)后,PyroMind 在 2026 年 9 月 1 日于官方 GitHub 仓库发布了面向软件工程的 Agentic-Preview 更新:

•控制符升级为 0–9 思考档位选择:从 <|llm_offload|> 升级为 <|llm_offload|>N<|/llm_offload|>(N 为 0 到 9 的整数),小模型在交接时可动态指定大模型的思考深度档位(Thinking Level)。

•引入 OPD 蒸馏并初验编程基准:Stage 2 升级为 SFT + OPD(On-Policy Distillation,即让小模型在自己采样生成的轨迹上向大模型对齐输出分布),在 ScaleSWE 与 TMax 数据集上训练,Qwen3.5-4B 可分别搭配 DeepSeek-V4-Flash 或 GLM-5.2(下图展示了相关模型与 λ = 0.3、λ = 0.1 策略的对照);在 SWE-Bench Verified 基准上初步实现约 20% 成本节省,官方文字称准确率下降约 5%,但没有在该说明中明确这是相对降幅还是百分点;这一预览口径不与数学基准混用。

2026 年 9 月 Agentic-Preview 编程基准成本与准确率成本—准确率分布:横轴在 300 美元前采用线性刻度、之后采用对数刻度,展示两种 λ 策略与独立模型的结果 | 来源:PyroMind Dynamics 官方 GitHub 仓库

这些是仓库披露的预览进展,完整细节尚待补充。结合论文第 6 节的限制与部署条件,落地前还需核对四件事:

1. 单轮单次交接 vs 多轮 Agent 循环:7 月论文验证的是单轮至多一次单向交接;在多步工具调用的 Agent 循环中,多轮往返的上下文缓存(Prompt Caching)命中率与交接 Prefill 成本如何折算,仍需新的完整实测,9 月预览不能直接补齐这些结论。

2. 折算费用与实际部署开销:论文实验是在服务端 GPU 集群(4 张 H100 跑 Qwen3.5-4B,8 张 B200 跑 GLM-5.2-FP8,vLLM 部署)上按公开标价折算美元的。若实际采用“端侧设备跑 4B + 公网调云端 API”,需要实测小模型生成、跨网传输和云端首字等待的总延迟,并核算设备、服务和缓存费用,不能把论文的 Token 折价直接当作端云总成本。

3. 触发位置的微观归因尚未完全验证:论文统计了整体准确率与账单,但尚未逐题检验小模型发出 <|llm_offload|> 的具体 Token 位置是否严格对应其推理即将出错的临界点。论文也未单独比较归一化成本惩罚与直接 Token 成本惩罚,不能把归一化本身说成已被消融证明的增益来源。

4. 模型搭档与价格比绑定:训练数据与奖励基线针对特定大小模型的能力差及价格比定制。如果业务中更换后端闭源大模型或前端小模型尺寸,适合的交接策略可能随之改变,需要重新评估乃至训练交接策略,不能只改一个运行时参数就沿用原成绩。

对预算有限的团队,PyroDash 提供了一个具体的设计思路:让小模型的中间工作参与最终答案,并把交接带来的输入费一起纳入训练目标。它的工程价值在于用不同训练策略选择可接受的准确率与成本组合;至于能否搬进自己的业务,还要用真实请求、真实价格和真实部署条件重新算账。

论文: https://arxiv.org/abs/2607.20327代码: https://github.com/Pyromind-Dynamics/PyroDash项目主页: https://Pyromind-Dynamics.github.io/PyroDash数据集: https://huggingface.co/datasets/pyromind/easyhard-24k模型(Hugging Face): https://huggingface.co/pyromind模型(魔搭社区): https://modelscope.cn/models/Pyromind/PyroDash-4B-GRPO-Lambda-0.05

参考资料

•PyroDash 论文(Cost-Efficient Token-Level Small-Large Language Model Collaborative Inference,arXiv:2607.20327v1): https://arxiv.org/abs/2607.20327

•PyroMind Dynamics 官方 GitHub 仓库与 Agentic-Preview 更新说明: https://github.com/Pyromind-Dynamics/PyroDash

•PyroDash 官方项目主页: https://Pyromind-Dynamics.github.io/PyroDash

•FrugalGPT 原始论文(请求级 API 级联): https://arxiv.org/abs/2305.05176

•EasyHard-24k 数据集(Hugging Face): https://huggingface.co/datasets/pyromind/easyhard-24k

蘑菇头表情图片来源:蚊子动漫

THE END

文章仅做学术分享,如有侵权请联系删除,非常感谢!

【声明】内容源于网络
0
0
AI大模型智能体前沿
分享AI大模型智能体前沿知识,探寻多元应用,洞察未来趋势,带你一路 “卷” 赢行业!🔥
内容 1165
粉丝 0
AI大模型智能体前沿 分享AI大模型智能体前沿知识,探寻多元应用,洞察未来趋势,带你一路 “卷” 赢行业!🔥
总阅读23.8k
粉丝0
内容1.2k