Graph Relation Embedding Network for Click-Through Rate Prediction

Wu Yixuan,Hu Youpeng,Xiong Xin,Li Xunkai,Guo Ronghui,Deng Shuiguang
DOI: https://doi.org/10.1007/s10115-022-01714-4
IF: 2.7
2022-01-01
Knowledge and Information Systems
Abstract:Most deep click-through rate (CTR) prediction models utilize a mainstream framework, which consists of the embedding layer and the feature interaction layer. Embeddings rich in semantic information directly benefit the downstream frameworks to mine potential information and achieve better performance. However, the embedding layer is rarely optimized in the CTR field. Although mapped into a low-dimensional embedding space, discrete features are still sparse. To solve this problem, we build graph structures to mine the similar interest of users and the co-occurrence relationship of items from click behavior sequences, and regard them as prior information for embedding optimization. For interpretable graph structures, we further propose graph relation embedding networks (GREENs), which utilize adapted order-wise graph convolution to alleviate the problems of data sparsity and over-smoothing. Moreover, we also propose a graph contrastive regularization module, which further normalizes graph embedding by maintaining certain graph structure information. Extensive experiments have proved that by introducing our embedding optimization methods, significant performance improvement is achieved.
What problem does this paper attempt to address?