近日,加州大学伯克利分校的Sky Computing Lab 团队发布了Sky-T1-32B-Preview,这是一款开放源代码的推理型人工智能模型,标志着推理型AI 的研发变得更加容易和便宜。该模型在多个关键基准测试中表现出色,甚至与OpenAI 早期版本的o1相媲美。
Sky-T1的训练成本令人瞩目,仅为450美元,这意味着高水平推理能力的复制已变得更加可负担且高效。虽然450美元的费用可能听起来不算低,但与几年前动辄数百万美元的训练成本相比,已是巨大的下降。借助合成训练数据,即由其他模型生成的训练数据,成本得以显着降低。 AI 公司Writer 最近发布的Palmyra X004几乎完全依赖合成数据,开发成本也仅为70万美元。
与大多数AI 不同,推理模型能够有效地自我核查,这使得它们在处理一些常见问题时更加可靠。推理模型通常在得出解决方案时需要更多时间,可能需要几秒到几分钟,但在物理、科学和数学等领域,其可靠性优势显着。
NovaSky 团队利用了另一种推理模型—— 阿里巴巴的QwQ-32B-Preview,生成Sky-T1的初始训练数据,并对数据进行了“策划”,之后使用OpenAI 的GPT-4o-mini 将数据重新整理成更可操作的格式。训练32亿参数的Sky-T1只需约19小时,使用的是一组8个Nvidia H100GPU。参数数量与模型的解决问题能力大致相关。
根据NovaSky 团队的说法,Sky-T1在MATH500这一包含“竞赛级” 数学挑战的集合上表现超越了o1的早期预览版本。此外,Sky-T1在LiveCodeBench 中遇到的难题上也超过了o1的预览版本。然而,在涉及物理、生物和化学等领域的问题上,Sky-T1在GPQA-Diamond 测试中的表现却逊色于o1预览版。
需要指出的是,OpenAI 的GA 版本o1比预览版更为强大,而且OpenAI 预计将在未来几周内发布更为优秀的推理模型o3。不过,NovaSky 团队表示,Sky-T1仅是他们开发具备先进推理能力的开源模型旅程的开始。
“展望未来,我们将专注于开发更高效的模型,以保持强大的推理性能,并探索进一步提升模型效率和准确性的高级技术,” 团队在博客中写道。 “请继续关注我们在这些激动人心的项目上的进展。”