[控场AI]
模型

Keera

一个97.5M参数的Python代码语言模型,架构类似GPT-2 small,由开发者从零训练完成,包含自研分词器、预训练和微调流程。

时间轴 (近 90 天)

10月18日

开发者从零构建了一个97.5M参数的Python代码模型Keera,训练总成本为1.40美元。

待验证50%
10月18日

Keera的架构基于GPT-2 small:12层Transformer块,隐藏维度768,总参数量97.5M。

待验证50%
10月18日

Keera使用自研分词器,词表大小16,384,在Python代码上训练,平均3.9字符/token。

待验证50%
10月18日

Keera使用RoPE位置编码、RMSNorm、权重共享嵌入和1,024 token上下文。

待验证50%
10月18日

Keera在2B token的codeparrot-clean数据集上预训练,遵循Chinchilla约20 token/参数的规则。

待验证50%
10月18日

在Vast.ai上租用RTX 5090(每小时0.584美元),使用bf16和torch.compile,Keera预训练吞吐约234,000 token/秒,2B token训练耗时2小时23分钟,验证损失1.501。

待验证50%
10月18日

Keera使用self-oss-instruct-sc2-exec-filter-50k数据集微调,该数据集含50k条Python指令和答案,答案代码均实际运行并通过测试。

待验证50%
10月18日

在HumanEval上,Keera预训练模型pass@1为3/164(1.8%),微调后为9/164(5.5%)。

待验证50%
10月18日

作者明确标注HumanEval结果存在局限:未检查数据污染、预训练与微调评估设置不同(非受控测量)、保存的微调逐题结果文件含NUL字节不完整。

待验证50%
10月18日

在HumanEval pass@1对比中,phi-1-small(350M参数)达到45%,Keera微调(97.5M参数)为5.5%。

待验证50%

全部知识事实 (10)

来源文章