skip to main content
10.1145/3448218.3448232acmotherconferencesArticle/Chapter ViewAbstractPublication PagescceaiConference Proceedingsconference-collections
short-paper

Improved CTR Prediction Algorithm based on LSTM and Attention

Authors Info & Claims
Published:15 February 2021Publication History

ABSTRACT

Click-through rate prediction is a core research issue in recommendation systems and online advertising, playing a pivotal role in the entire Internet field. This article mainly integrates the LSTM module and the attention module into the DeepFM framework, and proposes LDeepFM and LADeepFM which can extract time series information more effectively. We use real data sets to evaluate the prediction performance of the model. The experiment results demonstrate that our proposed approach can increase the accuracy of click-through rate prediction more effectively than traditional models.

References

  1. Richardson M, Dominowska E, Ragno R. Predicting clicks: estimating the click-through rate for new ads[C]//Proceedings of the 16th international conference on World Wide Web. 2007: 521--530.Google ScholarGoogle Scholar
  2. McMahan H B, Holt G, Sculley D, et al. Ad click prediction: a view from the trenches[C]//Proceedings of the 19th ACM SIGKDD international conference on Knowledge discovery and data mining. 2013: 1222--1230.Google ScholarGoogle Scholar
  3. McMahan H B, Holt G, Sculley D, et al. Ad click prediction: a view from the trenches[C]//Proceedings of the 19th ACM SIGKDD international conference on Knowledge discovery and data mining. 2013: 1222--1230.Google ScholarGoogle Scholar
  4. Blondel M, Ishihata M, Fujino A, et al. Polynomial networks and factorization machines: New insights and efficient training algorithms[J]. arXiv preprint arXiv:1607.08810, 2016.Google ScholarGoogle Scholar
  5. Rendle S. Factorization machines[C]//2010 IEEE International Conference on Data Mining. IEEE, 2010: 995--1000.Google ScholarGoogle Scholar
  6. Zhang W, Du T, Wang J. Deep learning over multi-field categorical data[C]//European conference on information retrieval. Springer, Cham, 2016: 45--57.Google ScholarGoogle Scholar
  7. Qu Y., Cai H., Ren K., et al. Product-based neural networks for user response prediction[C]. Data Mining (ICDM), 2016 IEEE 16th International Conference on, 2016, IEEE: 1149--1154Google ScholarGoogle ScholarCross RefCross Ref
  8. Cheng H T, Koc L, Harmsen J, et al. Wide & Deep Learning for Recommender Systems[J]. 2016.Google ScholarGoogle ScholarDigital LibraryDigital Library
  9. Guo H, Tang R, Ye Y, et al. DeepFM: a factorization-machine based neural network for CTR prediction[J]. arXiv preprint arXiv:1703.04247, 2017.Google ScholarGoogle Scholar
  10. Oentaryo R J, Lim E P, Low J W, et al. Predicting response in mobile advertising with hierarchical importance-aware factorization machine[C]//Proceedings of the 7th ACM international conference on Web search and data mining. 2014: 123--132.Google ScholarGoogle Scholar
  11. Juan Y, Zhuang Y, Chin W S, et al. Field-aware factorization machines for CTR prediction[C]//Proceedings of the 10th ACM Conference on Recommender Systems. 2016: 43--50.Google ScholarGoogle Scholar
  12. Graepel T, Candela J Q, Borchert T, et al. Web-scale bayesian click-through rate prediction for sponsored search advertising in microsoft's bing search engine[C]. Omnipress, 2010.Google ScholarGoogle Scholar
  13. Koren Y. Factorization meets the neighborhood: a multifaceted collaborative filtering model[C]//Proceedings of the 14th ACM SIGKDD international conference on Knowledge discovery and data mining. 2008: 426--434.Google ScholarGoogle Scholar
  14. Wang R, Fu B, Fu G, et al. Deep & cross network for ad click predictions[M]//Proceedings of the ADKDD'17. 2017: 1--7.Google ScholarGoogle Scholar
  15. Zaremba W, Sutskever I, Vinyals O. Recurrent neural network regularization[J]. arXiv preprint arXiv:1409.2329, 2014.Google ScholarGoogle Scholar
  16. Hochreiter S, Schmidhuber J. Long short-term memory[J]. Neural computation, 1997, 9(8): 1735--1780.Google ScholarGoogle ScholarDigital LibraryDigital Library
  17. Greff K, Srivastava R K, Koutník J, et al. LSTM: A search space odyssey[J]. IEEE transactions on neural networks and learning systems, 2016, 28(10): 2222--2232.Google ScholarGoogle Scholar
  18. Bahdanau D, Cho K, Bengio Y. Neural machine translation by jointly learning to align and translate[J]. arXiv preprint arXiv:1409.0473, 2014.Google ScholarGoogle Scholar
  19. Mnih V, Heess N, Graves A. Recurrent models of visual attention[C]//Advances in neural information processing systems. 2014: 2204--2212.Google ScholarGoogle Scholar
  20. Bahdanau D, Cho K, Bengio Y. Neural machine translation by jointly learning to align and translate[J]. arXiv preprint arXiv:1409.0473, 2014.Google ScholarGoogle Scholar
  21. Li L, Greene T, Hu B. A simple method to estimate the time-dependent receiver operating characteristic curve and the area under the curve with right censored data[J]. Statistical methods in medical research, 2018, 27(8): 2264--2278.Google ScholarGoogle Scholar
  22. Vovk V. The fundamental nature of the log loss function[M]//Fields of Logic and Computation II. Springer, Cham, 2015: 307--318.Google ScholarGoogle Scholar

Index Terms

  1. Improved CTR Prediction Algorithm based on LSTM and Attention

      Recommendations

      Comments

      Login options

      Check if you have access through your login credentials or your institution to get full access on this article.

      Sign in
      • Published in

        cover image ACM Other conferences
        CCEAI '21: Proceedings of the 5th International Conference on Control Engineering and Artificial Intelligence
        January 2021
        165 pages
        ISBN:9781450388870
        DOI:10.1145/3448218

        Copyright © 2021 ACM

        Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than ACM must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected]

        Publisher

        Association for Computing Machinery

        New York, NY, United States

        Publication History

        • Published: 15 February 2021

        Permissions

        Request permissions about this article.

        Request Permissions

        Check for updates

        Qualifiers

        • short-paper
        • Research
        • Refereed limited

      PDF Format

      View or Download as a PDF file.

      PDF

      eReader

      View online with eReader.

      eReader