SLM 如何在推理任务中击败大型模型

传统上,LLMs 的进步主要依赖于增加训练时间的计算量,即训练更大的模型。然而,这种方法成本昂贵,资源需求巨大。扩展测试时间计算提供了一种有效的替代方案,允许模型在推理过程中“思考更长时间”,从而在不增加模型参数的情况下提升性能。

文章主要介绍了三种测试时间计算扩展的策略:

  1. Best-of-N 方法:生成多个候选答案,使用奖励模型(reward model)对它们进行评分,选择得分最高的答案。加权的 Best-of-N 变体还考虑了答案的出现频率,优先选择高质量且高频率的答案。

  2. 束搜索(Beam Search):利用过程奖励模型(Process Reward Model,PRM)在生成过程中逐步引导模型,选择每一步最有可能通向正确答案的路径。相比于 Best-of-N,束搜索在相同计算预算下取得了更高的准确率。

  3. 多样化验证器树搜索(Diverse Verifier Tree Search,DVTS):这是对束搜索的改进,旨在增加生成答案的多样性。DVTS 将初始束分为多个独立的子树,在较大的计算预算下表现出色,尤其是在处理较简单的问题时。

    通过一系列实验,使用开源的 Llama 模型和 MATH-500 数据集,验证了这些方法的有效性。结果显示,即使是参数量较小的模型(如 1B 和 3B 的 Llama Instruct 模型),在采用适当的测试时间计算策略后,其性能可以超过更大的模型(如 8B 和 70B 的模型)。

    不同策略在不同问题难度和计算预算下的表现是不同的,参考“计算最优扩展”的概念,即针对特定的计算预算,选择能达到最佳性能的策略。对于简单问题和较低的计算预算,Best-of-N 表现更好;而对于复杂问题和较高的计算预算,束搜索和 DVTS 更具优势。

    在未来,提升验证器的质量,实现模型的自验证,融入更深入的推理过程,以及将搜索方法用于数据生成等。这些方向都有望进一步提升 LLMs 的性能,特别是在资源受限的情况下。

一、几种解码技术的区别

在这里插入图片描述

1. 贪婪解码(Greedy Decoding)

怎么工作:

  • 在每一步,选择概率最高的下一个词或标记。
  • 生成一个单一的、最有可能的序列,直到结束标记或达到最大长度。

特点:

  • 快速简单:计算效率高,适合实时应用。
  • 缺点
    局部最优:只关注当前步骤的最优选择,可能错过全局更优的解决方案。
    缺乏多样性:生成的序列缺少变化,可能导致重复或不自然的输出。

2. 束搜索(Beam Search)

怎么工作:

  • 保留多个候选序列:在每一步,保留固定数量(束宽度为 k)的最有可能的部分序列。
  • 扩展候选序列:对每个部分序列,生成可能的下一个词,用于扩展序列。
  • 选择最佳候选:计算每个新序列的累计概率,保留累计概率最高的 k 个序列,继续下一步扩展。
  • 重复上述步骤,直到生成结束标记或达到最大长度。

特点:

  • 平衡了精度和计算成本:相比贪婪解码,可以找到更接近全局最优的序列。
  • 提供一定的多样性:保留了多个候选序列,但多样性受束宽度限制。
  • 缺点
    计算量增加:束宽度越大,计算成本越高。
    可能仍受限于束宽度:无法保证找到全局最优解。

3. 多样化验证器树搜索(DVTS)

怎么工作:

  • 分裂初始束:将初始束分成多个独立的子树,增加初始解的多样性。
  • 使用验证器指导:在每个子树中,使用**过程奖励模型(Process Reward Model,PRM)**对生成的步骤进行评分。
  • 独立扩展子树:每个子树独立地进行扩展,按照验证器的反馈选择最有希望的路径。
  • 组合结果:最终从各个子树中选取验证器评分最高的答案。

特点:

  • 增加多样性:通过独立的子树,探索更多可能的解答路径。
  • 利用验证器引导:在生成过程中实时评估,提升生成结果的准确性和质量。
  • 适用于复杂任务:在需要深度推理或精确回答的任务中表现出色。
  • 缺点
    计算成本高:由于需要管理更多的搜索路径和验证器的计算。
    实现复杂:需要精心设计搜索策略和验证器模型。

4. 多数投票(Majority Voting)

怎么工作:

  • 多次独立生成:使用随机采样(如 Top-k 或 Top-p 采样)等方法,生成多个(如 N 个)独立的候选答案。
  • 统计答案频率:对生成的候选答案进行统计,记录每个独特答案出现的次数。
  • 选择出现次数最多的答案:将频率最高的答案作为最终输出。

特点:

  • 简单易行:不需要复杂的模型调整或外部评估模型。
  • 提高稳定性:通过统计,提高答案的一致性和可靠性。
  • 减少随机性影响:平滑随机采样带来的波动,过滤掉偶然的错误答案。
  • 缺点
    无法保证正确性:如果模型本身倾向于错误答案,可能多数投票的结果也是错误的。
    计算成本增加:需要多次生成答案,增加了计算开销。
    多样性可能不足:如果生成的答案过于多样,可能没有明确的多数答案。

它们之间的区别与联系

搜索空间的大小

  • 贪婪解码
    最小搜索空间:只探索一条路径,即每一步都选择概率最高的词。

  • 束搜索
    中等搜索空间:根据束宽度 k,同时探索 k 条路径。

  • DVTS
    最大搜索空间:通过独立的子树和验证器的引导,探索更加广泛的路径。

  • 多数投票
    扩展搜索空间:通过多次独立生成,获得不同的候选答案,但每次生成仍是一条路径。

多样性

  • 贪婪解码
    最低多样性:只有一个输出序列。

  • 束搜索
    一定的多样性:受限于束宽度,提供多个候选序列。

  • DVTS
    高多样性:独立的子树和验证器的引导,增加了解答的多样性。

  • 多数投票
    中等多样性:依赖于生成次数和随机采样的程度,可能获得多个不同的候选答案。

计算成本

  • 贪婪解码
    最低计算成本:每一步只需计算概率最高的词。

  • 束搜索
    中等计算成本:取决于束宽度 k,计算量随 k 增加而增加。

  • DVTS
    最高计算成本:需要管理多个子树、维护验证器的计算和评分。

  • 多数投票
    中等到高的计算成本:取决于生成的次数 N,生成次数越多,计算成本越高。

适用的场景和任务

贪婪解码

  • 快速生成结果:适用于对速度要求高、对结果质量要求不高的简单任务。
  • 示例:实时对话系统的快速回复。

束搜索

  • 平衡质量和效率:适用于需要一定精度,但计算资源有限的任务。
  • 示例:机器翻译、文本摘要。

DVTS

  • 复杂、需要深入推理的任务:适用于有足够计算资源,希望获得高质量答案的场景。
  • 示例:数学问题求解、代码生成、复杂问答。

多数投票

  • 提高答案稳定性:适用于需要减少随机性影响、希望获得一致性答案的任务。
  • 示例:知识问答、关键事实的确认。

简单类比

  • 贪婪解码
    类比:像在每个十字路口都选择看起来最直接的道路,可能错过更好的路线。

  • 束搜索
    类比:像在每个十字路口选择几条看起来不错的道路,走一段后再根据情况选择最佳路线。

  • DVTS
    类比:像派出多个探索队伍,从不同的路线出发,同时使用指南(验证器)引导,每个队伍独立寻找最佳路径。

  • 多数投票
    类比:像询问多个人同一个问题,然后选择大多数人都同意的答案。

总结

通过将多数投票加入讨论,我们可以更全面地了解不同的解码和生成策略的特点、优点和适用场景。每种方法都有其独特的优势和局限性,选择适当的方法需要根据具体任务的需求、可用的计算资源以及对结果质量的要求来决定。

  • 贪婪解码适合简单、快速、对质量要求不高的任务。

  • 束搜索在质量和效率之间取得平衡,适用于一般复杂度的任务。

  • 多样化验证器树搜索(DVTS) 适用于复杂、需要高准确性的任务,但计算成本高。

  • 多数投票通过多次生成和统计,减少了随机性,提高了答案的稳定性,但无法保证一定正确,需要谨慎使用。

    选择合适的策略,有助于充分利用大型语言模型的能力,满足各种应用场景的需求。

二、 train-time compute的变革

在过去的几年里,训练时间计算(train-time compute)的扩展主导了大型语言模型(LLMs)的进展。尽管这种模式已被证明异常有效,但预训练更大模型所需的资源变得昂贵得令人望而却步,数十亿美元规模的集群已在眼前。这个趋势引发了对一种互补方法的极大兴趣:测试时间计算扩展(test-time compute scaling)。测试时间方法并不依赖于越来越大的预训练预算,而是使用动态推理策略,让模型在解决更难的问题时“思考更长时间”。一个突出的例子是 OpenAI 的 o1 模型,它在增加测试时间计算量时,对困难的数学问题表现出持续的改进:

DeepMind 的最新研究表明,可以通过迭代自我改进(iterative self-refinement)或使用奖励模型(reward model)在解空间中进行搜索等策略,来最优地扩展测试时间计算。通过为每个提示自适应地分配测试时间计算,小模型可以与更大、更耗资源的模型媲美,甚至有时还能超越它们。当内存受限且可用硬件不足以运行更大的模型时,扩展测试时间计算尤其有利。然而,这种有前景的方法是使用闭源模型演示的,并未发布任何实现细节或代码。

  • 计算最优扩展(Compute-optimal scaling):实现 DeepMind 的方案,在测试时间提升开源模型的数学能力。

  • 多样化验证器树搜索(Diverse Verifier Tree Search,DVTS):开发的验证器引导树搜索(verifier-guided tree search)技术的未发表扩展。这种简单而有效的方法提高了多样性,特别是在较高的测试时间计算预算下,提供了更好的性能。

  • ** 搜索与学习(Search and Learn)**:一个用于在大型语言模型上实现搜索策略的轻量级工具包,使用 vLLM 构建,速度极快。

那么,计算最优扩展在实践中效果如何呢?看看这个图表,在具有挑战性的 MATH-500 基准上,如果给予足够的“思考时间”,微小的 1B 和 3B Llama Instruct 模型竟然超过了它们更大的 8B 和 70B 同系列模型的表现 :

三、test-time compute scaling的策略

扩展测试时间计算有两种主要策略:

  1. 自我改进(Self-Refinement):模型通过在后续迭代中识别和纠正错误,迭代地改进它们自己的输出或“想法”。虽然在某些任务上有效,但这种策略通常要求模型具有内置的自我改进机制,这可能会限制其适用性。

  2. 针对验证器的搜索(Search Against a Verifier):这种方法专注于生成多个候选答案,并使用验证器(verifier)选择最佳答案。验证器可以是从硬编码的启发式方法到学习的奖励模型(reward model),后面我们将重点关注学习的验证器。它包括 Best-of-N 采样和树搜索等技术。搜索策略更灵活,可以适应问题的难度,尽管它们的性能受限于验证器的质量。

四、验证器到底是什么?

验证器通常是一个奖励模型(Reward Model,RM)过程奖励模型(Process Reward Model,PRM),它们被训练来对生成的内容进行评估。

  • 定义:在本文的上下文中,验证器是一个辅助模型或机制,用于评估和评分大型语言模型(LLM)生成的输出或部分输出

  • 作用

  • 评估质量:验证器对生成的候选答案或中间步骤进行评估,判断其正确性、可信度或质量。

  • 指导搜索:通过对候选进行评分,验证器帮助模型在生成过程中选择更有可能通向正确答案的路径

Best-of-N、束搜索(Beam Search)和多样化验证器树搜索(DVTS)都是解码技术,它们被用来指导大型语言模型(LLM)在生成过程中如何产生输出。将这些解码技术与验证器(Verifier)结合,以优化模型在测试时间的性能。

  • 提高准确性:验证器有助于过滤掉错误或低质量的生成结果,提高模型输出的准确性,减少幻觉的产生。

  • 优化生成过程:在生成过程中实时评估和引导,使模型更有效地探索解答空间,尤其在复杂任务上表现出色。

  • 增强小模型的性能:通过结合验证器,较小的模型也可以在特定任务上达到或超过大模型的性能。

五、效果验证

实验设置

如上图所示,我们的实验设置涉及以下步骤的流程:

  1. 我们首先将一个数学问题输入大型语言模型(LLM),它生成 N 个部分解,例如推导中的一个中间步骤。

  2. 每个步骤由过程奖励模型(PRM)评分,它估计每个步骤最终到达正确最终答案的概率。

  3. 这些步骤和 PRM 分数然后被给定的搜索策略使用,以选择哪些部分解应被进一步探索,以生成下一轮中间步骤。

  4. 一旦搜索策略终止,最终的候选解由 PRM 排序,产生最终答案。

    为了比较各种搜索策略,我们使用了以下开源模型和数据集:

  • 模型:我们使用了 meta-llama/Llama-3.2-1B-Instruct 作为扩展测试时间计算的主要模型。由于其 10 亿参数的轻量级特性,可以实现快速迭代,其在数学基准测试中的未饱和性能使其成为突出扩展优势的理想选择。

  • 过程奖励模型(PRM):为了指导我们的搜索策略,我们使用了 RLHFlow/Llama3.1-8B-PRM-Deepseek-Data,这是一个使用过程监督(process supervision)训练的 80 亿参数的奖励模型。过程监督是一种训练方法,模型在推理过程的每个步骤(而不仅仅是最终结果)都能收到反馈。我们选择这个模型是因为它属于与我们的策略相同的模型家族,并且在我们测试的这个参数量级中,比我们测试的其他 PRM(如 Math-Shepherd)给出了更好的结果。

  • 数据集:我们在 MATH 基准的 MATH-500 子集上进行了评估,这是 OpenAI 作为其过程监督研究的一部分发布的数据集。这些数学问题涵盖了七个学科,对人类和大多数大型语言模型来说都具有挑战性。看看下面的数据集浏览器,感受一下问题的难度吧!我们在从每个提示生成 1 到 256 个生成的计算预算上测试了每个搜索策略,并使用五个随机种子运行数据生成流程,以估计运行间的方差。你可以在这个集合中找到我们分析的模型和数据集。

为了热身,我们将从一个简单的基线开始,并逐步加入其他技术来提高性能。

多数投票:一个简单的基线

多数投票——或者如果你想用更高端的说法,可以称为自一致性解码(self-consistency decoding)——是聚合大型语言模型输出的最直接方法。顾名思义,对于给定的数学问题,我们生成 N 个候选解并选择最频繁的答案。在我们所有的实验中,我们以温度 T = 0.8 采样了最多 N = 256 个候选解,每个问题生成最多 2048 个标记。

MATH 基准有一个独特之处,即答案必须以 LaTeX 盒子的形式格式化,如 \boxed{answer}。我们最初为 Llama 3.2 1B 尝试了以下简单的系统提示:

请逐步思考,并将你的最终答案放在 \boxed{} 中。

但发现使用贪婪解码(T = 0)得到的准确率远低于 Meta 在其发布中报告的 30.6%。幸运的是,Meta 也发布了他们用于评估的提示,切换我们的系统提示到他们的后,效果发生了巨大变化:

高效而清晰地解决以下数学问题:
  
- 对于简单的问题(2 步或更少):
  提供简洁的解答,尽量减少解释。
  
- 对于复杂的问题(3 步或更多):
  使用以下的逐步格式:
  
  ## 第 1 步:[简洁的描述]  
  [简短的解释和计算]  
  
  ## 第 2 步:[简洁的描述]  
  [简短的解释和计算]  
  
  ...  
  
无论采用何种方法,总是以以下内容结束:
  
因此,最终答案是:$\boxed{answer}$。我希望这是正确的。
  
其中 [answer] 是解决问题的最终数字或表达式。

评估数学问题的答案有一个细微之处,即像 1/33/3 这样的字符串是不同的,但代表数学上等价的答案。处理这种情况的标准方法是将一对答案转换为 SymPy 对象,然后检查减去两个对象并应用 sympy.simplify 是否得到零。

虽然这种方法在比较少量候选答案时效果很好,但我们发现当在一个包含 N 个候选答案的列表中比较许多对时,非常慢;在某些情况下,比最初生成候选答案还要慢!为了解决这个问题,我们首先将每个答案简化为其规范形式,然后计算每种形式的频率来确定多数投票。如果你对代码如何实现感兴趣,可以展开下面的细节。

实现细节

这里是将多数投票应用于 Llama 3.2 1B Instruct 的生成时的表现:

结果表明,多数投票相比于贪婪解码基线的确带来了显著的改进,但其增益在大约 N = 64 代之后开始趋于平稳。这种限制的出现是因为多数投票在需要细致推理或错误在代际间一致的任务上表现不佳。如果你也想知道为什么当 N = 1 和 2 时,多数投票的准确率比零次提示链式思考(0-shot CoT)基线更差,那是因为我们以 T = 0.8 进行采样,这使得在少数候选中产生正确答案的可能性较小。

基于多数投票的局限性,让我们看看引入奖励模型如何提升性能。

Best-of-N

Best-of-N 是多数投票的一个简单但有效的扩展,它使用奖励模型来确定最可能的答案。该方法有两种主要变体:

  1. 原始的 Best-of-N:生成 N 个独立的回复,选择奖励模型(RM)得分最高的作为最终答案。这确保选择最自信的单个回复,但不考虑答案之间的一致性。

  2. 加权的 Best-of-N:汇总所有相同回复的分数,选择总奖励最高的答案。这种方法通过重复出现来提升分数,优先考虑高质量的答案。数学上,对答案 (a_i) 的加权如下:

    [
    a_{\text{weighted}} = \arg\max_{a} \sum_{i=1}^{N} I(a_i = a) \cdot RM(p, s_i),
    ]

    其中 (RM(p, s_i)) 是问题 § 的第 (i) 个解 (s_i) 的奖励模型得分。

    通常,人们使用结果奖励模型(ORM)来获得单个解决方案级别的得分。但为了与后面讨论的其他搜索策略进行公平比较,我们将使用相同的 PRM 来对 Best-of-N 的解决方案进行评分。如下面所示,PRM 对每个解决方案产生一个累积的步骤级别的得分序列,因此我们需要对步骤进行归约以获得单个解决方案级别的得分:


    在文献中,最常见的归约方法如下:

  • 最小值(Min):使用所有步骤中的最小得分。

  • 乘积(Prod):使用步骤级别得分的乘积。

  • 最后(Last):使用步骤中的最终得分。这个得分包含了所有先前步骤的累积信息,因此有效地将 PRM 视为能够对部分解进行评分的 ORM。

    我们对每种归约方法进行了实验,发现在我们的任务和 PRM 选择中表现最佳。我们在所有实验中都使用了这种聚合,你可以展开下面的细节,看看我们如何实现它,以及上述的加权过程。

    这里是应用 Best-of-N 两种变体得到的结果:

结果显示了明显的优势:加权的 Best-of-N 在更大的生成预算下,一直优于原始的 Best-of-N。它能够对相同的回复汇总分数,确保即使是出现频率较低但质量更高的答案也能被有效地优先考虑。

然而,尽管有这些改进,我们仍然无法达到 Llama 8B 模型的性能,而且 Best-of-N 方法在 N = 256 代时开始趋于平稳。我们能否通过逐步监督搜索过程来进一步突破边界?

使用过程奖励模型的束搜索

束搜索(Beam search)是一种系统地探索解空间的结构化搜索方法,使其成为在测试时间改进模型输出的强大工具。当与过程奖励模型(PRM)结合使用时,束搜索可以同时优化问题解决中间步骤的生成和评估。其工作方式如下:

  1. 通过保持固定数量的“束”或活动路径 N,迭代地生成多个候选解。

  2. 在第一次迭代中,以温度 T 从 LLM 采样 N 个独立的步骤,以引入回复的多样性。这些步骤通常由停止条件定义,如在新行 \n 或双新行 \n\n 处终止。

  3. 使用 PRM 对每个步骤进行评分,选择前 N / M 个步骤作为下一轮生成的候选。在这里,M 表示给定活动路径的“束宽度”(beam width)。与 Best-of-N 一样,我们在每次迭代中使用“最后”归约来对部分解进行评分。

  4. 从步骤(3)中选定的节点生成 M 个新步骤,并选择 PRM 得分最高的步骤。

  5. 重复步骤(3)和(4),直到到达 EOS 标记或超过最大搜索深度。

    通过允许 PRM 评估中间步骤的正确性,束搜索可以在过程的早期识别和优先考虑有希望的路径。这种逐步评估对于像数学这样需要复杂推理的任务特别有益,其中验证部分解可以显著提高最终结果。

    实现细节

    在我们的实验中,我们遵循了 DeepMind 的超参数选择,使用以下设置运行束搜索:

  • 在计算扩展为 4、16、64、256 时,使用 N 个束

  • 固定束宽 M = 4

  • 以温度 T = 0.8 进行采样

  • 最多 40 次迭代,即最大深度为 40 步的树如下面所示,结果非常惊人:在测试时间预算为 N = 4 时,束搜索实现了与 Best-of-N 在 N = 16 时相同的准确率,即计算效率提高了 4 倍!此外,束搜索仅用每个问题 N = 32 个解就匹配了 Llama 3.1 8B 的性能。计算机科学博士生在 MATH 上的平均表现约为 40%,所以对于一个 10 亿参数的模型达到近 55% 并不算太差。

束搜索最擅长解决哪些问题?

虽然总体来看,束搜索显然是比 Best-of-N 或多数投票更好的搜索策略,但 DeepMind 的论文表明,每种策略都有取舍,取决于问题难度和测试时间计算预算。

为了了解哪种策略最适合哪些问题,DeepMind 计算了估计的问题难度分布,然后将结果分成五分位数。换句话说,每个问题被分配到 5 个级别之一,其中级别 1 表示较容易的问题,级别 5 表示最难的问题。为了估计问题难度,DeepMind 为每个问题以标准采样生成了 2048 个候选解,然后提出了以下启发式方法:

  • Oracle:使用真实标签来估计每个问题的 pass@1 得分。对 pass@1 得分的分布进行分箱以确定五分位数。

  • 模型:使用每个问题的平均 PRM 得分分布来确定五分位数。直观地说,较难的问题得分会较低。

    以下是在四个测试时间计算预算 N = [4, 16, 64, 256] 和 pass@1 得分下,各种方法的表现:

    在这个图中,每个柱状图表示一个测试时间计算预算,在每个柱状图内,我们显示每种方法的相对准确率。例如,在难度级别 2 的四个柱状图中,我们看到:

  • 多数投票在所有计算预算下都是表现最差的,除了 N = 256 时,束搜索最差。

  • 束搜索在 N = [4, 16, 64] 情况下最好,但在 N = 256 时,Best-of-N 最好。

    虽然我们看到束搜索在中等和困难问题(级别 3-5)中提供了持续的增益,但在较简单的问题上(尤其是在大的计算预算下),它的表现往往比 Best-of-N更差。

    通过查看束搜索生成的结果树,我们意识到,如果一个步骤被赋予高奖励,那么整个树就会收敛到该路径,从而影响多样性。这促使我们探索一种扩展束搜索的方法,以最大化多样性——让我们来看看!

DVTS:通过多样性提升性能

正如我们上面所见,束搜索在 Best-of-N 上表现出色,但在较简单的问题和较大的测试时间计算预算下往往表现不佳。为了解决这个问题,我们开发了一种扩展,称为多样化验证器树搜索(Diverse Verifier Tree Search,DVTS),旨在在较大的 N 值下最大化多样性。

DVTS 的工作方式与束搜索类似,但有以下修改:

  1. 对于给定的 N 和 M,将初始束集合扩展为 N / M 个独立的子树。

  2. 对于每个子树,选择 PRM 得分最高的步骤。

  3. 从步骤(2)中选定的节点生成 M 个新步骤,选择 PRM 得分最高的步骤。

  4. 重复步骤(3),直到到达 EOS 标记或最大树深度。

    以下是将 DVTS 应用于 Llama 1B 的结果:

如我们所见,DVTS 为束搜索提供了一种补充策略:在较小的 N 值下,束搜索更有效地找到正确的解,但在较大的 N 值下,DVTS 候选解的多样性开始发挥作用,我们获得了更好的性能。

我们还可以从问题难度的分解中看到这一点,DVTS 在大的 N 值下增强了在简单/中等问题上的性能,而束搜索在小的 N 值下在各种问题难度上表现最佳:

六、总结

总结:测试时计算扩展的最佳策略

  1. 计算最优的扩展策略(Compute-Optimal Scaling)

    核心思想:

    在给定的计算预算下,选择能够实现最佳性能的搜索方法和超参数组合。

    公式表示:

    θ*(N) = argmax_θ [ E_{y ∼ Target(θ, N, q)} [ 1_{y = y*(q)} ] ]

    挑战:直接计算 θ*(N) 较为困难。

    解决方案:

    DeepMind 提出了基于问题难度的近似方法,根据不同难度级别,确定最佳的搜索策略和计算资源分配:

  • 简单问题、低计算预算: 使用 Best-of-N 等简单方法。

  • 复杂问题、高计算预算: 使用束搜索(Beam Search)等高级方法。

  • θ*(N):给定计算预算 N 的最优参数和策略组合。

  • y*(q):问题 q 的真实答案。

  • θ:搜索方法和超参数的组合。

  1. 向更大模型的扩展

    目的: 探究计算最优策略在更大模型上的效果,以及过程奖励模型(PRM)在较大模型中是否仍然有益。

    发现:

  • 计算最优扩展策略效果显著。

  • 即使在大型模型上,使用计算最优策略的较小模型(如 Llama 13B)性能可超过更大的模型(如 Llama 2 70B Instruct)。

  1. 未来方向和挑战
  • 增强验证器的能力:提升验证器的鲁棒性和泛化能力,对于改进模型性能至关重要。

  • 实现自我验证(Self-Verification):使模型能够自主验证输出,提高可靠性,需要比标准监督微调更复杂的策略。

  • 融入“思考”过程:**在生成过程中加入显式的中间步骤或推理过程,增强模型的推理能力。

  • 搜索作为数据生成工具:利用搜索方法生成高质量训练数据,进一步微调和改进模型。

  • 开发更多过程奖励模型(PRMs):丰富的 PRM 有助于提升不同领域的模型性能。

  • 扩展至非可验证领域: 将方法应用于结构较弱或主观性较强的任务,需要新的策略。

    结论:

    找到合适的解码方式并结合强大的验证器,是提升大型语言模型性能的关键。最佳解码策略取决于问题难度和计算预算,没有一种通用的方法。优化模型性能需要综合考虑任务需求和可用资源。****

解码方法可以单独使用,也可以结合使用,具体取决于任务需求和目标。以下是一些常见的情况:

  1. 单独使用:
  • 贪婪解码:适用于简单任务,计算效率高。

  • 束搜索:适用于需要生成高质量文本的任务,尽管计算复杂度较高。

  • 温度采样、Top-k 采样、Top-p 采样:用于控制生成文本的随机性和多样性。

  1. 结合使用:
  • 在一些复杂任务中,可能会结合多种解码方法。例如,先使用束搜索生成多个候选,然后再用温度采样或Top-p采样从中选择最优解。

  • 结合使用可以在保证生成质量的同时,增加文本的多样性和创意性。

总结:

在追求最佳性能的过程中,应根据具体任务和资源限制,选择最合适的解码方法,并视情况决定是否结合验证器。针对任务特点,择优选择解码策略,而非同时使用多种方法。


如何系统学习掌握AI大模型?

AI大模型作为人工智能领域的重要技术突破,正成为推动各行各业创新和转型的关键力量。抓住AI大模型的风口,掌握AI大模型的知识和技能将变得越来越重要。

学习AI大模型是一个系统的过程,需要从基础开始,逐步深入到更高级的技术。

这里给大家精心整理了一份全面的AI大模型学习资源,包括:AI大模型全套学习路线图(从入门到实战)、精品AI大模型学习书籍手册、视频教程、实战学习、面试题等,资料免费分享

1. 成长路线图&学习规划

要学习一门新的技术,作为新手一定要先学习成长路线图方向不对,努力白费

这里,我们为新手和想要进一步提升的专业人士准备了一份详细的学习成长路线图和规划。可以说是最科学最系统的学习成长路线。
在这里插入图片描述

2. 大模型经典PDF书籍

书籍和学习文档资料是学习大模型过程中必不可少的,我们精选了一系列深入探讨大模型技术的书籍和学习文档,它们由领域内的顶尖专家撰写,内容全面、深入、详尽,为你学习大模型提供坚实的理论基础(书籍含电子版PDF)

在这里插入图片描述

3. 大模型视频教程

对于很多自学或者没有基础的同学来说,书籍这些纯文字类的学习教材会觉得比较晦涩难以理解,因此,我们提供了丰富的大模型视频教程,以动态、形象的方式展示技术概念,帮助你更快、更轻松地掌握核心知识

在这里插入图片描述

4. 2024行业报告

行业分析主要包括对不同行业的现状、趋势、问题、机会等进行系统地调研和评估,以了解哪些行业更适合引入大模型的技术和应用,以及在哪些方面可以发挥大模型的优势。

在这里插入图片描述

5. 大模型项目实战

学以致用 ,当你的理论知识积累到一定程度,就需要通过项目实战,在实际操作中检验和巩固你所学到的知识,同时为你找工作和职业发展打下坚实的基础。

在这里插入图片描述

6. 大模型面试题

面试不仅是技术的较量,更需要充分的准备。

在你已经掌握了大模型技术之后,就需要开始准备面试,我们将提供精心整理的大模型面试题库,涵盖当前面试中可能遇到的各种技术问题,让你在面试中游刃有余。

在这里插入图片描述

全套的AI大模型学习资源已经整理打包,有需要的小伙伴可以微信扫描下方CSDN官方认证二维码,免费领取【保证100%免费

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值