Cargando…

An adaptive hybrid XdeepFM based deep Interest network model for click-through rate prediction system

Recent advances in communication enable individuals to use phones and computers to access information on the web. E-commerce has seen rapid development, e.g., Alibaba has nearly 12 hundred million customers in China. Click-Through Rate (CTR) forecasting is a primary task in the e-commerce advertisem...

Descripción completa

Detalles Bibliográficos
Autores principales: Lu, Qiao, Li, Silin, Yang, Tuo, Xu, Chenheng
Formato: Online Artículo Texto
Lenguaje:English
Publicado: PeerJ Inc. 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8459778/
https://www.ncbi.nlm.nih.gov/pubmed/34616892
http://dx.doi.org/10.7717/peerj-cs.716
_version_ 1784571597804273664
author Lu, Qiao
Li, Silin
Yang, Tuo
Xu, Chenheng
author_facet Lu, Qiao
Li, Silin
Yang, Tuo
Xu, Chenheng
author_sort Lu, Qiao
collection PubMed
description Recent advances in communication enable individuals to use phones and computers to access information on the web. E-commerce has seen rapid development, e.g., Alibaba has nearly 12 hundred million customers in China. Click-Through Rate (CTR) forecasting is a primary task in the e-commerce advertisement system. From the traditional Logistic Regression algorithm to the latest popular deep neural network methods that follow a similar embedding and MLP, several algorithms are used to predict CTR. This research proposes a hybrid model combining the Deep Interest Network (DIN) and eXtreme Deep Factorization Machine (xDeepFM) to perform CTR prediction robustly. The cores of DIN and xDeepFM are attention and feature cross, respectively. DIN follows an adaptive local activation unit that incorporates the attention mechanism to adaptively learn user interest from historical behaviors related to specific advertisements. xDeepFM further includes a critical part, a Compressed Interactions Network (CIN), aiming to generate feature interactions at a vectorwise level implicitly. Furthermore, a CIN, plain DNN, and a linear part are combined into one unified model to form xDeepFM. The proposed end-to-end hybrid model is a parallel ensemble of models via multilayer perceptron. CIN and xDeepFM are trained in parallel, and their output is fed into a multilayer perceptron. We used the e-commerce Alibaba dataset with the focal loss as the loss function for experimental evaluation through online complex example mining (OHEM) in the training process. The experimental result indicates that the proposed hybrid model has better performance than other models.
format Online
Article
Text
id pubmed-8459778
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher PeerJ Inc.
record_format MEDLINE/PubMed
spelling pubmed-84597782021-10-05 An adaptive hybrid XdeepFM based deep Interest network model for click-through rate prediction system Lu, Qiao Li, Silin Yang, Tuo Xu, Chenheng PeerJ Comput Sci Agents and Multi-Agent Systems Recent advances in communication enable individuals to use phones and computers to access information on the web. E-commerce has seen rapid development, e.g., Alibaba has nearly 12 hundred million customers in China. Click-Through Rate (CTR) forecasting is a primary task in the e-commerce advertisement system. From the traditional Logistic Regression algorithm to the latest popular deep neural network methods that follow a similar embedding and MLP, several algorithms are used to predict CTR. This research proposes a hybrid model combining the Deep Interest Network (DIN) and eXtreme Deep Factorization Machine (xDeepFM) to perform CTR prediction robustly. The cores of DIN and xDeepFM are attention and feature cross, respectively. DIN follows an adaptive local activation unit that incorporates the attention mechanism to adaptively learn user interest from historical behaviors related to specific advertisements. xDeepFM further includes a critical part, a Compressed Interactions Network (CIN), aiming to generate feature interactions at a vectorwise level implicitly. Furthermore, a CIN, plain DNN, and a linear part are combined into one unified model to form xDeepFM. The proposed end-to-end hybrid model is a parallel ensemble of models via multilayer perceptron. CIN and xDeepFM are trained in parallel, and their output is fed into a multilayer perceptron. We used the e-commerce Alibaba dataset with the focal loss as the loss function for experimental evaluation through online complex example mining (OHEM) in the training process. The experimental result indicates that the proposed hybrid model has better performance than other models. PeerJ Inc. 2021-09-17 /pmc/articles/PMC8459778/ /pubmed/34616892 http://dx.doi.org/10.7717/peerj-cs.716 Text en ©2021 Lu et al. https://creativecommons.org/licenses/by/4.0/This is an open access article distributed under the terms of the Creative Commons Attribution License (https://creativecommons.org/licenses/by/4.0/) , which permits unrestricted use, distribution, reproduction and adaptation in any medium and for any purpose provided that it is properly attributed. For attribution, the original author(s), title, publication source (PeerJ Computer Science) and either DOI or URL of the article must be cited.
spellingShingle Agents and Multi-Agent Systems
Lu, Qiao
Li, Silin
Yang, Tuo
Xu, Chenheng
An adaptive hybrid XdeepFM based deep Interest network model for click-through rate prediction system
title An adaptive hybrid XdeepFM based deep Interest network model for click-through rate prediction system
title_full An adaptive hybrid XdeepFM based deep Interest network model for click-through rate prediction system
title_fullStr An adaptive hybrid XdeepFM based deep Interest network model for click-through rate prediction system
title_full_unstemmed An adaptive hybrid XdeepFM based deep Interest network model for click-through rate prediction system
title_short An adaptive hybrid XdeepFM based deep Interest network model for click-through rate prediction system
title_sort adaptive hybrid xdeepfm based deep interest network model for click-through rate prediction system
topic Agents and Multi-Agent Systems
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8459778/
https://www.ncbi.nlm.nih.gov/pubmed/34616892
http://dx.doi.org/10.7717/peerj-cs.716
work_keys_str_mv AT luqiao anadaptivehybridxdeepfmbaseddeepinterestnetworkmodelforclickthroughratepredictionsystem
AT lisilin anadaptivehybridxdeepfmbaseddeepinterestnetworkmodelforclickthroughratepredictionsystem
AT yangtuo anadaptivehybridxdeepfmbaseddeepinterestnetworkmodelforclickthroughratepredictionsystem
AT xuchenheng anadaptivehybridxdeepfmbaseddeepinterestnetworkmodelforclickthroughratepredictionsystem
AT luqiao adaptivehybridxdeepfmbaseddeepinterestnetworkmodelforclickthroughratepredictionsystem
AT lisilin adaptivehybridxdeepfmbaseddeepinterestnetworkmodelforclickthroughratepredictionsystem
AT yangtuo adaptivehybridxdeepfmbaseddeepinterestnetworkmodelforclickthroughratepredictionsystem
AT xuchenheng adaptivehybridxdeepfmbaseddeepinterestnetworkmodelforclickthroughratepredictionsystem