召回06 双塔模型two-tower(DSSM)

双塔模型的结构

相比于矩阵补充模型,双塔模型考虑了id之外的多种特征作为输入。

对于每个离散特征用单独的enbadding层得到向量,对于性别等类别较少的特征直接使用one-hot编码;对于连续特征,最简单的直接归一化,对于长尾分布的特征取log或做分桶等。

神经网络可以是简单的全连接网络,或更复杂的结构,例如深度交叉网络等。输出向量,即表征,用于召回。

余弦相似度范围 [-1 ,1],相当于对两个向量a b归一化然后求内积。

双塔模型的训练

pointwise: 把正样本和负样本组成一个数据集,在数据集上做随机梯度下降训练双塔模型;

pairwise: 每次取一个正样本和负样本组成一个二元组,损失函数用triplet hinge loss或者 triplet logistic loss;

listwise: 每次取一个正样本和多个负样本组成一个list,训练类似于多元分类。

pointwise训练

正负样本数量约定俗成

pairwise训练

正负样本embadding层使用一样的参数

m是超参数,可以调整,例如设置为1。训练就是最小化损失函数,用梯度更新双塔神经网络的参数,triplet hinge loss是一种损失函数;triplet logistic loss函数中的sigma是一个大于0的超参数,控制损失函数的形状,手动设置。

listwise训练:

鼓励正样本余弦相似度接近1,负样本都接近于-1。输入到softmax激活函数,输出分数s介于0-1之间。希望s+接近1,s-接近0,鼓励s接近y。用y和s的交叉熵-logs+作为损失函数,训练来最小化交叉熵,即鼓励softmax输出s接近标签y,也就是最大化s+。

总结:

这是粗排和精排,前期融合,将特征在神经网络之前就拼接融合,不适用于召回,适用于排序模型;召回使用双塔模型,是在最终输出相似度的时候才融合。

### DSSM双塔模型架构与实现 #### 架构概述 DSSM(Deep Structured Semantic Model)是一种经典的双塔模型,最初由微软提出并应用于搜索引擎领域[^2]。该模型的核心思想是通过神经网络学习查询(query)和文档(document)之间的语义相似度。具体来说,DSSM将输入的文本转换为低维向量表示,并计算它们之间的余弦相似度作为匹配分数。 #### 模型结构 DSSM的主要组成部分包括以下几个层次: 1. **Embedding 层** 输入通常是稀疏的词袋(Bag-of-Words, BoW)或字符 n-gram 特征。这些特征会被映射到密集的嵌入空间中[^3]。例如,对于单词序列 `w_1, w_2, ..., w_n`,可以将其转化为一组 one-hot 向量,再通过嵌入矩阵得到连续向量表示。 2. **隐藏层** 嵌入后的向量会经过多层全连接神经网络(Fully Connected Layers),逐步提取高层次的语义特征[^1]。每一层都会引入非线性激活函数(如 ReLU 或 tanh),从而捕捉复杂的模式关系。 3. **规范化处理** 在最终输出之前,通常会对隐含层的结果应用 L2 归一化操作,确保生成的向量长度一致,便于后续计算余弦距离[^4]。 4. **相似度计算** 查询和文档分别通过各自的塔传递后获得两个固定维度的向量 \( \vec{q} \) 和 \( \vec{d} \),接着利用如下公式衡量两者间的接近程度: \[ sim(q,d)=\frac{\vec{q}\cdot\vec{d}}{\|\vec{q}\|_{L2}\times\|\vec{d}\|_{L2}} \] #### 训练策略 针对排序任务,常采用 pairwise 方法构建训练样本集[^5]。即选取每一对包含正反馈 (clicked document) 和负反馈 (unclicked document) 的实例构成对比对子。损失函数可以选择基于 margin 的 hinge loss 或者 logistic regression 类型的形式表达如下: \[ loss=\sum_i max(0,m-sim(q,p)+sim(q,n)) \] 其中 m 表示预设的安全边界参数;p/n 分别代表正面/负面样例。 以下是 Python 实现的一个简化版本代码片段展示如何搭建这样的框架: ```python import tensorflow as tf from tensorflow.keras import layers def create_tower(input_dim, embedding_dim=128): input_layer = layers.Input(shape=(input_dim,)) embedded = layers.Embedding(input_dim=input_dim+1, output_dim=embedding_dim)(input_layer) flatten = layers.Flatten()(embedded) dense1 = layers.Dense(256, activation='relu')(flatten) dropout1 = layers.Dropout(0.5)(dense1) dense2 = layers.Dense(128, activation='tanh')(dropout1) norm = tf.math.l2_normalize(dense2, axis=-1) model = tf.keras.Model(inputs=[input_layer], outputs=[norm]) return model # Define Query Tower and Document Tower with same structure but different weights. query_input_size = vocab_size_for_queries # Replace this value accordingly. doc_input_size = vocab_size_for_documents # And replace this. query_model = create_tower(query_input_size) doc_model = create_tower(doc_input_size) # Combine both towers into a single model for training purposes. positive_doc_output = doc_model([pos_doc_inputs]) # Assume pos_doc_inputs defined elsewhere. negative_doc_output = doc_model([neg_doc_inputs]) cosine_sim_pos = tf.reduce_sum(tf.multiply(query_model.output, positive_doc_output), axis=-1) cosine_sim_neg = tf.reduce_sum(tf.multiply(query_model.output, negative_doc_output), axis=-1) margin_loss = tf.maximum(cosine_sim_neg - cosine_sim_pos + MARGIN_CONSTANT, 0.) total_loss = tf.reduce_mean(margin_loss) train_step = tf.optimizers.Adam().minimize(total_loss, var_list=query_model.trainable_variables + doc_model.trainable_variables) ``` 上述脚本定义了一个基础版的双通道网络及其相应的优化流程。实际部署时可能还需要考虑更多细节调整以及超参寻优等问题。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值