用 PMI 监督点积:对比学习的表示损失新设计

serrjoa · x · 2026-10-21

Alan Baade 分享了一种表示学习损失的设计思路:对数据的两个视图 x 和 y,像对比学习一样嵌入为向量 R(x)、R(y),然后监督其点积等于二者的 PMI(逐点互信息)。PMI 由一个联合训练的掩码生成模型计算。配图给出了具体损失公式:LPMI = ||⟨R(x),R(y)⟩ − stopgrad(log PG(y|x) − log PG(y))||²,即用生成模型的条件对数概率差作为互信息估计,并以 stopgrad 防止梯度回传污染生成模型。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →