HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

TAGNN: Target Attentive Graph Neural Networks for Session-based Recommendation

Feng Yu Yanqiao Zhu Qiang Liu Shu Wu Liang Wang Tieniu Tan

TAGNN: Target Attentive Graph Neural Networks for Session-based Recommendation

Abstract

Session-based recommendation nowadays plays a vital role in many websites, which aims to predict users' actions based on anonymous sessions. There have emerged many studies that model a session as a sequence or a graph via investigating temporal transitions of items in a session. However, these methods compress a session into one fixed representation vector without considering the target items to be predicted. The fixed vector will restrict the representation ability of the recommender model, considering the diversity of target items and users' interests. In this paper, we propose a novel target attentive graph neural network (TAGNN) model for session-based recommendation. In TAGNN, target-aware attention adaptively activates different user interests with respect to varied target items. The learned interest representation vector varies with different target items, greatly improving the expressiveness of the model. Moreover, TAGNN harnesses the power of graph neural networks to capture rich item transitions in sessions. Comprehensive experiments conducted on real-world datasets demonstrate its superiority over state-of-the-art methods.

Code Repositories

CRIPAC-DIG/TAGNN
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
session-based-recommendations-on-digineticaTAGNN
Hit@20: 51.31
MRR@20: 18.03
session-based-recommendations-on-yoochoose1TAGNN
MRR@20: 31.12
Precision@20: 71.02
session-based-recommendations-on-yoochoose1-1TAGNN
HR@20: 71.02
MRR@20: 31.12

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
TAGNN: Target Attentive Graph Neural Networks for Session-based Recommendation | Papers | HyperAI