S3AC:用于大语言模型推理的稳定化、选择性且可扩展的演员-评论家方法

Hugging Face Daily Papers 论文

摘要

S3AC 提出一种稳定的 Actor-Critic 训练方法与难度感知提示采样(DAPS),在不依赖分组 rollout 的情况下改进大模型的长链推理,性能优于 GRPO-X 等现代基线,并可在 35B 模型上维持单 rollout 训练。

无需评论器的方法广泛用于面向长思维链(CoT)推理的可验证奖励强化学习(RLVR),避免了困难且成本高昂的评论器训练。常见的替代方法缺乏词元级信用分配,且通常依赖分组采样来估计提示特定的基线。评论器可以解决这两项局限,但长思维链评论器的训练受到两个问题的阻碍:对不同长度轨迹的覆盖不足,以及训练早期的梯度被新初始化的价值头主导。我们提出了用于大语言模型推理的 S3AC(稳定、选择性且可扩展的演员-评论器方法),其核心是一套实用的评论器训练方案。训练后的评论器提供词元特定的优势值和提示特定的基线。我们利用提示级基线估计难度,并据此提出难度感知提示采样(DAPS)。DAPS 将这一估计与历史结果相结合,在生成前选择提示,无需离线初始化难度即可减少采样浪费。S3AC 在参数规模从 15 亿到 3970 亿的稠密模型和混合专家模型上均提升了分组采样的性能,优于现代 GRPO 变体(GRPO-X)。在 350 亿参数规模下,它能够持续进行单次采样训练,而所测试的无需评论器的基线方法在初期取得提升后性能出现退化。
查看原文
查看缓存全文

缓存时间: 2026/10/28 01:59

论文页面 - S3AC:用于大语言模型推理的稳定、选择性且可扩展的演员-评论家方法

来源:https://huggingface.co/papers/2610.35518

摘要

S3AC 引入了一种稳定的演员-评论家训练方法,结合难度感知提示采样,无需依赖分组 rollout 即可提升大语言模型的长程推理能力。

在用于长思维链 (CoT) 推理的可验证奖励强化学习 (https://huggingface.co/papers?q=reinforcement%20learning%20with%20verifiable%20rewards)(RLVR) 中,无评论家方法被广泛采用,以避免困难且昂贵的评论家训练。常见的替代方法缺乏词元级信用分配 (https://huggingface.co/papers?q=token-level%20credit%20assignment),且通常依赖分组 rollout 来估计特定提示的基线。评论家可以解决这两个局限,但长思维链的评论家训练受到两方面阻碍:对不同长度轨迹的覆盖不足,以及早期梯度被新初始化的价值头 (https://huggingface.co/papers?q=value%20head) 主导。我们提出用于大语言模型推理的 S3AC(稳定、选择性且可扩展的演员-评论家 (https://huggingface.co/papers?q=Actor-Critic)),其核心是一套实用的评论家训练方案。训练后的评论家提供词元特定的优势值和提示特定的基线。利用提示级基线估计难度,我们提出了难度感知提示采样 (https://huggingface.co/papers?q=Difficulty-Aware%20Prompt%20Sampling)(DAPS)。DAPS 将这一估计与历史结果相结合,在生成前选择提示,无需离线难度初始化即可减少 rollout 浪费。S3AC 在参数量从 1.5B 到 397B 的稠密模型和混合专家 (https://huggingface.co/papers?q=mixture-of-experts)模型上均提升了分组 rollout 的性能,优于一种现代 GRPO (https://huggingface.co/papers?q=GRPO) 变体 (GRPO (https://huggingface.co/papers?q=GRPO)-X)。在 35B 参数规模下,它能够持续进行单次 rollout 训练,而受测的无评论家基线在初期取得提升后性能出现下降。

查看 arXiv 页面 (https://arxiv.org/abs/2610.35518)查看 PDF (https://arxiv.org/pdf/2610.35518)GitHub5.22k (https://github.com/InternLM/xtuner)添加到收藏集 (https://huggingface.co/login?next=%2Fpapers%2F2610.35518)

在你的智能体中获取这篇论文:

hf papers read 2610\.35518

没有最新版 CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash

引用这篇论文的模型0

没有模型链接到这篇论文

在模型的 README.md 中引用 arxiv.org/abs/2610.35518,即可从本页面链接到该模型。

引用这篇论文的数据集0

没有数据集链接到这篇论文

在数据集的 README.md 中引用 arxiv.org/abs/2610.35518,即可从本页面链接到该数据集。

引用这篇论文的 Space0

没有 Space 链接到这篇论文

在 Space 的 README.md 中引用 arxiv.org/abs/2610.35518,即可从本页面链接到该 Space。

包含这篇论文的收藏集0

没有收藏集包含这篇论文

将这篇论文添加到收藏集 (https://huggingface.co/new-collection),即可从本页面链接到该收藏集。

相似文章

Z.ai的稳定异步强化学习(13分钟阅读)

TLDR AI

本文介绍了单rollout异步优化(SAO)技术,用于解决LLM后训练中异步RL的稳定性和异策略挑战,并证明SAO在智能体编码和推理基准测试上持续优于GRPO。