
初创公司 Reflection AI 今日在 Hugging Face 平台公开了其首款大语言模型 Beam 的权重,并同步发布相关技术资源,支持开发者对算法进行定制化修改。
Beam 于上周首次亮相,凭借约 5000 亿参数规模,成为业界最先进的开源大语言模型之一。基准测试显示,其表现接近参数量为其四倍的模型,且算力与内存消耗显著更低。
高效训练与可靠性优化
Reflection AI 动用数千张高端显卡,历时两个月完成 Beam 的训练。针对大规模训练中易发的质量下降问题,工程师开发了一套自定义可靠性优化算法,重点解决由环境配置细微不一致引发的“训练-推理不匹配”缺陷。
此外,公司还构建了一套自动修复系统,可在检测到特定数据质量错误时,将训练进程回滚至较早阶段继续,而非从头重来。得益于上述优化,Beam 的训练过程实现了近乎零集群停机时间。
多样化版本与推理加速
此次发布的 Beam 权重包含标准版及多个缩减规模的量化变体。后者通过缩小数据存储单元,以轻微牺牲输出质量为代价,换取更高的硬件效率。
随权重一同发布的还有“训练配方”(即微调代码模板)及配套评估工具。开发者可据此创建定制工作流,例如增强模型的编程能力,并衡量微调对功能及安全护栏的影响。
为降低推理硬件门槛,Reflection AI 发布了用于“推测解码”的草稿模型(drafter model)。该轻量级模型先生成响应初稿,再由 Beam 校验纠错,从而提升效率。同时,官方提供了适配主流服务引擎的即用型配置及融合内核(fused kernels),将多次推理计算压缩为单次操作,进一步加快处理速度。
Reflection AI 计划在未来几天内发布基于“自蒸馏”技术的额外训练配方,以进一步优化输出质量,并以开源贡献形式分享模型开发过程中的经验教训。