Exploiting Intra and Inter-field Feature Interaction with Self-Attentive Network for CTR Prediction

Shenghao Zheng, Xuefeng Xian, Yongjing Hao, Victor S. Sheng, Zhiming Cui, Pengpeng Zhao

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Click-Through Rate (CTR) prediction models have achieved huge success mainly due to the ability to model arbitrary-order feature interactions. Recently, Self-Attention Network (SAN) has achieved significant success in CTR prediction. However, most of the existing SAN-based methods directly perform feature interaction operations on raw features. We argue that such operations, which ignore the intra-field information and inter-field affinity, are not designed to model richer feature interactions. In this paper, we propose an Intra and Inter-field Self-Attentive Network (IISAN) model for CTR prediction. Specifically, we first design an effective embedding block named Gated Fusion Layer (GFL) to refine raw features. Then, we utilize self-attention to model the feature interactions for each item to form meaningful high-order features via a multi-head attention mechanism. Next, we use the attention mechanism to aggregate all interactive embeddings. Finally, we assign DNNs in the prediction layer to generate the final output. Extensive experiments on three real public datasets show that IISAN achieves better performance than existing state-of-the-art approaches for CTR prediction.

Original languageEnglish
Title of host publicationWeb Information Systems Engineering - WISE 2021 - 22nd International Conference on Web Information Systems Engineering, WISE 2021, Proceedings
EditorsWenjie Zhang, Lei Zou, Zakaria Maamar, Lu Chen
PublisherSpringer Science and Business Media Deutschland GmbH
Pages34-49
Number of pages16
ISBN (Print)9783030915599
DOIs
StatePublished - 2021
Event22nd International Conference on Web Information Systems Engineering, WISE 2021 - Melbourne, Australia
Duration: Oct 26 2021Oct 29 2021

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13081 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference22nd International Conference on Web Information Systems Engineering, WISE 2021
Country/TerritoryAustralia
CityMelbourne
Period10/26/2110/29/21

Keywords

  • CTR prediction
  • Feature interaction
  • Recommender system
  • Self-attentive network

Fingerprint

Dive into the research topics of 'Exploiting Intra and Inter-field Feature Interaction with Self-Attentive Network for CTR Prediction'. Together they form a unique fingerprint.

Cite this