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...

Descripción completa

Detalles Bibliográficos
Autores principales: Zhang, Wei, Kang, Zhaobin, Song, Lingling, Qu, Kaiyuan
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