论文笔记-DeepFM: A Factorization-Machine based Neural Network for CTR Prediction

烈酒焚心 提交于 2020-01-06 05:19:17

针对交叉(高阶)特征学习提出的DeepFM是一个end-to-end模型,不需要像wide&deep那样在wide端人工构造特征。

 

网络结构:

sparse features的构造:类别型特征one-hot,连续型特征数值表示,或者分段离散后one-hot

 

FM与NN分别输出预测y后,对两个结果进行sigmoid

 

FM部分:

paper指出在数据稀疏的情况下,FM仍能有效地学出二阶特征,最后FM的预测为:

 

deep部分:

papar指出这个网络结构两个特别的点:

1)虽然input的field vector长度不一,但是它们embedding出来的长度是固定的

2)FM的latent vector V向量作为原始特征到embedding vector的权重矩阵,放在网络里学习,如下图。

(个人理解,也就是说在网络的一二层,学出来的weight就是FM的latent vector,然后在FM component中使用,计算y_fromFM)

 

deepFM中FM与deep部分共享一份embedding数据,有两点好处:

1)能够从原始特征同时学习低阶和高阶特征

2)不需要像W&D做特征工程

 

paper之后比较了该模型与W&D在内的好几个模型,因为还没有读它们的原始论文,暂不记了。

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!