首页 | 本学科首页   官方微博 | 高级检索  
     

基于域内特征间相似性的点击率预估优化
引用本文:雷李想,武志昊,刘钰,周子站.基于域内特征间相似性的点击率预估优化[J].计算机工程,2023,49(2):238-245.
作者姓名:雷李想  武志昊  刘钰  周子站
作者单位:1. 北京交通大学 计算机与信息技术学院, 北京 100044;2. 北京交通大学 网络科学与智能系统研究所, 北京 100044;3. 中国民航信息网络股份有限公司, 北京 101318;4. 民航旅客服务智能化应用技术重点实验室, 北京 101318
基金项目:国家自然科学基金(61603028);中国民航信息网络股份有限公司和民航旅客服务智能化应用技术重点实验室基金项目(K20L00070)。
摘    要:基于深度学习的点击率预估模型多数通过建模各个域的特征之间的交互关系提升预估准确率。特征嵌入向量对模型效果具有重要影响,而现有的CTR模型中不同特征的嵌入向量学习过程相互独立,且由于特征长尾分布导致大部分低频特征不能学习到较好的向量表示,严重影响模型的预测效果。基于域内特征间存在隐含的相似性,提出两种分别基于特征间共现概率和游走概率的相似度定义和对应的相似性图构建方法,并给出结合剪枝策略的广度优先遍历算法实现相似特征的高效计算。在此基础上,基于域内特征相似性图,设计一种嵌入生成器,对于低频特征,在域内特征相似性图上通过图神经网络聚合与其相似的特征信息,生成新的特征嵌入,作为预处理过程对特征嵌入向量进行数据增强,提升嵌入向量的表示学习质量。在公开数据集Criteo、Avazu上的实验结果表明,该方法明显提升点击率预估模型的预测准确率,其中对代表性点击率预估模型xDeepFM和AutoInt,AUC指标分别提升了0.007和0.008,LogLoss则下降了0.009和0.006,证明了嵌入生成模型的有效性。

关 键 词:点击率预估  稀疏特征  特征嵌入  特征相似性  图神经网络
收稿时间:2022-03-14
修稿时间:2022-04-23

Click-Through Rate Prediction and Optimization Based on Intra-Field Features Similarity
LEI Lixiang,WU Zhihao,LIU Yu,ZHOU Zizhan.Click-Through Rate Prediction and Optimization Based on Intra-Field Features Similarity[J].Computer Engineering,2023,49(2):238-245.
Authors:LEI Lixiang  WU Zhihao  LIU Yu  ZHOU Zizhan
Affiliation:1. School of Computer and Information Technology, Beijing Jiaotong University, Beijing 100044, China;2. Institute of Network Science and Intelligent System, Beijing Jiaotong University, Beijing 100044, China;3. TravelSky Technology Limited, Beijing 101318, China;4. Key Laboratory of Intelligent Application Technology for Civil Aviation Passenger Services, Beijing 101318, China
Abstract:There exist numerous deep learning-based Click-Through Rate (CTR) models;however, most of them improve prediction accuracy by modeling feature interaction between different fields.The feature embedding vectors have a significant impact on model performance; existing CTR models independently learn the embedding vectors of different features in a field.Consequently, most low-frequency features cannot attain sufficiently good embeddings because of the long-tail feature distribution, which seriously affects model accuracy.Noticing that implicit similarity exists between features of the same field, this study proposes a two intra-field feature similarity based on co-occurrence probability and walk probability, respectively.Moreover, the proposed method develops the corresponding similarity graph construction method and designs a breadth first traversal algorithm combined with pruning strategy to efficiently calculate similar features.Based on the intra-field feature similarity graph, an embedding generator is also proposed.For low-frequency features, information of similar features is aggregated on the similarity graph through a graph neural network.This data augmentation method is used as a preprocessing step to improve the learning quality of feature embedding vectors.Extensive experiments conducted on the public data sets Criteo and Avazu demonstrate that the proposed method improves the prediction accuracy of several classical CTR models.Regarding the representative CTR models xDeepFM and AutoInt, the AUC is increased by 0.007 and 0.008, respectively, while the LogLoss is decreased by 0.009 and 0.006, respectively, which proves the effectiveness of the embedding generator.
Keywords:Click-Through Rate(CTR) prediction  sparse feature  feature embedding  feature similarity  graph neural network  
点击此处可从《计算机工程》浏览原始摘要信息
点击此处可从《计算机工程》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号