对大模型输出的 logits 进行处理,从而控制文本的生成

对大模型输出的 logits 进行处理,从而控制文本的生成

flyfish

在文本生成任务中,模型输出的 logits 代表了每个词被选为下一个生成词的未归一化概率得分。通过对 logits 进行处理,可以精确地控制文本的生成

基本原理

在每一步生成过程中,模型会输出一个 logits 向量,其长度等于词汇表的大小,每个元素对应词汇表中一个词的得分。通常,会对 logits 应用 softmax 函数将其转换为概率分布,然后根据这个概率分布来选择下一个生成的词。而 logits_processor 就是在应用 softmax 函数之前,对 logits 进行修改,从而改变最终的概率分布和词的选择。

具体控制方式

1. 避免重复
  • 重复惩罚(RepetitionPenaltyLogitsProcessor
    • 机制:对于已经在生成文本中出现过的词,降低其 logits 的值。具体来说,会将这些词的 logits 除以一个大于 1 的惩罚系数,使得它们在后续生成中被选中的概率降低。
    • 示例:假设生成的文本中已经出现了“苹果”这个词,当模型再次预测下一个词时,“苹果”对应的 logits 会被惩罚,从而减少再次生成“苹果”的可能性,避免文本中出现过多重复内容。
  • 不重复 n - gram(NoRepeatNGramLogitsProcessor
    • 机制:检查生成的文本中是否已经存在某个 n - gram(连续的 n 个词),如果存在,则将可能导致该 n - gram 重复出现的词的 logits 设为负无穷。这样,在后续的概率计算中,这些词的概率会变为 0,不会被选中。
    • 示例:如果 n = 2,当前生成的文本是“我 喜欢”,那么在选择下一个词时,会避免选择那些会导致“我 喜欢”这个 2 - gram 重复出现的词,如“我”或“喜欢”,从而提高文本的多样性。
2. 控制生成长度
  • 最小长度限制(MinLengthLogitsProcessor
    • 机制:在生成的文本长度未达到指定的最小长度之前,将结束标记(EOS)的 logits 设为负无穷。这样,在 softmax 处理后,结束标记的概率会变为 0,模型不会选择结束生成,确保文本达到一定的长度。
    • 示例:如果设置最小长度为 10 个词,在生成的词数小于 10 时,结束标记的 logits 始终为负无穷,模型会继续生成,直到达到最小长度要求。
  • 最小新标记数(MinNewTokensLengthLogitsProcessor
    • 机制:类似于最小长度限制,不过是针对新生成的标记数量。在新生成的标记数未达到指定数量之前,降低结束标记的 logits,保证生成足够数量的新内容。
3. 采样策略调整
  • 温度调整(TemperatureLogitsWarper
    • 机制:将 logits 除以一个温度参数 temperature。温度越高,logits 之间的差异会被缩小,经过 softmax 处理后,概率分布会更加均匀,采样会更随机;温度越低,logits 之间的差异会被放大,概率分布会更集中,更倾向于选择概率最大的词。
    • 示例:当 temperature = 1 时,保持原始的 logits 分布;当 temperature > 1 时,模型可能会生成一些更具创意但可能不太准确的文本;当 temperature < 1 时,模型会更保守,生成的文本更符合常见的表达。
  • Top - k 采样(TopKLogitsWarper
    • 机制:只保留 logits 中概率最高的 k 个词,将其余词的 logits 设为负无穷。这样,在后续的采样中,只会从这 k 个词中选择下一个生成的词,限制了采样范围,提高了生成的稳定性。
    • 示例:如果 k = 5,模型会在每次生成时,只考虑概率最高的 5 个词,排除其他词的干扰。
  • Top - p 采样(TopPLogitsWarper
    • 机制:选择累积概率达到 p 的最小词集合,只保留这些词的 logits,其余词的 logits 设为负无穷。这种方法结合了概率和词的数量,既能控制采样范围,又能适应不同的概率分布。
    • 示例:如果 p = 0.9,模型会选择累积概率达到 0.9 的最小词集合,从这个集合中进行采样。
4. 约束生成内容
  • 禁用词过滤(NoBadWordsLogitsProcessor
    • 机制:将禁用词的 logits 设为负无穷,使得这些词在后续的概率计算中概率为 0,不会被选中,从而避免生成包含禁用词的文本。
    • 示例:如果禁用词列表中包含“脏话”,那么在生成过程中,“脏话”对应的 logits 会被设为负无穷,不会出现在生成的文本中。
  • 前缀约束(PrefixConstrainedLogitsProcessor
    • 机制:根据给定的前缀允许标记函数,限制生成的词必须符合特定的前缀约束。不符合约束的词的 logits 会被设为负无穷,从而保证生成的文本符合特定的前缀要求。
    • 示例:如果要求生成的文本必须以“今天”开头,那么在生成第一个词时,只有与“今天”相关的词的 logits 会被保留,其他词的 logits 会被设为负无穷。

配置参数

参数数据类型默认值含义
guidance_scalefloatNone引导比例,用于无批量分类器自由引导,值不为 1 时会添加相应的 logits 处理器,影响生成过程的引导程度。
sequence_bias-None序列偏差,用于控制特定序列的生成概率,设置后会添加序列偏差 logits 处理器。
diversity_penaltyfloatNone多样性惩罚,大于 0 时会添加汉明多样性 logits 处理器,鼓励生成结果更具多样性。
encoder_repetition_penaltyfloatNone编码器重复惩罚,不为 1 且编码器输入 ID 形状符合要求时,会添加编码器重复惩罚 logits 处理器,减少编码器输入相关的重复内容。
repetition_penaltyfloatNone重复惩罚,不为 1 时会添加重复惩罚 logits 处理器,防止生成结果出现过多重复。
no_repeat_ngram_sizeintNone不重复 n - gram 大小,大于 0 时会添加不重复 n - gram logits 处理器,避免生成的文本中出现重复的 n - gram 片段。
encoder_no_repeat_ngram_sizeintNone编码器不重复 n - gram 大小,大于 0 且编码器输入 ID 形状符合要求时,会添加编码器不重复 n - gram logits 处理器,减少编码器输入相关的重复 n - gram 内容。
bad_words_ids-None禁用词 ID,设置后会添加禁用词 logits 处理器,防止生成包含指定禁用词的文本。
min_lengthintNone最小长度,大于 0 且有结束标记张量时,会添加最小长度 logits 处理器,确保生成的文本达到最小长度要求。
min_new_tokensintNone最小新标记数,大于 0 且有结束标记张量时,会添加最小新标记长度 logits 处理器,保证生成的新标记数量达到要求。
forced_bos_token_idintNone强制起始标记 ID,设置后会添加强制起始标记 logits 处理器,确保生成的文本以指定的标记开始。
forced_eos_token_idintNone强制结束标记 ID,设置后会添加强制结束标记 logits 处理器,使生成的文本在达到指定标记时结束。
remove_invalid_valuesboolFalse是否移除无效值,为 True 时会添加移除无穷大和 NaN 值的 logits 处理器,保证生成过程中 logits 的有效性。
exponential_decay_length_penalty-None指数衰减长度惩罚,设置后会添加指数衰减长度惩罚处理器,对生成文本的长度进行惩罚控制。
suppress_tokens-None抑制标记,设置后会添加抑制标记 logits 处理器,降低指定标记的生成概率。
begin_suppress_tokens-None起始抑制标记,设置后会添加起始抑制标记 logits 处理器,在生成的起始阶段抑制指定标记的生成。
forced_decoder_ids-None强制解码器 ID,不建议使用,设置后会抛出异常,提示使用 input_idsdecoder_input_ids 代替。
do_sampleboolFalse是否使用采样策略,为 True 时会根据其他采样相关参数添加相应的 logits 调整器。
temperaturefloatNone采样温度,不为 1 时会添加温度 logits 调整器,控制采样的随机性,值越大随机性越强。
top_kintNonetop - k 采样值,不为 0 时会添加 top - k logits 调整器,只考虑概率最高的 k 个标记进行采样。
top_pfloatNonetop - p 采样值,小于 1 时会添加 top - p logits 调整器,只考虑累积概率达到 p 的标记进行采样。
min_pfloatNone最小概率阈值,设置后会添加最小概率 logits 调整器,在温度缩放后应用,控制采样的最小概率。
typical_pfloatNone典型概率采样值,小于 1 时会添加典型概率 logits 调整器,基于典型概率进行采样。
epsilon_cutofffloatNoneepsilon 截断值,在 0 到 1 之间时会添加 epsilon logits 调整器,用于截断低概率标记。
eta_cutofffloatNoneeta 截断值,在 0 到 1 之间时会添加 eta logits 调整器,结合设备信息对低概率标记进行截断。
watermarking_config-None水印配置,设置后会添加水印处理器,在生成的文本中添加水印。
renormalize_logitsboolFalse是否重新归一化 logits,为 True 时会添加 logit 归一化处理器,确保 logits 归一化。

logits 说明

logits 是模型在进行分类或预测任务时,最后一层神经元的原始输出值,它是未经过归一化处理的数值。在文本生成场景中,logits 代表了模型预测词汇表中每个词作为下一个生成词的得分,这些得分反映了模型对每个词成为下一个词的相对可能性判断,但并非是概率值。

数学公式

1. 线性变换得到 logits

在许多深度学习模型中,logits 通常是通过对前一层的输出进行线性变换得到的。假设模型前一层的输出为向量 h \mathbf{h} h,权重矩阵为 W \mathbf{W} W,偏置向量为 b \mathbf{b} b,则 logits 向量 z \mathbf{z} z 的计算公式如下:

z = W h + b \mathbf{z} = \mathbf{W}\mathbf{h} + \mathbf{b} z=Wh+b

其中, h \mathbf{h} h 是前一层输出的特征向量,维度通常为 d h d_h dh W \mathbf{W} W 是权重矩阵,维度为 V × d h V \times d_h V×dh V V V 是词汇表的大小; b \mathbf{b} b 是偏置向量,维度为 V V V z \mathbf{z} zlogits 向量,维度为 V V V,每个元素 z i z_i zi 对应词汇表中第 i i i 个词的得分。

2. logits 转换为概率分布

为了将 logits 转换为概率分布,通常会使用 softmax 函数。softmax 函数可以将 logits 向量中的每个元素转换为一个在 [ 0 , 1 ] [0, 1] [0,1] 范围内的值,且所有元素之和为 1,符合概率分布的定义。softmax 函数的数学公式如下:

P ( y i ) = e z i ∑ j = 1 V e z j P(y_i) = \frac{e^{z_i}}{\sum_{j=1}^{V} e^{z_j}} P(yi)=j=1Vezjezi

其中, P ( y i ) P(y_i) P(yi) 是词汇表中第 i i i 个词被选为下一个生成词的概率, z i z_i zilogits 向量中第 i i i 个元素的值, V V V 是词汇表的大小。

示例

假设词汇表大小 V = 3 V = 3 V=3,模型输出的 logits 向量为 z = [ 2 , 1 , 3 ] \mathbf{z} = [2, 1, 3] z=[2,1,3],下面计算经过 softmax 函数处理后的概率分布:

首先,计算分母的值:

∑ j = 1 3 e z j = e 2 + e 1 + e 3 ≈ 7.389 + 2.718 + 20.086 = 30.193 \sum_{j=1}^{3} e^{z_j} = e^2 + e^1 + e^3 \approx 7.389 + 2.718 + 20.086 = 30.193 j=13ezj=e2+e1+e37.389+2.718+20.086=30.193

然后,分别计算每个词的概率:

P ( y 1 ) = e 2 30.193 ≈ 7.389 30.193 ≈ 0.245 P(y_1) = \frac{e^2}{30.193} \approx \frac{7.389}{30.193} \approx 0.245 P(y1)=30.193e230.1937.3890.245

P ( y 2 ) = e 1 30.193 ≈ 2.718 30.193 ≈ 0.090 P(y_2) = \frac{e^1}{30.193} \approx \frac{2.718}{30.193} \approx 0.090 P(y2)=30.193e130.1932.7180.090

P ( y 3 ) = e 3 30.193 ≈ 20.086 30.193 ≈ 0.665 P(y_3) = \frac{e^3}{30.193} \approx \frac{20.086}{30.193} \approx 0.665 P(y3)=30.193e330.19320.0860.665

可以看到,经过 softmax 函数处理后,得到了一个概率分布 [ 0.245 , 0.090 , 0.665 ] [0.245, 0.090, 0.665] [0.245,0.090,0.665],表示词汇表中三个词被选为下一个生成词的概率。

在模型中的作用

在文本生成任务中,模型会根据 logits 转换后的概率分布来选择下一个生成的词。常见的选择方法有贪心搜索(选择概率最大的词)、采样搜索(根据概率分布随机采样)等。同时,logits_processor 会对 logits 进行调整,从而影响最终的概率分布和词的选择,以控制文本生成的行为和质量。

代码说明

logits_processor 是 _get_logits_processor 方法的一个参数,它是一个可选的 LogitsProcessorList 对象。这个方法会根据 GenerationConfig 中的各种配置参数,创建一系列不同的 LogitsProcessor 实例,并将它们添加到 processors 列表中。最后,如果传入了 logits_processor,还会将其与新创建的处理器列表进行合并。

def _get_logits_processor(
        self,
        generation_config: GenerationConfig,
        input_ids_seq_length: int,
        encoder_input_ids: torch.LongTensor,
        prefix_allowed_tokens_fn: Callable[[int, torch.Tensor], List[int]],
        logits_processor: Optional[LogitsProcessorList],
        device: str = None,
        model_kwargs: Optional[Dict[str, Any]] = None,
        negative_prompt_ids: Optional[torch.Tensor] = None,
        negative_prompt_attention_mask: Optional[torch.Tensor] = None,
    ) -> LogitsProcessorList:
        """
        此函数返回一个 `LogitsProcessorList` 对象,该对象包含所有用于修改语言模型头部得分的相关 `LogitsProcessor` 实例。
        这些处理器会对模型预测的 logits 进行调整,以控制文本生成的行为,例如避免重复、控制生成长度等。

        参数:
            generation_config (GenerationConfig): 生成配置对象,包含了文本生成过程中的各种配置参数。
            input_ids_seq_length (int): 输入 ID 序列的长度。
            encoder_input_ids (torch.LongTensor): 编码器的输入 ID。
            prefix_allowed_tokens_fn (Callable[[int, torch.Tensor], List[int]]): 一个可调用对象,用于指定允许的前缀标记。
            logits_processor (Optional[LogitsProcessorList]): 可选的 logits 处理器列表。
            device (str, optional): 设备名称,如 'cuda' 或 'cpu'。默认为 None。
            model_kwargs (Optional[Dict[str, Any]], optional): 模型的其他关键字参数。默认为 None。
            negative_prompt_ids (Optional[torch.Tensor], optional): 负提示的 ID。默认为 None。
            negative_prompt_attention_mask (Optional[torch.Tensor], optional): 负提示的注意力掩码。默认为 None。

        返回:
            LogitsProcessorList: 包含所有 logits 处理器的列表。
        """
        # 实例化一个空的处理器列表
        processors = LogitsProcessorList()

        # 如果配置了引导比例且不为 1,则添加无批量分类器自由引导 logits 处理器
        if generation_config.guidance_scale is not None and generation_config.guidance_scale != 1:
            processors.append(
                UnbatchedClassifierFreeGuidanceLogitsProcessor(
                    generation_config.guidance_scale,
                    self,
                    unconditional_ids=negative_prompt_ids,
                    unconditional_attention_mask=negative_prompt_attention_mask,
                    use_cache=generation_config.use_cache,
                )
            )
        # 如果配置了序列偏差,则添加序列偏差 logits 处理器
        if generation_config.sequence_bias is not None:
            processors.append(SequenceBiasLogitsProcessor(sequence_bias=generation_config.sequence_bias))

        # 如果配置了多样性惩罚且大于 0,则添加汉明多样性 logits 处理器
        if generation_config.diversity_penalty is not None and generation_config.diversity_penalty > 0.0:
            processors.append(
                HammingDiversityLogitsProcessor(
                    diversity_penalty=generation_config.diversity_penalty,
                    num_beams=generation_config.num_beams,
                    num_beam_groups=generation_config.num_beam_groups,
                )
            )
        # 如果配置了编码器重复惩罚且不为 1,并且编码器输入 ID 的形状为二维,则添加编码器重复惩罚 logits 处理器
        if (
            generation_config.encoder_repetition_penalty is not None
            and generation_config.encoder_repetition_penalty != 1.0
        ):
            if len(encoder_input_ids.shape) == 2:
                processors.append(
                    EncoderRepetitionPenaltyLogitsProcessor(
                        penalty=generation_config.encoder_repetition_penalty,
                        encoder_input_ids=encoder_input_ids,
                    )
                )
            else:
                # 如果编码器输入 ID 形状不符合要求,发出警告
                warnings.warn(
                    "Passing `encoder_repetition_penalty` requires some form of `input_ids` to be passed to "
                    "`generate`, ignoring the argument.",
                    UserWarning,
                )
        # 如果配置了重复惩罚且不为 1,则添加重复惩罚 logits 处理器
        if generation_config.repetition_penalty is not None and generation_config.repetition_penalty != 1.0:
            processors.append(RepetitionPenaltyLogitsProcessor(penalty=generation_config.repetition_penalty))
        # 如果配置了不重复 n-gram 大小且大于 0,则添加不重复 n-gram logits 处理器
        if generation_config.no_repeat_ngram_size is not None and generation_config.no_repeat_ngram_size > 0:
            processors.append(NoRepeatNGramLogitsProcessor(generation_config.no_repeat_ngram_size))
        # 如果配置了编码器不重复 n-gram 大小且大于 0,并且编码器输入 ID 的形状为二维,则添加编码器不重复 n-gram logits 处理器
        if (
            generation_config.encoder_no_repeat_ngram_size is not None
            and generation_config.encoder_no_repeat_ngram_size > 0
        ):
            if len(encoder_input_ids.shape) == 2:
                processors.append(
                    EncoderNoRepeatNGramLogitsProcessor(
                        generation_config.encoder_no_repeat_ngram_size,
                        encoder_input_ids,
                    )
                )
            else:
                # 如果编码器输入 ID 形状不符合要求,发出警告
                warnings.warn(
                    "Passing `encoder_no_repeat_ngram_size` requires some form of `input_ids` to be passed to "
                    "`generate`, ignoring the argument.",
                    UserWarning,
                )
        # 如果配置了禁用词 ID,则添加禁用词 logits 处理器
        if generation_config.bad_words_ids is not None:
            processors.append(
                NoBadWordsLogitsProcessor(
                    generation_config.bad_words_ids,
                    generation_config._eos_token_tensor,
                )
            )
        # 如果配置了最小长度且大于 0,并且有结束标记张量,则添加最小长度 logits 处理器
        if (
            generation_config.min_length is not None
            and generation_config._eos_token_tensor is not None
            and generation_config.min_length > 0
        ):
            processors.append(
                MinLengthLogitsProcessor(
                    generation_config.min_length,
                    generation_config._eos_token_tensor,
                    device=device,
                )
            )
        # 如果配置了最小新标记数且大于 0,并且有结束标记张量,则添加最小新标记长度 logits 处理器
        if (
            generation_config.min_new_tokens is not None
            and generation_config._eos_token_tensor is not None
            and generation_config.min_new_tokens > 0
        ):
            processors.append(
                MinNewTokensLengthLogitsProcessor(
                    input_ids_seq_length,
                    generation_config.min_new_tokens,
                    generation_config._eos_token_tensor,
                    device=device,
                )
            )
        # 如果提供了前缀允许标记函数,则添加前缀约束 logits 处理器
        if prefix_allowed_tokens_fn is not None:
            processors.append(
                PrefixConstrainedLogitsProcessor(
                    prefix_allowed_tokens_fn,
                    generation_config.num_beams // generation_config.num_beam_groups,
                )
            )
        # 如果配置了强制起始标记 ID,则添加强制起始标记 logits 处理器
        if generation_config.forced_bos_token_id is not None:
            processors.append(
                ForcedBOSTokenLogitsProcessor(
                    generation_config.forced_bos_token_id,
                )
            )
        # 如果配置了强制结束标记 ID,则添加强制结束标记 logits 处理器
        if generation_config.forced_eos_token_id is not None:
            processors.append(
                ForcedEOSTokenLogitsProcessor(
                    generation_config.max_length,
                    generation_config.forced_eos_token_id,
                    device=device,
                )
            )
        # 如果配置了移除无效值,则添加移除无穷大和 NaN 值的 logits 处理器
        if generation_config.remove_invalid_values is True:
            processors.append(InfNanRemoveLogitsProcessor())
        # 如果配置了指数衰减长度惩罚,则添加指数衰减长度惩罚处理器
        if generation_config.exponential_decay_length_penalty is not None:
            processors.append(
                ExponentialDecayLengthPenalty(
                    generation_config.exponential_decay_length_penalty,
                    generation_config._eos_token_tensor,
                    input_ids_seq_length,
                )
            )
        # 如果配置了抑制标记,则添加抑制标记 logits 处理器
        if generation_config.suppress_tokens is not None:
            processors.append(
                SuppressTokensLogitsProcessor(
                    generation_config.suppress_tokens,
                    device=device,
                )
            )
        # 如果配置了起始抑制标记,则添加起始抑制标记 logits 处理器
        if generation_config.begin_suppress_tokens is not None:
            begin_index = input_ids_seq_length
            begin_index = (
                begin_index
                if (input_ids_seq_length > 1 or generation_config.forced_bos_token_id is None)
                else begin_index + 1
            )
            processors.append(
                SuppressTokensAtBeginLogitsProcessor(
                    generation_config.begin_suppress_tokens,
                    begin_index,
                    device=device,
                )
            )
        # 如果配置了强制解码器 ID,则抛出异常,提示使用 input_ids 或 decoder_input_ids 代替
        if generation_config.forced_decoder_ids is not None:
            # TODO (sanchit): move this exception to GenerationConfig.validate() when TF & FLAX are aligned with PT
            raise ValueError(
                "You have explicitly specified `forced_decoder_ids`. Please remove the `forced_decoder_ids` argument "
                "in favour of `input_ids` or `decoder_input_ids` respectively.",
            )

        # 合并自定义的 logits 处理器列表
        processors = self._merge_criteria_processor_list(processors, logits_processor)

        # 以下处理器之前被称为 `LogitsWarpers`,仅在采样策略下应用
        if generation_config.do_sample:
            # 在束搜索方法中,我们需要至少保留一个非结束标记来探索可能有更好得分的延续(即保留 len(list(generation_config._eos_token_tensor)) + 1)
            if generation_config.num_beams > 1:
                if isinstance(generation_config._eos_token_tensor, list):
                    min_tokens_to_keep = len(generation_config._eos_token_tensor) + 1
                elif isinstance(generation_config._eos_token_tensor, torch.Tensor):
                    min_tokens_to_keep = generation_config._eos_token_tensor.shape[0] + 1
                else:
                    min_tokens_to_keep = 2
            else:
                min_tokens_to_keep = 1

            # 以下思路主要借鉴自这个 PR: https://github.com/huggingface/transformers/pull/5420/files
            # 所有采样器可以在 `generation_utils_samplers.py` 中找到
            # 如果配置了温度且不为 1,则添加温度 logits 调整器
            if generation_config.temperature is not None and generation_config.temperature != 1.0:
                processors.append(TemperatureLogitsWarper(generation_config.temperature))
            # 如果配置了 top-k 采样且不为 0,则添加 top-k logits 调整器
            if generation_config.top_k is not None and generation_config.top_k != 0:
                processors.append(
                    TopKLogitsWarper(top_k=generation_config.top_k, min_tokens_to_keep=min_tokens_to_keep)
                )
            # 如果配置了 top-p 采样且小于 1,则添加 top-p logits 调整器
            if generation_config.top_p is not None and generation_config.top_p < 1.0:
                processors.append(
                    TopPLogitsWarper(top_p=generation_config.top_p, min_tokens_to_keep=min_tokens_to_keep)
                )
            # 如果配置了最小概率阈值,则添加最小概率 logits 调整器
            if generation_config.min_p is not None:
                # 在温度缩放后应用(见 https://github.com/ggerganov/llama.cpp/pull/3841#issuecomment-2073826084)
                processors.append(
                    MinPLogitsWarper(min_p=generation_config.min_p, min_tokens_to_keep=min_tokens_to_keep)
                )
            # 如果配置了典型概率采样且小于 1,则添加典型概率 logits 调整器
            if generation_config.typical_p is not None and generation_config.typical_p < 1.0:
                processors.append(
                    TypicalLogitsWarper(mass=generation_config.typical_p, min_tokens_to_keep=min_tokens_to_keep)
                )
            # 如果配置了 epsilon 截断且在 0 到 1 之间,则添加 epsilon logits 调整器
            if generation_config.epsilon_cutoff is not None and 0.0 < generation_config.epsilon_cutoff < 1.0:
                processors.append(
                    EpsilonLogitsWarper(
                        epsilon=generation_config.epsilon_cutoff, min_tokens_to_keep=min_tokens_to_keep
                    )
                )
            # 如果配置了 eta 截断且在 0 到 1 之间,则添加 eta logits 调整器
            if generation_config.eta_cutoff is not None and 0.0 < generation_config.eta_cutoff < 1.0:
                processors.append(
                    EtaLogitsWarper(
                        epsilon=generation_config.eta_cutoff, min_tokens_to_keep=min_tokens_to_keep, device=device
                    )
                )

        # 水印处理应该在所有 logits 处理完成后进行(见 #34630)
        if generation_config.watermarking_config is not None:
            processors.append(
                generation_config.watermarking_config.construct_processor(self.config.vocab_size, device)
            )

        # `LogitNormalization` 应该始终是最后一个 logit 处理器(如果存在)
        if generation_config.renormalize_logits is True:
            processors.append(LogitNormalization())
        return processors
transformers/src/transformers/generation/utils.py
<think>好的,用户想了解机器学习中的logits的定义和用法。首先我需要明确logits在机器学习中的基本概念。Logits通常指的是模型输出的原始分数,还没有经过归一化处理,比如softmax转换前的值。这个术语在分类任务中经常出现,尤其是在神经网络中。 接下来,我需要回忆一下logits的具体应用场景。例如,在二分类问题中,模型可能输出一个logit值,然后通过sigmoid函数将其转换为概率。而在多分类问题中,多个logits经过softmax处理得到各个类别的概率分布。这里需要区分logits和概率之间的区别,强调logits是未经处理的原始分数。 然后,用户可能想知道为什么使用logits而不是直接输出概率。这时候需要解释数值稳定性的问题,比如在计算交叉熵损失时,直接使用logits可以避免数值下溢或溢出的问题。此外,在生成对抗网络(GANs)中,判别器的输出通常也是logits,这可能也是需要提到的点。 还要注意用户提供的引用内容,里面有提到RNN在处理序列数据中的应用,可能和自然语言处理相关。这时候可以结合NLP中的例子,比如语言模型预测下一个词时,模型输出的是logits,再转换为概率分布进行采样。 需要检查是否有相关的引用文献可以支持解释。例如,引用[2]提到RNN用于序列数据,处理长期依赖关系,可能和logits在序列生成中的应用相关。不过用户提供的引用似乎和logits没有直接关系,可能需要指出这一点,或者说明logits在这些模型中的普遍存在性。 此外,用户的问题可能隐含他们想了解logits在实际应用中的具体用法,比如在损失函数中的使用,或者在模型推理时的处理步骤。这时候需要举例说明,比如在PyTorch或TensorFlow中如何操作logits,以及常见的错误,比如忘记应用softmax导致的问题。 最后,生成相关问题需要考虑用户可能进一步想了解的内容,比如logits与概率的区别、如何处理多标签分类中的logits、在损失函数中的具体应用,以及在不同机器学习框架中的实现方式。这些问题可以帮助用户更全面地理解logits的各个方面。 总结一下,回答的结构应该是先定义logits,然后解释其在不同场景下的应用,结合实例说明,最后提到相关的注意事项和常见问题。同时,需要确保使用正确的中文表达,并正确引用提供的文献,尽管可能关联性不大,但可以适当提及相关模型中的使用情况。</think>### Logits的定义与用途 在机器学习中,**logits**是模型输出的原始未归一化分数,通常出现在分类任务的最后一层。具体来说,它们是激活函数(如softmax或sigmoid)处理前的数值,表示模型对每个类别的“置信度”。 #### 核心定义 - **数学形式**:对于多分类问题,假设有$K$个类别,模型最后一层输出为向量$\mathbf{z} = (z_1, z_2, \dots, z_K)$,这些$z_i$即为logits。通过softmax函数可将其转换为概率分布: $$ p_i = \frac{e^{z_i}}{\sum_{j=1}^K e^{z_j}} $$ #### 主要用途 1. **分类任务的基础** Logits通过softmax转换为概率后,用于交叉熵损失函数。例如,在自然语言处理(NLP)中,语言模型预测下一个词时,输出logits后计算损失[^2]。 2. **数值稳定性** 直接使用logits计算损失(如`tf.nn.softmax_cross_entropy_with_logits`)可避免数值溢出或下溢。例如: ```python # TensorFlow示例 loss = tf.nn.softmax_cross_entropy_with_logits(labels=y_true, logits=model_output) ``` 3. **生成对抗网络(GAN)** 判别器的输出常为logits,以区分真实样本与生成样本的置信度。 #### 与其他概念的对比 - **概率**:Logits是未归一化的分数,而概率通过softmax/sigmoid约束到$[0,1]$。 - **Log Probability**:Logits与对数概率的关系为$\log p_i = z_i - \log \sum e^{z_j}$。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

二分掌柜的

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值