Graph Masked Autoencoder for Sequential Recommendation

Yaowen Ye, Lianghao Xia, Chao Huang
5/8/2023
cs.IRcs.AI

Official code: HKUDS/MAERec (66 stars, last commit 2023-06-01).

Abstract

While some powerful neural network architectures (e.g., Transformer, Graph Neural Networks) have achieved improved performance in sequential recommendation with high-order item dependency modeling, they may suffer from poor representation capability in label scarcity scenarios. To address the issue of insufficient labels, Contrastive Learning (CL) has attracted much attention in recent methods to perform data augmentation through embedding contrasting for self-supervision. However, due to the hand-crafted property of their contrastive view generation strategies, existing CL-enhanced models i) can hardly yield consistent performance on diverse sequential recommendation tasks; ii) may not be immune to user behavior data noise. In light of this, we propose a simple yet effective Graph Masked AutoEncoder-enhanced sequential Recommender system (MAERec) that adaptively and dynamically distills global item transitional information for self-supervised augmentation. It naturally avoids the above issue of heavy reliance on constructing high-quality embedding contrastive views. Instead, an adaptive data reconstruction paradigm is designed to be integrated with the long-range item dependency modeling, for informative augmentation in sequential recommendation. Extensive experiments demonstrate that our method significantly outperforms state-of-the-art baseline models and can learn more accurate representations against data noise and sparsity. Our implemented model code is available at https://github.com/HKUDS/MAERec.

View on arXivView PDF

Code Implementations(1)

HKUDS/MAERecOfficial100%

[SIGIR'2023] "MAERec: Graph Masked Autoencoder for Sequential Recommendation"

667Python3 years ago
graph-neural-networksmasked-autoencoderrecommender-systemsself-supervised-learningsequential-recommendation

Cite this paper

@article{ye2023graph,
  title  = {Graph Masked Autoencoder for Sequential Recommendation},
  author = {Yaowen Ye and Lianghao Xia and Chao Huang},
  year   = {2023},
  eprint = {2305.04619},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2305.04619v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2305.04619.svg)](https://codeofpaper.com/paper/2305.04619)

Discussion