SWAG: Item Recommendations using Convolutions on Weighted Graphs

Abstract

Recent advancements in deep neural networks for graph-structured data have led to state-of-the-art performance on recommender system benchmarks. In this work, we present a Graph Convolutional Network (GCN) algorithm SWAG (Sample Weight and AGgregate), which combines efficient random walks and graph convolutions on weighted graphs to generate embeddings for nodes (items) that incorporate both graph structure as well as node feature information such as item-descriptions and item-images. The three important SWAG operations that enable us to efficiently generate node embeddings based on graph structures are (a) Sampling of graph to homogeneous structure, (b) Weighting the sampling, walks and convolution operations, and (c) using AGgregation functions for generating convolutions. The work is an adaptation of graphSAGE over weighted graphs. We deploy SWAG at Target and train it on a graph of more than 500K products sold online with over 50M edges. Offline and online evaluations reveal the benefit of using a graph-based approach and the benefits of weighing to produce high quality embeddings and product recommendations.

arXiv comment10 pages, 8 figures, 2019 IEEE BigData special session

References in corpus (15)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Distributed Representations of Words and Phrases and their Compositionality
• DeepWalk: Online Learning of Social Representations
• Semi-Supervised Classification with Graph Convolutional Networks
• LINE: Large-scale Information Network Embedding
• Geometric deep learning: going beyond Euclidean data
• Graph Convolutional Neural Networks for Web-Scale Recommender Systems
• Spectral Networks and Locally Connected Networks on Graphs
• Graph Embedding Techniques, Applications, and Performance: A Survey
• Modeling polypharmacy side effects with graph convolutional networks
• Graph Neural Networks: A Review of Methods and Applications
• Variational Graph Auto-Encoders
• Diffusion-Convolutional Neural Networks
• The History Began from AlexNet: A Comprehensive Survey on Deep Learning Approaches
• Geometric Matrix Completion with Recurrent Multi-Graph Neural Networks