Kolibri 不只是一次学术性的 MoE 尝试:Aleph Alpha 打造了一款支持英语和德语的 Transformer,在庞大的总参数规模与较少的单 token 激活参数之间取得平衡。Kolibri 总计包含 384 个专家,每个 token 会被路由至 6 个路由专家和 1 个共享专家,因此每个 token 的计算量约为 34.6 亿参数。这样的稀疏设计,加上由 50 个 block 组成的 Transformer 堆栈和混合注意力机制,让实际应用能够运行具备前沿能力的模型,同时避免密集模型通常带来的计算负担——后者往往只能依赖云端托管。
Kolibri 的注意力设计是影响实际部署的关键因素。Aleph Alpha 将 40 层滑动窗口注意力与 10 层全注意力结合,只让少数 block 的 KV 缓存随上下文长度增长,从而使内存用量的增长低于上下文长度的线性增幅。结果是,据称模型原生支持最多 1,048,576 个 token。对于需要推理超长文档的企业场景——如法律案件、航空航天日志和技术手册——这意味着可以用一个自托管模型通读完整档案,无须依赖成本高昂、严重依赖检索的变通方案。
从部署角度看,Kolibri 的 FP8 检查点(约 78GB)和 Apache-2.0 许可证改变了实现自主可控部署的成本考量。Aleph Alpha 说明,模型可通过 vLLM 在 B200/B300/H200 单卡上运行,也可使用 H100 SXM5 双卡配置。对许多受监管机构而言,在本地可控的基础设施上托管开放权重,可以减少供应商锁定,也更容易满足数据驻留要求。不过,要实现稳定、可预测的延迟与吞吐量,仍需精心调优,尤其要处理好生产请求分布下的路由和专家负载均衡。
Kolibri 的训练方式和分词器选择也会影响实际使用。其 128k UniBPE 词表相较于某些前沿分词器能减少德语分词开销,训练数据也着重覆盖德语内容。这些特点能在以德语为主的企业工作负载中带来可衡量的成本和性能收益。同时,Aleph Alpha 披露的训练流程包含超长序列预训练和基于 RL 的后训练阶段,表明模型针对持续推理进行了调优,并在上下文不足时具备拒答倾向。


