60 秒回答模板

InfoNCE 会把一个 query 和正样本、多个负样本的相似度放进 softmax,温度系数通常写在相似度除以 tau 的位置。tau 小时,相似度差异会被放大,softmax 更尖锐,模型更关注最难负样本,梯度更集中,但过小容易训练不稳定或过拟合噪声。tau 大时,分布更平滑,负样本权重更平均,训练更温和,但可能区分度不足。实际选择温度要结合 embedding 归一化、batch 内负样本数量、相似度尺度、数据噪声和下游指标调参。

考点 控制分布尖锐度
难度 真实面经题
回答目标 讲清原理、实现和边界

深入解析

01

温度进入 softmax

InfoNCE 的分母是正负样本相似度的 softmax 归一化。把相似度除以 tau 后,tau 直接改变 logits 的尺度,从而改变概率分布和梯度分配。

02

小温度更尖锐

tau 越小,相似度差距越被放大,最像正样本的负样本会获得更高权重。这样能强化判别性,但如果负样本有假负例或噪声,模型也会更容易被误导。

03

大温度更平滑

tau 越大,正负样本概率差距被压平,训练信号更平均,优化过程通常更稳定。但温度过大时,模型对难负样本不够敏感,embedding 聚类边界可能不够清晰。

04

与负样本数量耦合

batch 内负样本越多,出现难负样本和假负样本的概率越高,温度需要配合 batch size、采样策略和去噪策略一起调,而不是固定套一个经验值。

05

调参看下游指标

温度不是只看训练 loss。推荐、检索或匹配场景要看 Recall、NDCG、AUC、线上点击转化和 embedding 分布,必要时还要按用户或物品长尾切片评估。

易错点

  • 只说温度是超参数,不解释它如何改变 softmax 和梯度。
  • 认为温度越小越好,忽略噪声和假负例会被放大。
  • 脱离 batch size、负样本采样和相似度尺度谈温度。
  • 只看训练 loss,不看召回、排序和线上效果。

面试官追问

温度过小会有什么问题?

softmax 过尖,梯度集中在少数样本上,遇到假负例或噪声会更不稳定,也可能导致表示空间过度分离。

温度过大会有什么问题?

正负样本差异被抹平,模型对难负样本不敏感,学到的 embedding 判别性可能不足。

embedding 做归一化后温度还重要吗?

仍然重要。归一化限制了相似度范围,温度继续控制 logits 尺度和 softmax 梯度分配。

推荐场景如何选择温度?

用验证集和线上指标调参,同时观察召回、排序指标、热门长尾切片和假负样本影响。