Cargando…
Graph Attention Interaction Aggregation Network for Click-Through Rate Prediction
Click-through rate prediction is a critical task for computational advertising and recommendation systems, where the key challenge is to model feature interactions between different feature domains. At present, the main click-through rate prediction models model feature interactions in an implicit w...
Autores principales: | , , , |
---|---|
Formato: | Online Artículo Texto |
Lenguaje: | English |
Publicado: |
MDPI
2022
|
Materias: | |
Acceso en línea: | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9784643/ https://www.ncbi.nlm.nih.gov/pubmed/36560060 http://dx.doi.org/10.3390/s22249691 |
_version_ | 1784857858197684224 |
---|---|
author | Zhang, Wei Kang, Zhaobin Song, Lingling Qu, Kaiyuan |
author_facet | Zhang, Wei Kang, Zhaobin Song, Lingling Qu, Kaiyuan |
author_sort | Zhang, Wei |
collection | PubMed |
description | Click-through rate prediction is a critical task for computational advertising and recommendation systems, where the key challenge is to model feature interactions between different feature domains. At present, the main click-through rate prediction models model feature interactions in an implicit way, which leads to poor interpretation of the model, and the interaction between each pair of features may introduce noise into the model, thus limiting the predictive ability of the model. In response to the above problems, this paper proposes a click-through rate prediction model (GAIAN) based on the graph attention interactive aggregation network, which explicitly obtains cross features on the graph structure. Our specific method is to design a feature interactive selection mechanism to select cross features that are beneficial to model prediction, reducing model noise and reducing the risk of model overfitting. On this basis, the bilinear interaction function is integrated into the aggregation strategy of the graph neural network, and the fine-grained intersection features are extracted in a flexible and explicit way, which makes graph neural networks more suitable for modeling feature interactions and enhances the interpretability of the model. Compared with several other state-of-the-art models on the Criteo and Avazu datasets, the experimental results show the superiority of the model. |
format | Online Article Text |
id | pubmed-9784643 |
institution | National Center for Biotechnology Information |
language | English |
publishDate | 2022 |
publisher | MDPI |
record_format | MEDLINE/PubMed |
spelling | pubmed-97846432022-12-24 Graph Attention Interaction Aggregation Network for Click-Through Rate Prediction Zhang, Wei Kang, Zhaobin Song, Lingling Qu, Kaiyuan Sensors (Basel) Article Click-through rate prediction is a critical task for computational advertising and recommendation systems, where the key challenge is to model feature interactions between different feature domains. At present, the main click-through rate prediction models model feature interactions in an implicit way, which leads to poor interpretation of the model, and the interaction between each pair of features may introduce noise into the model, thus limiting the predictive ability of the model. In response to the above problems, this paper proposes a click-through rate prediction model (GAIAN) based on the graph attention interactive aggregation network, which explicitly obtains cross features on the graph structure. Our specific method is to design a feature interactive selection mechanism to select cross features that are beneficial to model prediction, reducing model noise and reducing the risk of model overfitting. On this basis, the bilinear interaction function is integrated into the aggregation strategy of the graph neural network, and the fine-grained intersection features are extracted in a flexible and explicit way, which makes graph neural networks more suitable for modeling feature interactions and enhances the interpretability of the model. Compared with several other state-of-the-art models on the Criteo and Avazu datasets, the experimental results show the superiority of the model. MDPI 2022-12-10 /pmc/articles/PMC9784643/ /pubmed/36560060 http://dx.doi.org/10.3390/s22249691 Text en © 2022 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/). |
spellingShingle | Article Zhang, Wei Kang, Zhaobin Song, Lingling Qu, Kaiyuan Graph Attention Interaction Aggregation Network for Click-Through Rate Prediction |
title | Graph Attention Interaction Aggregation Network for Click-Through Rate Prediction |
title_full | Graph Attention Interaction Aggregation Network for Click-Through Rate Prediction |
title_fullStr | Graph Attention Interaction Aggregation Network for Click-Through Rate Prediction |
title_full_unstemmed | Graph Attention Interaction Aggregation Network for Click-Through Rate Prediction |
title_short | Graph Attention Interaction Aggregation Network for Click-Through Rate Prediction |
title_sort | graph attention interaction aggregation network for click-through rate prediction |
topic | Article |
url | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9784643/ https://www.ncbi.nlm.nih.gov/pubmed/36560060 http://dx.doi.org/10.3390/s22249691 |
work_keys_str_mv | AT zhangwei graphattentioninteractionaggregationnetworkforclickthroughrateprediction AT kangzhaobin graphattentioninteractionaggregationnetworkforclickthroughrateprediction AT songlingling graphattentioninteractionaggregationnetworkforclickthroughrateprediction AT qukaiyuan graphattentioninteractionaggregationnetworkforclickthroughrateprediction |