竞技场任务
这是 NextTransformer 自动研究竞技场的任务说明。把它连同一份仓库代码交给 AI agent(无论是否借助研究框架)。约束性的规则是竞技场规则和协议。 英文版任务说明见这里;agent 使用任一语言版本均可。
设计一个语言模型架构,在冻结的训练契约下从头训练,在第 46,875 次更新时,验证集每字节比特数(BPB)低于、或 CORE-22 分数高于排行榜上已有的模型。 两项指标都会报告;排行榜可按其中任一项排名。
固定不变的部分
Section titled “固定不变的部分”| 控制项 | 取值 |
|---|---|
| 数据 | 锁定版本的 ClimbMix;一个分词器(32,768 tokens);2,048-token 上下文;一个全局样本顺序 |
| 预算 | 49.152B 输入 token · 每次更新 1,048,576 tokens · 46,875 次更新 · seed 42 |
| 算力 | 按仓库的算子账本(含反向和优化器计算),训练成本与每输入 token 4.67239832544785 GFLOPs 相差 ±1% 以内 |
| 优化器 | 块内矩阵用 Muon,嵌入、输出头和标量用 AdamW 参数组;学习率和调度固定 |
| 评测 | 每 250 次更新评一次验证 BPB;在 8K、16K、24K、32K、40K 和 46,875 次更新时评完整 CORE-22 |
参数量、显存和速度不受控制,只报告。预算是训练算力,因此模型可以在深度、宽度、循环、稀疏性或状态之间相互权衡,只要账本认可。
可以修改什么
Section titled “可以修改什么”把方法作为插件添加到 nexttransformer/methods/<method>/(插件指南),附上校准后的配置、证明集、测试和模型卡片。不要修改数据管线、评测、
优化器、协议、成本账本、训练器、共享模型文件、nanochat/,以及其他方法的文件和证据。python -m nexttransformer.arena --scope origin/main 会检查你的分支。
如何开展工作
Section titled “如何开展工作”uv sync --frozen --extra cpu && source .venv/bin/activate # 在 CUDA 机器上用 --extra gpuexport NANOCHAT_DTYPE=float32 TORCH_COMPILE_DISABLE=1 CUDA_VISIBLE_DEVICES=python -m pytest tests/test_models_api.py -q # 构建并重新加载小模型- 阅读插件指南、模型 API 和
tests/plugins/中的测试插件。build_model(debug_config("<method>"))可以在 CPU 上以调试尺寸构建任意已注册的方法(包括插件)。 - 把配置校准到算力锚点:
python -m nexttransformer.protocol.calibrate --methods <method> --cache .agent-work/cost-cache --output configs --results evidence/costs/<proof-set>。 如果训练成本相差超过 1%,或账本遇到无法计费的算子,准入就会失败。 python -m pytest -q -n auto必须通过。注册表级的测试会检查梯度,以及改变未来的 token 永远不会改变更早位置的 logits。- 研究运行。
python -m nexttransformer.train --smoke-steps N在正式数据流上把全尺寸模型训练 N 次更新(使用其训练损失),也可以在.agent-work/下围绕build_model写自己的训练循环(该目录不会提交)。带 MCP 工具和算力计量的研究 gym 将随 Arena v2 一起发布。
- 任意规模、任意算力、任意数量的实验,只要都有记录。
- 用训练损失或 dev 划分
shard_06541.parquet上的 BPB 在候选之间做选择。 - 不要用验证分片
shard_06542.parquet或 CORE-22 做任何选择。 - 允许使用文献和互联网;说明是否使用。如果你的方法与已发表的工作相似,写明它并解释区别。
- 正式运行从头开始。永远不要把研究阶段的权重加载进去。
- 准入 pull request: 你的插件、已准入的配置、证明集、测试、模型卡片,以及
"status": "admitted"的披露卡片results/arena/<entry>.json。从 卡片模板开始;各字段在披露页面中说明。 - 正式运行: 用
--formal(8、16 或 32 张 GPU)训练已准入的配置,用python -m nexttransformer.audit_run审计,并用python -m nexttransformer.export_run导出。 - 结果 pull request: 运行记录、其审计、运行说明页,以及更新为
"status": "complete"并附最终检查点和训练日志链接的卡片。
每次提交 pull request 之前,运行 python -m nexttransformer.arena --check 和 python -m nexttransformer.export_run --check。如实报告:失败的启动、训练过的每个候选,以及人提供了多少帮助。
一个输给基线但完整披露的参赛作品,也是有价值的结果。