Lightsage: Graph Neural Networks For Large Scale Item Retrieval In Shopee's Advertisement Recommendation

Abstract

Graph Neural Network (GNN) is the trending solution for item retrieval in recommendation problems. Most recent reports, however, focus heavily on new model architectures. This may bring some gaps when applying GNN in the industrial setup, where, besides the model, constructing the graph and handling data sparsity also play critical roles in the overall success of the project. In this work, we report how GNN is applied for large-scale e-commerce item retrieval at Shopee. We introduce our simple yet novel and impactful techniques in graph construction, modeling, and handling data skewness. Specifically, we construct high-quality item graphs by combining strong-signal user behaviors with high-precision collaborative filtering (CF) algorithm. We then develop a new GNN architecture named LightSAGE to produce high-quality items' embeddings for vector search. Finally, we design multiple strategies to handle cold-start and long-tail items, which are critical in an advertisement (ads) system. O

Stats

• citations7
• S2 citations—
• github stars0
• HF likes0
• heat score6.77
• arxiv keynguyen2023lightsage

Related papers

• Searchgcn: Powering Embedding Retrieval By Graph Convolution Networks For E-commerce Search (2021)4.52
• Neural Graph Matching For Video Retrieval In Large-scale Video-driven E-commerce (2024)0.00
• Gpu-accelerated Multi-relational Parallel Graph Retrieval For Web-scale Recommendations (2025)0.00
• Neural IR Meets Graph Embedding: A Ranking Model For Product Search (2019)11.85
• Hierarchical Structured Neural Network: Efficient Retrieval Scaling For Large Scale Recommendation (2024)0.00
• Hessian-aware Quantized Node Embeddings For Recommendation (2023)2.26
• Combigcn: An Effective GCN Model For Recommender System (2025)6.77
• Learning To Hash With Graph Neural Networks For Recommender Systems (2020)14.02