当前位置：首页 > article >正文

语言模型的采样方法

article 2024/11/5 6:59:31

语言模型的采样方法

语言模型的输出为一个向量，该向量的每一维代表着词典中对应词的概率。

在采用自回归范式的文本生成任务中，语言模型将依次生成一组向量并将其解码为文本。将这组向量解码为文本的过程被成为语言模型解码。

解码过程显著影响着生成文本的质量。当前，两类主流的解码方法可以总结为 (1). 概率最大化方法; (2).随机采样方法。

自回归

自回归（Autoregression，简称AR）是一种统计模型，用于描述时间序列数据中各个时间点的值与其之前时间点值之间的关系。在自回归模型中，一个变量的当前值被认为是其过去值的函数，加上一个随机误差项。这种模型用于预测和分析时间序列数据。

基本概念

时间序列数据：指按时间顺序排列的数据点。
自回归模型：假设当前值 $Y_t$ 是过去 $p$ 个时间点值的线性组合加上一个随机误差项 $\epsilon_t$ 。数学表达式为：
$Y_t = c + \phi_1 Y_{t-1} + \phi_2 Y_{t-2} + \ldots + \phi_p Y_{t-p} + \epsilon_t$
其中， $c$ 是常数项， $\phi_1, \phi_2, \ldots, \phi_p$ 是模型参数， $\epsilon_t$ 是随机误差项。

模型类型

AR(1)：一阶自回归模型，只包含一个滞后项。
$Y_t = c + \phi_1 Y_{t-1} + \epsilon_t$
AR§：p阶自回归模型，包含 $p$ 个滞后项。
$Y_t = c + \phi_1 Y_{t-1} + \phi_2 Y_{t-2} + \ldots + \phi_p Y_{t-p} + \epsilon_t$

参数估计

自回归模型的参数通常通过最小化误差项的平方和来估计，这可以通过最大似然估计（MLE）或最小二乘法（OLS）来实现。

注意事项

自回归模型假设误差项是白噪声，即具有零均值和恒定方差。
模型的阶数 $p$ 需要根据数据和问题背景来确定，过高或过低的阶数都可能导致模型拟合不佳。

自回归模型是时间序列分析中的基础工具，通过捕捉数据的自相关性，帮助我们理解和预测时间序列数据的行为。

概率最大化方法

目标：
- 目标是最大化生成文本 $P(w_{N+1}:N+M)$ 的概率，即在给定前文的情况下，找到最有可能的后续词序列。
公式：
- 生成文本的概率可以表示为：
  $P(w_{N+1}:N+M) = \prod_{i=N}^{N+M-1} P(w_{i+1}|w_1:i) = \prod_{i=N}^{N+M-1} o_i[w_{i+1}]$

贪心搜索（Greedy Search）

初始化：
- 从一个初始状态或初始词开始，通常是特殊的起始符号，如 <s>。
迭代选择：
- 在每一步，模型根据当前的上下文（可能是空的或之前选择的词序列）计算所有可能的下一个词的概率。
选择概率最大的词：
- 选择概率最高的词作为当前步骤的输出：
  $w_{i+1} = \arg\max_{w \in D} o_i[w]$
  其中 $w_{i+1}$ 是下一步的词， $D$ 是词汇表， $o_i$ 是在给定上下文下计算的输出概率分布。
更新上下文：
- 将选择的词添加到当前上下文中，并更新模型的状态。
重复：
- 重复步骤2-4，直到达到终止条件，如特定的结束符号 </s> 或达到最大序列长度。
问题：
- 贪心搜索只考虑当前步骤中概率最大的词，可能导致后续词的概率都很小，陷入局部最优。

贪心搜索的问题

局部最优：
- 贪心搜索在每一步都选择局部最优的词，这可能导致整个序列的全局最优解被忽略。换句话说，它可能会陷入局部最优，而不是找到全局最优解。
缺乏回溯：
- 贪心搜索不进行回溯，一旦选择了一个词，就会一直沿着这个选择前进，不会考虑其他可能的路径。
忽略长距离依赖：
- 在某些情况下，早期的贪心选择可能会影响序列的长距离依赖关系，使得最终的输出序列质量下降。### 波束搜索（Beam Search）

波束搜索（Beam Search）

波束搜索（Beam Search）是一种在序列生成问题中常用的启发式搜索算法，特别是在自然语言处理领域，如机器翻译、文本摘要等任务中。它旨在找到使得整个序列的概率最大化的词序列。

初始化：
- 从一个初始状态开始，通常是特殊的起始符号，如 <s>。初始化一个波束，其中包含这个起始状态。
扩展波束：
- 在每一步 $i$ （从 $N$ 到 $N + M - 1$ ），对于波束中的每个状态，生成所有可能的下一个词。
- 计算每个扩展状态的概率，并选择概率最高的 $b$ 个状态作为新的波束。这里 $b$ 是波束的大小，表示在每一步保留的可能性最高的词的数量。
- 在每轮预测中保留 $b$ 个可能性最高的词，构成波束：
  $\min_{o_i[w] \text{ for } w \in B_i} > \max_{o_i[w] \text{ for } w \in D - B_i}$
终止条件：
- 继续扩展波束，直到达到预定的序列长度 $M$ 或遇到终止符号 </s>。
选择最优序列：
- 在所有 $M$ 个波束中，选择联合概率最大的词序列作为最终输出：
  $\{w_{N+1}, w_{N+2}, \ldots, w_{N+M}\} = \arg\max_{\{w_i \in B_i \text{ for } 1 \leq i \leq M\}} \prod_{i=1}^{M} o_{N+i}[w_i]$

波束搜索的关键点

波束修剪（Beam Pruning）：
- 在每一步，只保留概率最高的 $b$ 个状态，这有助于减少计算量并避免无效搜索。
长度归一化：
- 有时，波束搜索会考虑长度归一化，即通过惩罚过长的序列来平衡长度对概率的影响。
多样性：
- 为了增加结果的多样性，可以采用一些技术，如束熵搜索（Beam Entropy Search）或束多样性搜索（Beam Diversity Search）。

波束搜索的优势

全局最优：
- 与贪心搜索相比，波束搜索考虑了多个候选词，更有可能找到全局最优解。
避免局部最优：
- 波束搜索避免了贪心搜索中的局部最优问题，因为它在每一步都考虑了多个可能的下一步。
灵活性：
- 波束搜索的参数（如波束大小 $b$ ）可以根据具体任务进行调整，以平衡计算成本和结果质量。

概率最大化方法的问题

平庸文本：
- 概率最大化方法倾向于生成最常见的平庸文本，缺乏多样性和新颖性。
改进：
- 为了提升生成文本的新颖度和多样性，可以在解码过程中加入随机元素，如随机采样方法。

随机采样方法

为了增加生成文本的多样性，随机采样的方法在预测时增加了随机性。在每轮预测时，其先选出一组可能性高的候选词，然后按照其概率分布进行随机采样，采样出的词作为本轮的预测结果。

当前，主流的 Top-K 采样和 Top-P 采样方法分别通过指定候选词数量和划定候选词概率阈值的方法对候选词进行选择。

在采样方法中加入 Temperature 机制可以对候选词的概率分布进行调整。

Top-K 采样

Top-K 采样是一种结合了贪心搜索和随机采样的序列生成方法，它旨在平衡探索（exploration）和利用（exploitation）。这种方法在每轮预测中不是选择概率最高的单个词，而是选择概率最高的 $K$ 个词作为候选词集合，然后从这个集合中随机采样一个词作为下一步的输出。

选择候选词：
- 在每一步 $i$ ，从所有可能的词中选择概率最高的 $K$ 个词作为候选词集合。
概率归一化：
- 对这 $K$ 个候选词的概率进行归一化，使得它们的和为1。这通常通过softmax函数实现，但在这里只对 $K$ 个候选词应用，而不是整个词汇表。
随机采样：
- 根据归一化后的概率分布，从 $K$ 个候选词中随机采样一个词作为输出。

Top-K 采样的公式

对于每个候选词 $w_j$ （其中 $\ldots, K$ ），归一化的概率 $p(w_{i+1})$ 计算如下：
$p(w_{i+1}) = \left( \frac{\exp\left(\frac{o_i[w_1]}{T}\right)}{\sum_{j=1}^{K} \exp\left(\frac{o_i[w_j]}{T}\right)}, \ldots, \frac{\exp\left(\frac{o_i[w_K]}{T}\right)}{\sum_{j=1}^{K} \exp\left(\frac{o_i[w_j]}{T}\right)} \right)$
然后，根据这个分布随机采样 $w_{i+1}$ ：
$w_{i+1} \sim p(w_{i+1})$
其中：
- $o_i[w_j]$ 是第 $i$ 轮预测中词 $w_j$ 的原始概率。
- $T$ 是温度参数（temperature），用于控制概率分布的平滑程度。 $T$ 的值越大，分布越平滑，值越小，分布越尖锐。

Top-K 采样的优势

平衡探索和利用：
- Top-K 采样通过选择 $K$ 个最高概率的词，增加了生成多样性，避免了贪心搜索中的局部最优问题。
控制随机性：
- 通过调整 $K$ 的值，可以控制生成过程中的随机性。较小的 $K$ 值会增加随机性，而较大的 $K$ 值会接近贪心搜索。
灵活性：
- 温度参数 $T$ 提供了额外的控制，允许调整概率分布的形状，以适应不同的生成需求。

Top-P 采样

Top-P 采样，也称为 Nucleus 采样，是一种序列生成技术，它不是简单地选择概率最高的 $K$ 个词，而是选择累积概率达到某个阈值 $p$ 的所有词作为候选词集合。

这种方法旨在生成更多样化和连贯的文本，同时避免贪心搜索的局部最优问题。

选择候选词：
- 在每一步 $i$ ，从所有可能的词中选择累积概率达到阈值 (p) 的词作为候选词集合 $S_p$ 。
概率归一化：
- 对候选词集合 $S_p$ 中的词的概率进行归一化，使得它们的和为1。这通常通过softmax函数实现，但在这里只对候选词集合中的词应用。
随机采样：
- 根据归一化后的概率分布，从候选词集合 $S_p$ 中随机采样一个词作为输出。

Top-P 采样的公式

对于每个候选词 $w_j$ （其中 $\ldots, |S_p|$ ），归一化的概率 $p(w_{i+1})$ 计算如下：
$p(w_{i+1}) = \left( \frac{\exp\left(\frac{o_i[w_1]}{T}\right)}{\sum_{j=1}^{|S_p|} \exp\left(\frac{o_i[w_j]}{T}\right)}, \ldots, \frac{\exp\left(\frac{o_i[w_{|S_p|}]}{T}\right)}{\sum_{j=1}^{|S_p|} \exp\left(\frac{o_i[w_j]}{T}\right)} \right)$
然后，根据这个分布随机采样 $w_{i+1}$ ：
$w_{i+1} \sim p(w_{i+1})$
其中：

$o_i[w_j]$ 是第 $i$ 轮预测中词 $w_j$ 的原始概率。
$T$ 是温度参数（temperature），用于控制概率分布的平滑程度。 $T$ 的值越大，分布越平滑，值越小，分布越尖锐。
$S_p|$ 是候选词集合 $S_p$ 的大小。

Top-P 采样的优势

提高多样性：
- 通过选择累积概率达到阈值 (p) 的词，Top-P 采样可以生成更多样化的文本。
控制风险：
- 通过设置阈值 (p)，可以控制生成文本的冒险程度，避免生成概率极低的词。
灵活性：
- 温度参数 $T$ 提供了额外的控制，允许调整概率分布的形状，以适应不同的生成需求。

Temperature 机制的作用

Temperature 机制是一种调整 Softmax 函数输出概率分布的方法，它通过改变 Softmax 函数中输入的尺度来控制输出概率分布的形状。

这种机制在机器学习和深度学习中被用来控制模型的不确定性和随机性，特别是在生成模型和强化学习中。

增加随机性：当 Temperature $T > 1$ 时，Softmax 函数的输出分布变得更加平坦，这意味着各个类别的概率更加接近，从而增加了模型输出的随机性。
减少随机性：当 $0 < T < 1$ 时，Softmax 函数的输出分布变得更加集中，这意味着模型更倾向于输出概率最高的类别，从而减少了模型输出的随机性。

Temperature 机制的公式

对于一个给定的输入 $o_i$ 和类别集合 $S_p$ ，Softmax 函数的输出 $p(w_{i+1})$ 可以表示为：

$p(w_{i+1}) = \frac{\exp\left(\frac{o_i[w_1]}{T}\right)}{\sum_{j=1}^{|S_p|} \exp\left(\frac{o_i[w_j]}{T}\right)}, \ldots, \frac{\exp\left(\frac{o_i[w_{|S_p|}]}{T}\right)}{\sum_{j=1}^{|S_p|} \exp\left(\frac{o_i[w_j]}{T}\right)}$

其中：
- $o_i[w_j]$ 是模型对于第 $i$ 个输入和第 $j$ 个类别的原始输出（通常是神经网络的最后一层的输出）。
- $T$ 是 Temperature 参数，控制概率分布的形状。
- $S_p|$ 是类别集合 $S_p$ 中的类别数量。
- $\exp$ 是指数函数。

引入 Temperature

在引入 Temperature 机制后，Top-K 采样和 Top-P 采样的候选集分布都受到了 Temperature 参数 $T$ 的影响。这两种采样方法都是在生成模型中常用的技术，用于从模型输出的概率分布中选择最可能的候选集。

Top-K 采样

Top-K 采样是一种选择概率分布中前 $K$ 个最高概率值的方法。在引入 Temperature 机制后，Top-K 采样的候选集分布可以表示为：

$p(w_{i1+1}, \ldots, w_{iK+1}) = \left( \frac{\exp\left(\frac{o_i[w_{1i+1}]}{T}\right)}{\sum_{j=1}^{K} \exp\left(\frac{o_i[w_{ji+1}]}{T}\right)}, \ldots, \frac{\exp\left(\frac{o_i[w_{Ki+1}]}{T}\right)}{\sum_{j=1}^{K} \exp\left(\frac{o_i[w_{ji+1}]}{T}\right)} \right)$

这里：

$w_{i1+1}, \ldots, w_{iK+1}$ 表示选择的前 $K$ 个候选。
$o_i[w_{ji+1}]$ 是模型对于第 $i$ 个输入和第 $j$ 个候选的输出。
$T$ 是 Temperature 参数。

Top-P 采样

Top-P 采样是一种选择累积概率达到 $P$ 百分比的候选集的方法。在引入 Temperature 机制后，Top-P 采样的候选集分布可以表示为：

$p(w_{i1+1}, \ldots, w_{i|S_p|+1}) = \left( \frac{\exp\left(\frac{o_i[w_{1i+1}]}{T}\right)}{\sum_{j=1}^{|S_p|} \exp\left(\frac{o_i[w_{ji+1}]}{T}\right)}, \ldots, \frac{\exp\left(\frac{o_i[w_{|S_p|i+1}]}{T}\right)}{\sum_{j=1}^{|S_p|} \exp\left(\frac{o_i[w_{ji+1}]}{T}\right)} \right)$