STAM: A Spatiotemporal Aggregation Method for Graph Neural Network-based Recommendation

Zhen Yang, Ming Ding, Bin Xu, Hongxia Yang, Jie Tang

Research output: Chapter in book / Conference proceedingConference article published in proceeding or bookAcademic researchpeer-review

46 Citations (Scopus)

Abstract

Graph neural network-based recommendation systems are blossoming recently, and its core component is aggregation methods that determine neighbor embedding learning. Prior arts usually focus on how to aggregate information from the perspective of spatial structure information, but temporal information about neighbors is left insufficiently explored. In this work, we propose a spatiotemporal aggregation method STAM to efficiently incorporate temporal information into neighbor embedding learning. STAM generates spatiotemporal neighbor embeddings from the perspectives of spatial structure information and temporal information, facilitating the development of aggregation methods from spatial to spatiotemporal. STAM utilizes the Scaled Dot-Product Attention to capture temporal orders of one-hop neighbors and employs multi-head attention to perform joint attention over different latent subspaces. We utilize STAM for GNN-based recommendation to learn users and items embeddings. Extensive experiments demonstrate that STAM brings significant improvements on GNN-based recommendation compared with spatial-based aggregation methods, e.g., 24% for MovieLens, 8% for Amazon, and 13% for Taobao in terms of MRR@20.

Original languageEnglish
Title of host publicationWWW 2022 - Proceedings of the ACM Web Conference 2022
PublisherAssociation for Computing Machinery, Inc
Pages3217-3228
Number of pages12
ISBN (Electronic)9781450390965
DOIs
Publication statusPublished - 25 Apr 2022
Externally publishedYes
Event31st ACM Web Conference, WWW 2022 - Virtual, Lyon, France
Duration: 25 Apr 202229 Apr 2022

Publication series

NameWWW 2022 - Proceedings of the ACM Web Conference 2022

Conference

Conference31st ACM Web Conference, WWW 2022
Country/TerritoryFrance
CityVirtual, Lyon
Period25/04/2229/04/22

Keywords

  • GNN-based Recommendation
  • Self-Attention
  • Spatiotemporal Aggregation Method

ASJC Scopus subject areas

  • Computer Networks and Communications
  • Software

Fingerprint

Dive into the research topics of 'STAM: A Spatiotemporal Aggregation Method for Graph Neural Network-based Recommendation'. Together they form a unique fingerprint.

Cite this