当前位置:   article > 正文

论文总结 (一):Fi-GNN——通过GNN对特征交互建模进行CTR预测

fi-gnn

一. CTR(Click-through rate)简介

  • CTR=实际点击数/点击率;
  • CTR预测是一个分类问题 ,预测用户是否点击目标项目;
  • 是一个重要任务在推荐系统和在线广告平台等

二. Fi-GNN

在这里插入图片描述
Fi-GNN

2.1 已有模型的缺点

特征之间的交互是灵活显性的,而无结构的特征组合无法很好地进行提取。

2.2 Fi-GNN的创新点

  • 在图结构上以更灵活和明确的方式对特征之间的复杂交互进行建模
  • 图(graph)中的每个节点(node)都对应一个特征(field),并且不同的特征可以通过边(edge)进行交互。 因此,可以将对特征之间的复杂交互转换为对特征图上的节点的交互
  • 在每个时间步,节点都与邻居进行更深一层的交互。 特征交互程度量等于交互步数。
  • 边的权重代表了不同特征进行交互的重要性,节点的权重代表了每个特征的重要性。

2.3 具体做法

在这里插入图片描述

  • 1. Embedding Layer

E = [ e 1 , e 2 , e 3 , … , e m ] , e i ∈ R d , E ∈ R d × m \textbf{E}=[ \textbf{e}_1, \textbf{e}_2, \textbf{e}_3, …, \textbf{e}_m],\textbf{e}_i \in \mathbb{R}^d,\textbf{E} \in \mathbb{R}^{d×m} E=[e1,e2,e3,,em],eiRd,ERd×m

  • 2. Multi-head Self-attention Layer

    通过多头注意力机制捕捉成对特征的复杂依赖
    H i = s o f t m a x i ( Q K T d K ) V , H i ∈ R m × d i \textbf{H}_i=softmax_i(\frac{\textbf{Q}\textbf{K}^T}{\sqrt{d_\textbf{K}}})\textbf{V},\textbf{H}_i\in \mathbb{R}^{m×{d_i}} Hi=softmaxi(dK QKT)V,HiRm×di
    Q = W i ( Q ) E , K = W i ( K ) E , V = W i ( V ) E \textbf{Q}={\textbf{W}_i}^{{(\textbf{Q})}}\textbf{E},\textbf{K}={\textbf{W}_i}^{{(\textbf{K})}}\textbf{E},\textbf{V}={\textbf{W}_i}^{{(\textbf{V})}}\textbf{E} Q=Wi(Q)E,K=Wi(K)E,V=Wi(V)E
    w h e r e    W i ( Q ) , W i ( K ) , W i ( V ) ∈ R d i × d where\ \ {\textbf{W}_i}^{{(\textbf{Q})}},{\textbf{W}_i}^{{(\textbf{K})}},{\textbf{W}_i}^{{(\textbf{V})}}\in \mathbb{R}^{d_i×d} where  Wi(Q),Wi(K),Wi(V)Rdi×d
    H 1 = R E L U ( H 1 ⊕ H 2 ⊕ . . . ⊕ H h ) \textbf{H}^1=RELU(H_1 \oplus H_2\oplus...\oplus H_h) H1=RELU(H1H2...Hh)

  • 3. 构建Feature Graph

    加权的全连接图,其中边的权重则反映了不同特征交互的重要性。

  • 4. Feature Interaction Graph Neural Network

在这里插入图片描述
在每个交互步(interaction step)中,每个节点聚合邻居节点的状态信息,然后根据其聚合信息和历史信息通过GRU和残差连接更新其节点状态。
加权的全连接图,其中边的权重则反映了不同特征交互的重要性。

  • 1). 状态聚合(在交互步骤t,每个节点将汇总来自邻居的状态信息)

    转换函数邻接矩阵决定节点的交互作用。
    在这里插入图片描述

    a. 邻接函数
    在这里插入图片描述

    b. 转换矩阵
    在这里插入图片描述

  • 2). 状态更新

在这里插入图片描述

  • 5. Attentional Scoring Layer (a graph-level output to predict CTR)

在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/运维做开发/article/detail/888578
推荐阅读
相关标签
  

闽ICP备14008679号