The Majority is not always right:RL training for solution aggregation
很像Learning to Refine:Self-Refinement of Parallel Reasoning in LLMs和Parallel-R1:Towards Parallel Thinking via Reinforcement Learning。这篇是通过rl来训练总结的能力。
论文深度解读
这篇由 FAIR at Meta 和 CMU 的研究者们发表的论文,直面了当前大语言模型(LLMs)在解决复杂推理任务时的一个核心挑战。长久以来,为了提升模型在数学、编程等高难度任务上的表现,学术界和工业界普遍采用一种测试时计算扩展(test-time compute scaling)的策略。这种策略的本质很简单:与其让模型只生成一次答案,不如让它独立生成多个(例如100个)候选解决方案,然后通过某种机制从中选出最好的一个。最流行、最简单直接的机制就是多数投票(majority voting),也被称为自洽性(Self-Consistency)。它的逻辑是“真理掌握在多数人手中”,出现次数最多的答案最可能是正确答案。
然而,本文的标题一针见血地指出了该方法的局限性——“多数并非总是正确”。论文作者观察到,在许多困难问题上,正确的解决方案可能非常精巧或不寻常,导致模型只有很小的概率能生成它。在这种情况下,正确的答案往往是“少数派”,而被淹没在大量相似但错误的“多数派”答案中。此外,多数投票机制完全忽略了那些虽然最终答案错误,但中间步骤包含正确推理的候选方案,错失了“取其精华、去其糟粕”的机会。
基于这一洞察,论文提出了一个全新的范式:将“答案聚合”本身视为一种需要学习的推理技能,而非一个固定的、基于规则的启发式方法。他们设计的核心方法名为 AGGLM (Aggregation Language Model)。其工作流程如下:
- 生成候选方案:首先,使用一个基础的LLM(论文中称为 solution model)针对一个问题(如一道数学题)生成多个独立的候选解决方案。
- 聚合与综合:然后,将问题描述和所有这些候选方案一同输入给另一个专门训练过的模型,即 AGGLM。AGGLM 的任务不是简单地从中挑选一个,而是被指令去“审阅、核对、并综合(review, reconcile, and synthesize)”这些方案,最终生成一个全新的、质量更高的最终答案。
- 通过强化学习进行训练:这套方法的精髓在于 AGGLM
是如何被训练的。作者采用了基于可验证奖励的强化学习(Reinforcement
Learning from Verifiable Rewards,
RLVR)。在数学这类任务中,“奖励”的定义非常明确:最终答案是否与标准答案一致。如果
AGGLM
综合出的答案是正确的,它就得到一个正奖励(+1),否则就是零奖励。通过这种方式,AGGLM
在训练中不断试错,逐渐学会如何识别并整合不同方案中的正确部分,修正错误,甚至在所有候选方案都错误的情况下,通过汲取它们的“思想火花”来构建出一个全新的正确解法。正如论文附录中的
Example A所示,即使所有输入解法都错了,AGGLM 也能综合出一个正确答案。
Example A: No correct solutions were included in the prompt; a new, correct solution synthesized by AggLM-1.7B.
Problem: … [Multiple incorrect candidate solutions are provided] AggLM-1.7B Output: … ### Step 1: Understand the Inequalities We simplify the inequalities using the plane equation x + y + z = 75… … a + b = 507 + 3 = 510
一个尤为关键的训练细节是,作者精心设计了训练数据的构成。他们将训练样本分为两类:“简单样本(easy examples)”(多数投票结果正确的样本)和“困难样本(hard examples)”(多数投票结果错误的样本,即正确答案是少数派)。实验发现,只用困难样本训练会导致模型过于激进,而在只用简单样本训练则无法学会处理关键的少数派正确情况。最终,他们通过混合所有困难样本和一部分(50%)简单样本,达到了最佳的平衡,使得模型既能处理常规问题,又擅长攻克难题。
在实验验证部分,论文的论证极为扎实。他们使用了四个极具挑战性的数学竞赛数据集(AIME24,
AIME25, HMMT24, HMMT25)进行评测。结果显示,他们训练的
AggLM-1.7B
模型在所有数据集上都显著超越了所有基线方法。例如,在 AIME25
数据集上,基础模型 Qwen3-1.7B 的 pass@1
准确率(单次生成正确率)为 35.68%,传统的多数投票方法能将其提升到
45.89%,而 AggLM-1.7B 则达到了惊人的
50.00%。这不仅证明了方法的有效性,更体现了其卓越的效率——AggLM-1.7B
在仅聚合8个候选解时的表现,甚至优于多数投票聚合16个候选解的表现(见论文
Figure 3),这意味着它能用更少的计算成本达到更高的准确率。
此外,论文还展示了 AggLM-1.7B
强大的泛化能力。即便它只在 1.7B
模型产生的答案上进行训练,当把它用于聚合一个更强大的 8B
模型产生的答案时,它依然是表现最好的方法。这说明 AGGLM
学到的是一种通用的、跨模型的答案聚合与推理能力,而非仅仅记住了特定模型的输出模式。
总而言之,这篇论文的核心贡献在于,它将答案聚合从一个简单的后处理步骤,提升到了一个可以通过学习获得的、核心的推理能力层面。它不仅提出了一个效果卓越、计算高效、泛化能力强的 AGGLM 模型,更重要的是,它为如何突破现有LLM性能瓶颈提供了一个全新的、极具潜力的思路:与其盲目扩大模型规模,不如教会模型如何更“聪明”地利用和整合自己已有的知识。
接下来,我将按照您提出的六个问题,进行更详细的剖析。
### 1. 论文的研究目标与实际意义
研究目标: 本文的核心研究目标是设计并验证一种新的方法,用于有效聚合大语言模型生成的多个候选解决方案,从而在复杂的推理任务上取得比现有方法(如多数投票)更高的准确率。它旨在将聚合过程本身从一个固定的启发式规则,转变为一个通过强化学习训练出来的动态推理技能。
解决的实际问题: 该研究直接解决了在实际应用中如何最大化“测试时计算”投入回报的问题。目前,为了让LLM解决一个难题,我们常常会让它“多想几次”(即生成多个答案),这会消耗大量的计算资源。但如果最后聚合答案的方法很原始(如多数投票),就可能导致:
- 浪费正确信息:一个正确的“少数派”答案被错误的“多数派”答案覆盖。
- 无法组合优点:多个答案可能各自只对了一部分,但投票机制无法将它们的正确部分拼接成一个完整的正确答案。 AGGLM方法正是为了解决这种资源利用效率低下的问题,确保每一次额外的计算投入都能被更智能地转化为性能的提升。
行业重要意义: 这项研究对于整个大模型行业具有重要的战略意义。随着模型规模越来越大,训练成本呈指数级增长。因此,如何在不增加模型参数的情况下,通过更优的推理时策略(inference-time strategy)来压榨模型性能,成为一个极具性价比的技术方向。AGGLM展示了一条非常有前景的路径:即训练一个相对较小的“专家”模型(聚合器),来指挥和整合一个或多个大型“生成器”模型的工作成果。这为模型部署和优化提供了新的思路,即通过“智能调度”而非“暴力堆算力”来提升系统上限。
### 2. 论文提出的新思路、方法与优势
- 新思路与方法:
- 聚合即推理(Aggregation as Reasoning):这是本文最核心的思路转变。它不再将聚合看作是一个简单的选择题(从A、B、C中选一个),而是看作一个开放式的问答题:给你这些材料(候选答案),请你综合、批判并给出一个最终的、最好的答案。
- AggLM 模型:一个专门用于执行上述“聚合推理”任务的语言模型。它的输入是“问题 + 一系列候选答案”,输出是一个经过综合提炼后的新答案。
- 基于可验证奖励的强化学习(RLVR):这是训练 AggLM 的关键技术。它利用任务本身具有明确正确答案的特性(如数学题),为模型的输出提供了一个清晰、客观的奖励信号。这比依赖人类反馈的RLHF成本更低、效率更高、扩展性更强。
- 平衡的训练数据策略:论文明确提出并验证了在训练中混合“简单”和“困难”样本的重要性。这是一个非常实用的工程经验,确保模型既不会在简单问题上“退化”,又具备解决核心难题的能力。
- 与之前方法的比较与优势:
对比多数投票/自洽性: > But, as we will see, majority-based aggregation can overlook valuable minority solutions: correct answers are sometimes assigned low probability under the model due to modeling errors. 优势在于,AGGLM 能够发现并采纳“少数派”的正确答案,并且能通过识别和组合不同方案中的正确推理步骤来合成一个全新的答案,这是投票机制完全无法做到的。
对比基于奖励模型的选择(Best-of-N): 一些方法会训练一个奖励模型(Reward Model, RM)来给每个候选答案打分,然后选择分数最高的那个。 > This recombines frequency with a learned notion of quality, but it sometimes introduces more regression errors than improvements. 优势在于,AGGLM 不仅仅是“选择者”,更是“创造者”。奖励模型只能在现有答案中进行排序,其天花板受限于生成的所有答案中最好的那一个。而 AGGLM 则有可能创造出比所有原始候选方案都更好的答案。此外,训练一个能精确评估复杂推理过程的奖励模型本身就极其困难。
对比简单的提示聚合(Prompted Aggregation): 即用一个未经专门训练的通用大模型,通过提示词让它去聚合答案。 论文通过实验(见
Table 1,Qwen3-1.7BvsAggLM-1.7B)证明,虽然提示有一定效果,但经过RLVR专门训练的 AggLM 性能要强大得多。这说明聚合这种高级推理能力是需要通过针对性训练才能充分激发的。
### 3. 实验设计与结果分析
- 实验设计:
- 数据集:选用了 MathArena 基准中的四个现代数学竞赛数据集:AIME24, AIME25, HMMT24, HMMT25。这些数据集的特点是问题难度大、解法多样,且有唯一的数值答案,非常适合进行“可验证奖励”的实验。
- 评估协议(Protocol):为了保证结果的稳健性,作者设计了严谨的评估流程。对每个问题,独立采样128个解,然后将其分成16个组,每组8个解。对每一组,他们用不同的聚合方法进行处理,并计算 pass@1 准确率,最后将16组的结果平均,得到该问题的最终得分。这种分组求平均的方法可以有效降低随机性带来的误差。
- 基线模型(Baselines):实验设置了非常全面的基线,包括:
- 基础性能:
pass@1(原始模型单次生成) 和pass@8(8个答案中只要有1个正确就算对,代表理论上限)。 - 传统方法:
Majority Voting。 - 先进方法:
Best-of-N(使用强大的 AceMath 7B 和 72B 奖励模型) 和Weighted Majority(结合奖励模型分数的加权投票)。 - 消融研究:
Prompted Aggregation(使用未训练的 Qwen3-1.7B 模型进行聚合,以验证训练的必要性)。
- 基础性能:
- 实验数据与结果: 实验结果有力地证明了 AGGLM
的优越性。以下是
Table 1中,在与训练数据同分布的Qwen3-1.7B模型上的核心数据摘要:
| 聚合模型 (Aggregation Model) | AIME24 | AIME25 | HMMT24 | HMMT25 |
|---|---|---|---|---|
| 基线 (Baselines) | ||||
| pass@1 (无聚合) | 50.91 | 35.68 | 22.45 | 22.84 |
| 多数投票 (Majority Voting) | 67.92 | 45.89 | 29.01 | 26.72 |
| 我们的方法 | ||||
| RL训练的聚合 (AggLM-1.7B) | 70.69 | 50.00 | 33.34 | 32.07 |
从上表可以清晰地看到:
1. **显著超越**:在所有四个数据集上,`AggLM-1.7B` 的性能均**全方位超越**包括多数投票和基于强大奖励模型的选择在内的所有基线方法。
2. **提升巨大**:相较于最常用的多数投票,`AggLM-1.7B` 带来了 3 到 5 个百分点的绝对提升,这在极具挑战性的数学竞赛任务上是非常显著的成果。
3. **泛化性强**:`Table 2` 的数据显示,当聚合更强的 `Qwen3-8B` 模型生成的解时,`AggLM-1.7B` 依然是表现最好的方法,证明了其学习到的聚合能力是可迁移的。
### 4. 未来研究方向、挑战与机会
- 值得探索的问题与挑战:
- 推广到非可验证领域:本文最大的成功依赖于数学任务有明确的“对/错”之分。未来最大的挑战是如何将
RLVR 的思想推广到奖励模糊或主观的领域,例如:
- 代码生成:如何定义一段代码是“好”的?(可能需要结合编译通过、单元测试、效率、可读性等多个指标)
- 长文写作:如何评估一篇综合多个来源的文章的质量?(可能需要训练一个更复杂的、基于 AI 或人类偏好的奖励模型)
- 多模态任务:如何聚合生成的图像或视频?
- 聚合器与生成器的协同进化:本文中生成器是固定的。一个更高级的系统可以是,聚合器在聚合过程中发现的常见错误或优秀推理模式,可以被蒸馏(distill)回生成器模型中,从而让生成器本身也变得更强大,形成一个良性循环。
- 动态候选集大小:本文固定了聚合的候选解数量(如k=8)。未来的研究可以探索让模型动态决定需要生成多少个候选解,当置信度高时少生成,置信度低时多生成,以进一步优化计算效率。
- 推广到非可验证领域:本文最大的成功依赖于数学任务有明确的“对/错”之分。未来最大的挑战是如何将
RLVR 的思想推广到奖励模糊或主观的领域,例如:
- 新的技术和投资机会:
- 推理即服务(Inference-as-a-Service)的增值层:云服务商或AI公司可以提供一种“性能增强”服务。客户使用标准的大模型API,但可以选择开启“聚合推理”模式,在后端通过类似AGGLM的机制,以稍高的延迟和成本换取显著提升的准确率。
- 面向特定领域的“超级优化器”:可以针对金融、法律、医疗等垂直领域,训练专门的聚合模型。这些模型不仅能提升答案准确性,甚至可以确保输出内容符合行业规范和逻辑,成为专业人士的得力助手。
- AI Agent 框架的决策核心:在复杂的 Agent 系统中,一个任务可能被分解给多个子 Agent 并行处理。AGGLM 的思想可以被用来设计一个“总指挥”Agent,负责整合各个子 Agent 的方案,做出最终的、最鲁棒的决策。
### 5. 批判性视角:论文的不足与缺失
尽管这是一篇非常出色的论文,但从批判的角度看,仍存在一些可以探讨的局限性:
- 对可验证奖励的强依赖:如前所述,这是该方法最大的限制。它在数学、部分科学计算和有严格测试用例的编程任务上表现优异,但其向更广泛的开放式、创造性任务的可迁移性并未被证明,仍是一个开放问题。
- 训练成本与复杂性:虽然推理时可能更高效,但训练 AGGLM 本身需要一个复杂的 RL 流程,包括大量样本生成、奖励计算和策略优化。论文没有详细讨论训练这样一个聚合器需要多大的计算开销和时间成本。
- 聚合器规模的探讨不足:实验中,生成器和聚合器都是 1.7B 规模。一个有趣的问题是:一个小规模的聚合器(如1.7B)能否有效聚合一个超大规模模型(如100B)的输出?还是说聚合器必须与生成器规模相当才能充分理解其推理过程?这方面的 scaling law 值得进一步研究。
- “可解释性”问题:AGGLM 是一个黑箱。虽然它能给出好答案,但我们很难确切知道它是如何“取舍”和“组合”输入信息的。如果它合成了一个看似正确但实际上有微小错误的答案,其错误的根源将难以追溯。
### 6. 核心启发与知识补充
- 我应该从这篇论文中重点学什么?
- 范式转变的思维:最大的启发是“将后处理步骤看作可学习的关键模型组件”。在你的工作中,任何你使用启发式规则(hard-coded rules)来处理模型输出的地方,都可以思考一下:这个规则能否被一个小的、专门训练的模型所替代?
- 数据驱动的解决方案:AGGLM 的成功很大程度上归功于其精巧的数据策略(平衡简单/困难样本)。这提醒我们,在很多时候,智能地构建和平衡训练数据,其效果可能比调整模型架构或优化算法更显著。
- “小模型撬动大模型”的杠杆作用:你不一定总需要一个巨大的、端到端的模型。一个“大模型(生成器)+ 小模型(协调器/聚合器)”的组合架构,可能在成本和效益上达到更好的平衡。
- 我还需要补充了解哪些背景知识?
为了完全理解并应用这篇论文的思想,建议您了解以下背景知识:
- 思维链(Chain-of-Thought,
CoT)与自洽性(Self-Consistency):这是本文所要改进的基础技术。理解它们的工作原理和局限性,才能更好地体会
AGGLM 的创新之处。可以阅读 Google 的原始论文
Chain-of-Thought Prompting Elicits Reasoning in Large Language Models和Self-Consistency Improves Chain of Thought Reasoning in Large Language Models(https://arxiv.org/abs/2203.11171)。 - 强化学习基础与 PPO/GRPO 算法:理解强化学习的基本概念(策略、奖励、价值函数)和近端策略优化(PPO)算法,因为它是当前训练LLM最主流的RL算法之一。本文使用的 GRPO (Group-Relative Policy Optimization) 是其变种。
- 奖励模型(Reward Modeling):了解在 RLHF(基于人类反馈的强化学习)中,奖励模型是如何被训练和使用的。这将帮助你理解 AGGLM 与传统 Best-of-N 方法的根本区别。
- 思维链(Chain-of-Thought,
CoT)与自洽性(Self-Consistency):这是本文所要改进的基础技术。理解它们的工作原理和局限性,才能更好地体会
AGGLM 的创新之处。可以阅读 Google 的原始论文
希望这份详尽的解读能帮助您深入理解这篇论文的精髓,并从中获得有价值的启发。