OmniSage: Large Scale, Multi-Entity Heterogeneous Graph Representation Learning

OmniSage: Large Scale, Multi-Entity Heterogeneous Graph Representation Learning

22 April 2025

Anirudhan Badrinath

Prabhat Agarwal

Charles R. Rosenberg

Papers citing "OmniSage: Large Scale, Multi-Entity Heterogeneous Graph Representation Learning"

19 / 19 papers shown

Title
OmniSearchSage: Multi-Task Multi-Entity Embeddings for Pinterest Search Prabhat Agarwal SK MinhazulIslam Nikil Pancha Kurchi Subhra Hazra Jiajing Xu Chuck Rosenberg 55 5 0 25 Apr 2024
MLPInit: Embarrassingly Simple GNN Training Acceleration with MLP Initialization Xiaotian Han Tong Zhao Yozen Liu Xia Hu Neil Shah GNN 76 38 0 30 Sep 2022
ItemSage: Learning Product Embeddings for Shopping Recommendations at Pinterest Paul Baltescu Haoyu Chen Nikil Pancha Andrew Zhai J. Leskovec Charles R. Rosenberg DML 52 34 0 24 May 2022
Masked Autoencoders Are Scalable Vision Learners Kaiming He Xinlei Chen Saining Xie Yanghao Li Piotr Dollár Ross B. Girshick ViT TPM 345 7,600 0 11 Nov 2021
Billion-Scale Pretraining with Vision Transformers for Multi-Task Visual Representations Josh Beal Hao Wu Dong Huk Park Andrew Zhai Dmitry Kislyuk ViT 26 30 0 12 Aug 2021
Self-supervised Heterogeneous Graph Neural Network with Co-contrastive Learning Xiao Wang Nian Liu Hui-jun Han C. Shi SSL 50 379 0 19 May 2021
Scaling Graph Neural Networks with Approximate PageRank Aleksandar Bojchevski Johannes Klicpera Bryan Perozzi Amol Kapoor Martin J. Blais Benedek Rozemberczki Michal Lukasik Stephan Günnemann GNN 54 370 0 03 Jul 2020
Graph Enhanced Representation Learning for News Recommendation Suyu Ge Chuhan Wu Fangzhao Wu Tao Qi Yongfeng Huang AI4TS 32 123 0 31 Mar 2020
A Simple Framework for Contrastive Learning of Visual Representations Ting-Li Chen Simon Kornblith Mohammad Norouzi Geoffrey E. Hinton SSL 119 18,523 0 13 Feb 2020
PyTorch-BigGraph: A Large-scale Graph Embedding System Adam Lerer Ledell Yu Wu Jiajun Shen Timothée Lacroix Luca Wehrstedt Abhijit Bose A. Peysakhovich GNN 27 384 0 28 Mar 2019
AliGraph: A Comprehensive Graph Neural Network Platform Rong Zhu Kun Zhao Hongxia Yang Wei Lin Chang Zhou Baole Ai Yong Li Jingren Zhou GNN 67 387 0 23 Feb 2019
BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding Jacob Devlin Ming-Wei Chang Kenton Lee Kristina Toutanova VLM SSL SSeg 603 93,936 0 11 Oct 2018
Graph Convolutional Neural Networks for Web-Scale Recommender Systems Rex Ying Ruining He Kaifeng Chen Pong Eksombatchai William L. Hamilton J. Leskovec GNN BDL 151 3,513 0 06 Jun 2018
Graph Attention Networks Petar Velickovic Guillem Cucurull Arantxa Casanova Adriana Romero Pietro Lio Yoshua Bengio GNN 225 19,902 0 30 Oct 2017
Hash Embeddings for Efficient Word Representations Dan Svenstrup Jonas Meinertz Hansen Ole Winther 18 76 0 12 Sep 2017
Inductive Representation Learning on Large Graphs William L. Hamilton Z. Ying J. Leskovec 273 15,066 0 07 Jun 2017
Modeling Relational Data with Graph Convolutional Networks Michael Schlichtkrull Thomas Kipf Peter Bloem Rianne van den Berg Ivan Titov Max Welling GNN 114 4,772 0 17 Mar 2017
Semi-Supervised Classification with Graph Convolutional Networks Thomas Kipf Max Welling GNN SSL 310 28,795 0 09 Sep 2016
node2vec: Scalable Feature Learning for Networks Aditya Grover J. Leskovec 109 10,800 0 03 Jul 2016