arXiv:2505.17999cs.IRcs.LG2025-05KDD被引 8

用二次神经网络重新理解点击率预测中的特征交互机制

Revisiting Feature Interactions from the Perspective of Quadratic Neural Networks for Click-through Rate Prediction

  • 从二次神经网络视角解析哈达玛积的有效性
  • 提出QNN-alpha新结构,在6个数据集上达最优性能
  • 适合关注点击率预测模型设计的研究者

哈达玛积(HP)在点击率(CTR)预测中长期被广泛使用,因其简单高效且无需额外参数即可捕捉特征交互。然而其有效性背后的原理尚不明确。本文从二次神经网络(QNN)的角度重新审视HP,发现QNN能扩展特征空间并提供平滑非线性近似,且无需依赖激活函数。进一步实验表明,传统后激活无效,中段激活更优。通过对25种QNN神经元结构的理论分析与实证评估,识别出表现优异的变体并进行优化:提出多头Khatri-Rao积替代HP,设计具有动态集成能力的自集成损失函数,提升计算效率与性能。最终提出专为CTR任务设计的QNN-alpha神经元结构。实验显示,QNN-alpha在六个公开数据集上达到新最佳性能,同时保持低推理延迟、良好可扩展性和优异兼容性。代码、运行日志及超参数配置详见:https://github.com/salmon1802/QNN。

原文摘要 · Abstract (English)

Hadamard Product (HP) has long been a cornerstone in click-through rate (CTR) prediction tasks due to its simplicity, effectiveness, and ability to capture feature interactions without additional parameters. However, the underlying reasons for its effectiveness remain unclear. In this paper, we revisit HP from the perspective of Quadratic Neural Networks (QNN), which leverage quadratic interaction terms to model complex feature relationships. We further reveal QNN's ability to expand the feature space and provide smooth nonlinear approximations without relying on activation functions. Meanwhile, we find that traditional post-activation does not further improve the performance of the QNN. Instead, mid-activation is a more suitable alternative. Through theoretical analysis and empirical evaluation of 25 QNN neuron formats, we identify a good-performing variant and make further enhancements on it. Specifically, we propose the Multi-Head Khatri-Rao Product as a superior alternative to HP and a Self-Ensemble Loss with dynamic ensemble capability within the same network to enhance computational efficiency and performance. Ultimately, we propose a novel neuron format, QNN-alpha, which is tailored for CTR prediction tasks. Experimental results show that QNN-alpha achieves new state-of-the-art performance on six public datasets while maintaining low inference latency, good scalability, and excellent compatibility. The code, running logs, and detailed hyperparameter configurations are available at: https://github.com/salmon1802/QNN.

点击率预测二次神经网络特征交互模型优化

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