SuperAPI AI 工具集

nanoGRPO

nanoGRPO 是 Group Relative Policy Optimization(GRPO)的轻量级实现,面向语言模型训练。按照仓库说明,用户可以利用它训练自己的“tiny r1/o1”模型;作者给出了在 8GB RTX 4060 上训练 120 步后的结果。其核心公式为:J_GRPO(θ) = (1/G) Σ_{i=1}^{G} (min(π_θ(o_i|q)/π_θ_old(o_i|q) · A_i, clip(π_θ(o_i|q)/π_θ_old(o_i|q), 1-ε, 1+ε) · A_i))。该仓库指出,GRPO 可理解为使用 grouped output 的 z score 作为优势估计的 PPO,而不使用 value function,并提供了原始论文和 r1 论文链接。

访问官网

工具介绍

nanoGRPO 是 Group Relative Policy Optimization(GRPO)的轻量级实现,主要用于语言模型的策略优化训练。

用途

根据项目描述,nanoGRPO 用于语言模型训练,并强调用户可以“cooked you own tiny r1/o1”,即利用该项目训练自己的小型 r1/o1 类模型。它提供了 GRPO 的轻量级实现,适合希望了解或在有限硬件条件下实践 GRPO 的开发者。

主要能力

项目给出了 GRPO 的目标函数:

$$ J_{GRPO}(\theta) = \frac{1}{G} \sum_{i=1}^{G} \left( \min \left( \frac{\pi_{\theta}(o_i|q)}{\pi_{\theta_{old}}(o_i|q)} A_i, \text{clip} \left( \frac{\pi_{\theta}(o_i|q)}{\pi_{\theta_{old}}(o_i|q)}, 1 - \epsilon, 1 + \epsilon \right) A_i \right) \right) $$

同时说明:GRPO 是 PPO 的一种形式,其优势(advantage)被估计为分组输出(grouped output)的 z score,而不是使用 value function。

使用方式

项目在标题中给出官方地址:https://github.com/joey00072/nanoGRPO。文本中未提供具体安装命令、运行步骤或代码示例,相关细节暂未提供。

适用场景与限制

项目提到在 8GB RTX 4060 上训练 120 步后得到结果,说明其可在较小显存条件下使用。但除该硬件与步数信息外,文本未提供数据集、模型规模、超参数、性能指标或与其他实现的对比,也未说明许可证、依赖和适用范围,这些内容暂未提供。