RAIN: Reconstructed-aware in-context enhancement with graph denoising for session-based recommendation.

Journal: Neural networks : the official journal of the International Neural Network Society
Published Date:

Abstract

Session-based recommendation aims to recommend the next item based on short-term interactions. Traditional session-based recommendation methods assume that all interacted items are closely related to the user's interests. However, noise (e.g., accidental clicks) is inevitably present in the interaction data. Existing methods for session-based recommendation only focus on graph denoising for enhanced item embeddings, resulting in sub-optimal session representation learning. To address these issues, we propose RAIN: Reconstructed-Aware In-context eNhancement with Graph Denoising for session-based recommendation. RAIN performs denoising on both the graph and session in a step-by-step manner. Guided by self-supervised signals, we aim to enhance the clarity of edges by employing masking and reconstruction alongside training an edge indicator to effectively eliminate noisy edges. By leveraging the trained indicator and incorporating a self-attentive mechanism, we additionally incorporate reconstructed-aware in-context enhancement within the session. Comparative evaluations with current state-of-the-art methods demonstrate that RAIN achieves significant improvements, with gains up to 7.05% in Hit@20 and 1.53% in MRR@20 on four benchmark datasets. The experimental results and analysis provide evidence for the rationality and superiority of our proposed model. The source code is available at https://github.com/zengxy20/RAIN.

Authors

  • Xinyi Zeng
    Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing, 100190, China; Key Laboratory of Network Information System Technology (NIST), Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing, 100190, China; University of Chinese Academy of Sciences, Beijing, 100190, China; School of Electronic, Electrical and Communication Engineering, University of Chinese Academy of Sciences, Beijing, 100190, China.
  • Shuchao Li
    Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing, 100190, China; Key Laboratory of Network Information System Technology (NIST), Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing, 100190, China. Electronic address: lisc@aircas.ac.cn.
  • Zequn Zhang
    Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing 100190, China.
  • Li Jin
    State Key Laboratory of Genetic Engineering and Innovation Center of Genetics and Development, School of Life Sciences, Fudan University, Shanghai, China.
  • Zhi Guo
    Academy of Integrative Medicine, Fujian University of Traditional Chinese Medicine, Fuzhou, 350122, China.
  • Kaiwen Wei
    Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing, 100190, China; Key Laboratory of Network Information System Technology (NIST), Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing, 100190, China; University of Chinese Academy of Sciences, Beijing, 100190, China; School of Electronic, Electrical and Communication Engineering, University of Chinese Academy of Sciences, Beijing, 100190, China.