上一节展示了如何使用矩阵分解来学习嵌入。但矩阵分解存在一些局限性,包括:
深度神经网络(DNN)模型可以解决矩阵分解的这些局限性。DNN可以轻松地合并查询特征和项目特征(由于网络输入层的灵活性),这有助于捕获用户的特定兴趣并提高建议的相关性。
一种可能的DNN模型是softmax,它将问题看作多类预测问题,其中:
DNN的输入可包括:
与矩阵分解方法不同,还可以添加年龄或国家区域等侧面特征。这里用x表示输入向量。
模型架构决定了模型的复杂性和表现力。通过添加隐藏层和非线性激活函数(例如,ReLU),模型可以捕获数据中更复杂的关系。然而,增加参数的数量通常也使得模型更难以训练并且计算起来更复杂。最后一个隐藏层的输出用 ψ ( x ) ∈ R d \psi (x) \in \mathbb R^d ψ(x)∈Rd 表示。
模型映射最后一层的输出, ψ(X),通过softmax层到概率分布 p ^ = h ( ψ ( x ) V T ) \hat p = h(\psi(x) V^T) p^=h(ψ(x)VT),其中:
softmax层映射得分矢量 y ∈ R n y \in \mathbb R^n y∈Rn有时称为 (logits)为概率分布。
图3.预测的概率分布, p ^ = h ( ψ ( x ) V T ) \hat p = h(\psi(x) V^T) p^=h(ψ(x)VT)
你知道吗?
softmax这个名字是一个单词的游戏。“硬”(hard)表示将最大值的概率1分配给得分最高的项目 ÿ。相比之下,softmax为所有的项目分配非零概率,但是具有更高分数的项目其概率也越高。当 α→∞,分数缩放时,softmax h ( α y ) h(\alpha y) h(αy) 收敛到"硬"最大值。
最后,定义一个损失函数:
例如,可以使用交叉熵损失,因为正在比较两个概率分布。
项目 j j j的概率是 p ^ j = exp ( ⟨ ψ ( x ) , V j ⟩ ) Z \hat p_j = \frac{\exp(\langle \psi(x), V_j\rangle)}{Z} p^j=Zexp(⟨ψ(x),Vj⟩)得到,其中 Z Z Z 是一个不依赖 j j j的归一化常数。
换一种说法, log ( p ^ j ) = ⟨ ψ ( x ) , V j ⟩ − l o g ( Z ) \log(\hat p_j) = \langle \psi(x), V_j\rangle - log(Z) log(p^j)=⟨ψ(x),Vj⟩−log(Z),所以项目 j j j的对数概率 是(最多为加性常数)两个d维度向量的点积 ,可以将其解释为查询和项目的嵌入:
注意: 由于log函数是一个递增函数,项目 j j j概率最高的 p ^ j \hat p_j p^j 会获得最高点积的项目 ⟨ ψ ( x ) , V j ⟩ \langle \psi(x) , V_j\rangle ⟨ψ(x),Vj⟩。因此,点积可以被解释为该嵌入空间中的相似性度量。
图5.项目的嵌入 V j ∈ R d V_j \in \mathbb R^d Vj∈Rd
在softmax模型和矩阵分解模型中,系统都是对每一个项目 j j j学习嵌入向量 V j V_j Vj。在矩阵分解中,将其称之为项目嵌入矩阵 V ∈ R n × d V \in \mathbb R^{n \times d} V∈Rn×d;在DNN中,是通过softmax层的权重矩阵表示。
但是,查询嵌入的表示是不同的。在DNN中,不再是对每一个查询 i i i学习一个嵌入 U i U_i Ui,而是系统从查询特征中学习映射查询特征X为查询嵌入 ψ ( x ) ∈ R d \psi(x) \in \mathbb R^d ψ(x)∈Rd。因此,可以将DNN模型视为矩阵分解的泛化。
可以将相同的想法应用到学习项目嵌入上面吗?即,不是为每一个项目学习到一个嵌入,而是模型是否可以学习将项目特征映射到项目嵌入的非线性函数?答案是可以的。实现这一点需要使用双塔神经网络,该网络由两个神经网络组成:
模型的输出可以定义为点积 ⟨ ψ ( x query ) , ϕ ( x item ) ⟩ \langle \psi(x_{\text{query}}), \phi(x_{\text{item}}) \rangle ⟨ψ(xquery),ϕ(xitem)⟩。注意,这里不再是softmax模型。新模型预测每对 ( x query , x item ) (x_{\text{query}}, x_{\text{item}}) (xquery,xitem)值, 而不是每个查询 x q u e r y x_{query} xquery的概率向量。