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...
Autores principales: | , , , |
---|---|
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 |