arXiv:2411.07508cs.IR2024-11被引 6

通过融合自蒸馏提升点击率预测的特征交互能力

Feature Interaction Fusion Self-Distillation Network For CTR Prediction

  • 在每一层连接显式与隐式特征交互,增强信息共享
  • 用深层网络作为教师指导浅层训练,提升知识传递效率
  • 在四个数据集上验证,对噪声具有更强鲁棒性

点击率(CTR)预测在推荐系统、在线广告和搜索引擎中至关重要。现有方法多采用堆叠或并行结构建模特征交互,部分使用知识蒸馏进行模型压缩。然而,这些方法存在三方面局限:(1)并行结构中显式与隐式交互独立执行,导致特征间信息共享不足;(2)知识蒸馏引入复杂师生框架设计,知识传递效率低;(3)高阶特征交互构建过程及数据集本身含显著噪声,限制模型性能。为此,本文提出FSDNet,一个集成即插即用融合自蒸馏模块的CTR预测框架。FSDNet在每层建立显式与隐式特征交互间的连接,增强特征间信息流动;以最深层融合层作为教师模型,通过自蒸馏引导浅层训练。在四个基准数据集上的实证评估验证了该框架的有效性与泛化能力。代码已公开于https://anonymous.4open.science/r/FSDNet。

原文摘要 · Abstract (English)

Click-Through Rate (CTR) prediction plays a vital role in recommender systems, online advertising, and search engines. Most of the current approaches model feature interactions through stacked or parallel structures, with some employing knowledge distillation for model compression. However, we observe some limitations with these approaches: (1) In parallel structure models, the explicit and implicit components are executed independently and simultaneously, which leads to insufficient information sharing within the feature set. (2) The introduction of knowledge distillation technology brings about the problems of complex teacher-student framework design and low knowledge transfer efficiency. (3) The dataset and the process of constructing high-order feature interactions contain significant noise, which limits the model's effectiveness. To address these limitations, we propose FSDNet, a CTR prediction framework incorporating a plug-and-play fusion self-distillation module. Specifically, FSDNet forms connections between explicit and implicit feature interactions at each layer, enhancing the sharing of information between different features. The deepest fusion layer is then used as the teacher model, utilizing self-distillation to guide the training of shallow layers. Empirical evaluation across four benchmark datasets validates the framework's efficacy and generalization capabilities. The code is available on https://anonymous.4open.science/r/FSDNet.

点击率预测特征交互自蒸馏推荐系统

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。