English

Feature embedding in click-through rate prediction

Machine Learning 2022-09-21 v1

Abstract

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.

Keywords

Cite

@article{arxiv.2209.09481,
  title  = {Feature embedding in click-through rate prediction},
  author = {Samo Pahor and Davorin Kopič and Jure Demšar},
  journal= {arXiv preprint arXiv:2209.09481},
  year   = {2022}
}

Comments

25 pages, 8 figures, 7 tables

R2 v1 2026-06-28T01:42:44.554Z