arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2209.09481cs.LG
收藏

Feature embedding in click-through rate prediction

Zemanta, an Outbrain company(Zemanta(Outbrain旗下公司));Faculty of Computer and Information Science, University of Ljubljana(卢布尔雅那大学计算机与信息科学学院)

Samo Pahor, Davorin Kopič, Jure Demšar

英文摘要:

We tackle the challenge of feature embedding for the purposes of improving the click-through rate prediction process. We select three models: logistic regression, factorization machines and deep factorization machines, as our baselines and propose five different feature embedding modules: embedding scaling, FM embedding, embedding encoding, NN embedding and the embedding reweighting module. The embedding modules act as a way to improve baseline model feature embeddings and are trained alongside the rest of the model parameters in an end-to-end manner. Each module is individually added to a baseline model to obtain a new augmented model. We test the predictive performance of our augmented models on a publicly accessible dataset used for benchmarking click-through rate prediction models. Our results show that several proposed embedding modules provide an important increase in predictive performance without a drastic increase in training time.

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

↑