跳转到内容

搜索

Temperature & Top-p:调节大模型的创造力

拾知 2 min read

Temperature 调整不同词的概率差距,Top-p 控制入围门槛,去除长尾词。两者都是值越大输出结果越多样,值越小越输出结果稳定(如同不同温度下的分子运动)。

  • 大模型输出的三个步骤:生成分数、概率转换、加权采样,Temperature 影响概率转换环节的 Softmax 函数,调整不同词的概率差距,Top-p 作用在加权采样环节,控制入围门槛,去除长尾词
    • Temperature 与 Softmax 函数:Pi=ezi/τj=1Kezj/τP_i = \frac{e^{z_i / \tau}}{\sum_{j=1}^{K} e^{z_j / \tau}},其中,ziz_i 表示第 ii 个词的分数,kk 表示词的总个数,PiP_i 表示第 ii 个词出现的概率,τ\tau 表示温度系数。
    • Top-p 与加权采样:以 Top-p 为累加阈值丢掉长尾词后,对剩余词的概率同比例放大使得它们的和为100%后再进行加权采样。
  • 究竟调哪个、怎么调需要依据不同的情况、甚至不同的模型而定,比如 OpenAI 就建议只调其中一个就够了。

Reference