arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.17828cs.LGcs.AI
收藏

用于点击率预测模型快速训练的极限学习机

Extreme Learning Machines for Fast Training of Click-Through Rate Prediction Models

AI Enablement Huawei Türkiye R&D Center(华为土耳其研发中心AI赋能部门)

Ergun Biçici

AI总结:

本文提出一种结合嵌入层的极限学习机(ELM)模型用于点击率(CTR)预测,在Avazu和Criteo数据集上实现了具竞争力的F1结果并显著减少训练时间。

摘要:

极限学习机(ELM)为神经网络中传统的基于梯度的学习提供了一种快速的替代方案,具有训练迅速和泛化能力强的特点。其理论基础表明其具备通用近似能力。我们探讨了ELM在点击率(CTR)预测任务中的应用,由于该问题的高维特性,ELM在这一领域很大程度上尚未被探索。我们引入了一种通过嵌入层增强的基于ELM的模型,以提升在CTR任务上的性能,这是该领域的一项新颖补充。在包括Avazu和Criteo在内的基准数据集上的实验结果表明,我们提出的带有嵌入的ELM在取得具有竞争力的F1结果的同时,与Masknet等最先进的模型相比,显著减少了训练时间。我们的研究结果表明,ELM可用于CTR预测,尤其是在需要快速训练时。

英文摘要:

Extreme Learning Machines (ELM) provide a fast alternative to traditional gradient-based learning in neural networks, offering rapid training and robust generalization capabilities. Its theoretical basis shows its universal approximation capability. We explore the application of ELMs for the task of Click-Through Rate (CTR) prediction, which is largely unexplored by ELMs due to the high dimensionality of the problem. We introduce an ELM-based model enhanced with embedding layers to improve the performance on CTR tasks, which is a novel addition to the field. Experimental results on benchmark datasets, including Avazu and Criteo, demonstrate that our proposed ELM with embeddings achieves competitive F1 results while significantly reducing training time compared to state-of-the-art models such as Masknet. Our findings show that ELMs can be useful for CTR prediction, especially when fast training is needed.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑