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

Full description

Bibliographic Details
Main Authors: Wei Zhang, Zhaobin Kang, Lingling Song, Kaiyuan Qu
Format: Article
Language:English
Published: MDPI AG 2022-12-01
Series:Sensors
Subjects:
Online Access:https://www.mdpi.com/1424-8220/22/24/9691
_version_ 1797455370151526400
author Wei Zhang
Zhaobin Kang
Lingling Song
Kaiyuan Qu
author_facet Wei Zhang
Zhaobin Kang
Lingling Song
Kaiyuan Qu
author_sort Wei Zhang
collection DOAJ
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.
first_indexed 2024-03-09T15:52:28Z
format Article
id doaj.art-4c6f778249ac4eb1b7bceb430eb36c3b
institution Directory Open Access Journal
issn 1424-8220
language English
last_indexed 2024-03-09T15:52:28Z
publishDate 2022-12-01
publisher MDPI AG
record_format Article
series Sensors
spelling doaj.art-4c6f778249ac4eb1b7bceb430eb36c3b2023-11-24T17:53:45ZengMDPI AGSensors1424-82202022-12-012224969110.3390/s22249691Graph Attention Interaction Aggregation Network for Click-Through Rate PredictionWei Zhang0Zhaobin Kang1Lingling Song2Kaiyuan Qu3Department of Artificial Intelligence Education, Central China Normal University, Wuhan 430079, ChinaDepartment of Artificial Intelligence Education, Central China Normal University, Wuhan 430079, ChinaDepartment of Artificial Intelligence Education, Central China Normal University, Wuhan 430079, ChinaDepartment of Artificial Intelligence Education, Central China Normal University, Wuhan 430079, ChinaClick-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.https://www.mdpi.com/1424-8220/22/24/9691click-through rate predictionrecommender systemsfeature interactiongraph neural networkattention mechanism
spellingShingle Wei Zhang
Zhaobin Kang
Lingling Song
Kaiyuan Qu
Graph Attention Interaction Aggregation Network for Click-Through Rate Prediction
Sensors
click-through rate prediction
recommender systems
feature interaction
graph neural network
attention mechanism
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 click-through rate prediction
recommender systems
feature interaction
graph neural network
attention mechanism
url https://www.mdpi.com/1424-8220/22/24/9691
work_keys_str_mv AT weizhang graphattentioninteractionaggregationnetworkforclickthroughrateprediction
AT zhaobinkang graphattentioninteractionaggregationnetworkforclickthroughrateprediction
AT linglingsong graphattentioninteractionaggregationnetworkforclickthroughrateprediction
AT kaiyuanqu graphattentioninteractionaggregationnetworkforclickthroughrateprediction