给 Gemma 4 加上信心分,本地模型与云端 API 的混合路由怎么玩?
Cactus Hybrid 的核心逻辑是给 Gemma 4 增加了一个极其轻量级的“探测层(Probe Layer)”。这个层的参数量只有 68k,几乎可以忽略不计,但它的作用是让模型在输出答案的同时,能够给出一个 0 到 1 之间的置信度分数。
这里最值得深挖的是它的实现机制。很多早期的方案是让模型在回答完之后,再问一遍“你对这个答案有信心吗?”,这种自我反省(Self-Reflection)极其不可靠,模型往往会过度自信。另一种方案是观察 Token 的熵值(Entropy),但这更像是在抛硬币,因为高熵并不一定代表错误。Cactus Hybrid 绕过了这些坑,直接从模型的隐藏状态(Hidden State)中提取信号。这意味着它在观察模型内部的“思考状态”,而不是看最终输出的文字。
最让我惊讶的一个细节是,这个探测层在训练时并没有接触过音频数据,但实测在音频 Benchmark 上依然能精准判断输出是否正确。这说明它捕捉到的是一种跨模态的“正确性信号”,这种泛化能力在如此小规模的参数量下是非常罕见的。
在实际的混合路由架构中,Gemma 4 充当了前置过滤器的角色。当探测层给出的置信度分数高于某个阈值时,系统直接采用本地结果;而一旦分数过低,请求会被自动路由给 Gemini 3.1 Flash-Lite。根据实测数据,这种方案能将需要交给云端大模型的请求量压缩到 15%-55%,而在综合效果上,竟然能与全量使用大模型持平。对于开发者来说,这意味着在保证质量的前提下,成本直接下降了 50% 以上。
当然,在实际部署到生产环境前,有几个技术细节必须注意。首先是长度限制,目前该方案仅支持单序列解码,且 Token 上限被锁定在 1024 个,如果你处理的是长文档分析,这个方案可能不适用。其次是路由粒度的问题,这种方案在“按任务路由”(Task-level Routing)时表现最稳,如果你尝试在每一个 Token 生成步骤都进行路由,用户体验可能会因为频繁的切换而打折扣。最后,这种探测层是强绑定模型的,你不能直接把为 Gemma 4 训练的权重套用到 Llama 3 或其他模型上。
目前该方案已经支持 Transformers、MLX 以及 llama.cpp,部署门槛较低。如果你想尝试,可以直接去 GitHub 搜索 cactus-compute/cactus-hybrid 这个仓库。
总的来说,这种“本地小模型 + 信心分 + 云端大模型”的组合,比单纯死磕一个超大规模模型要灵活得多。它把本地模型的速度和云端模型的深度结合了起来,为 AI Agent 的低成本规模化落地提供了一个非常务实的参考路径。