Search Results for author: Karthik Subbian

Found 23 papers, 13 papers with code

STaRK: Benchmarking LLM Retrieval on Textual and Relational Knowledge Bases

1 code implementation19 Apr 2024 Shirley Wu, Shiyu Zhao, Michihiro Yasunaga, Kexin Huang, Kaidi Cao, Qian Huang, Vassilis N. Ioannidis, Karthik Subbian, James Zou, Jure Leskovec

Answering real-world user queries, such as product search, often requires accurate retrieval of information from semi-structured knowledge bases or databases that involve blend of unstructured (e. g., textual descriptions of products) and structured (e. g., entity relations of products) information.

Benchmarking Retrieval

Graph Coarsening via Convolution Matching for Scalable Graph Neural Network Training

1 code implementation24 Dec 2023 Charles Dickens, Eddie Huang, Aishwarya Reganti, Jiong Zhu, Karthik Subbian, Danai Koutra

Notably, CONVMATCH achieves up to 95% of the prediction performance of GNNs on node classification while trained on graphs summarized down to 1% the size of the original graph.

Link Prediction Node Classification

ForeSeer: Product Aspect Forecasting Using Temporal Graph Embedding

no code implementations7 Oct 2023 Zixuan Liu, Gaurush Hiranandani, Kun Qian, Eddie W. Huang, Yi Xu, Belinda Zeng, Karthik Subbian, Sheng Wang

ForeSeer transfers reviews from similar products on a large product graph and exploits these reviews to predict aspects that might emerge in future reviews.

Graph Embedding Link Prediction

Communication-Free Distributed GNN Training with Vertex Cut

no code implementations6 Aug 2023 Kaidi Cao, Rui Deng, Shirley Wu, Edward W Huang, Karthik Subbian, Jure Leskovec

Here, we introduce CoFree-GNN, a novel distributed GNN training framework that significantly speeds up the training process by implementing communication-free training.

Node Classification

Simplifying Distributed Neural Network Training on Massive Graphs: Randomized Partitions Improve Model Aggregation

1 code implementation17 May 2023 Jiong Zhu, Aishwarya Reganti, Edward Huang, Charles Dickens, Nikhil Rao, Karthik Subbian, Danai Koutra

Backed by our theoretical analysis, instead of maximizing the recovery of cross-instance node dependencies -- which has been considered the key behind closing the performance gap between model aggregation and centralized training -- , our framework leverages randomized assignment of nodes or super-nodes (i. e., collections of original nodes) to partition the training graph such that it improves data uniformity and minimizes the discrepancy of gradient and loss function across instances.

You Only Transfer What You Share: Intersection-Induced Graph Transfer Learning for Link Prediction

1 code implementation27 Feb 2023 Wenqing Zheng, Edward W Huang, Nikhil Rao, Zhangyang Wang, Karthik Subbian

We identify this setting as Graph Intersection-induced Transfer Learning (GITL), which is motivated by practical applications in e-commerce or academic co-authorship predictions.

Link Prediction Transfer Learning

Search Behavior Prediction: A Hypergraph Perspective

1 code implementation23 Nov 2022 Yan Han, Edward W Huang, Wenqing Zheng, Nikhil Rao, Zhangyang Wang, Karthik Subbian

With these hyperedges, we augment the original bipartite graph into a new \textit{hypergraph}.

Link Prediction

TuneUp: A Simple Improved Training Strategy for Graph Neural Networks

no code implementations26 Oct 2022 Weihua Hu, Kaidi Cao, Kexin Huang, Edward W Huang, Karthik Subbian, Kenji Kawaguchi, Jure Leskovec

Extensive evaluation of TuneUp on five diverse GNN architectures, three types of prediction tasks, and both transductive and inductive settings shows that TuneUp significantly improves the performance of the base GNN on tail nodes, while often even improving the performance on head nodes.

Data Augmentation

Text Enriched Sparse Hyperbolic Graph Convolutional Networks

no code implementations6 Jul 2022 Nurendra Choudhary, Nikhil Rao, Karthik Subbian, Chandan K. Reddy

In TESH-GCN, we extract semantic node information, which successively acts as a connection signal to extract relevant nodes' local neighborhood and graph-level metapath features from the sparse adjacency tensor in a reformulated hyperbolic graph convolution layer.

Language Modelling Link Prediction

Shopping Queries Dataset: A Large-Scale ESCI Benchmark for Improving Product Search

1 code implementation14 Jun 2022 Chandan K. Reddy, Lluís Màrquez, Fran Valero, Nikhil Rao, Hugo Zaragoza, Sambaran Bandyopadhyay, Arnab Biswas, Anlu Xing, Karthik Subbian

This paper introduces the "Shopping Queries Dataset", a large dataset of difficult Amazon search queries and results, publicly released with the aim of fostering research in improving the quality of search results.

Learning Backward Compatible Embeddings

1 code implementation7 Jun 2022 Weihua Hu, Rajas Bansal, Kaidi Cao, Nikhil Rao, Karthik Subbian, Jure Leskovec

We formalize the problem where the goal is for the embedding team to keep updating the embedding version, while the consumer teams do not have to retrain their models.

Fraud Detection Product Recommendation +1

Multilingual Knowledge Graph Completion with Self-Supervised Adaptive Graph Alignment

1 code implementation ACL 2022 Zijie Huang, Zheng Li, Haoming Jiang, Tianyu Cao, Hanqing Lu, Bing Yin, Karthik Subbian, Yizhou Sun, Wei Wang

In this paper, we explore multilingual KG completion, which leverages limited seed alignment as a bridge, to embrace the collective knowledge from multiple languages.

Knowledge Graph Completion

Task-Agnostic Graph Explanations

1 code implementation16 Feb 2022 Yaochen Xie, Sumeet Katariya, Xianfeng Tang, Edward Huang, Nikhil Rao, Karthik Subbian, Shuiwang Ji

They are also unable to provide explanations in cases where the GNN is trained in a self-supervised manner, and the resulting representations are used in future downstream tasks.

Probabilistic Entity Representation Model for Reasoning over Knowledge Graphs

1 code implementation NeurIPS 2021 Nurendra Choudhary, Nikhil Rao, Sumeet Katariya, Karthik Subbian, Chandan K. Reddy

Current approaches employ spatial geometries such as boxes to learn query representations that encompass the answer entities and model the logical operations of projection and intersection.

Knowledge Graph Embedding Knowledge Graphs +1

Task-Agnostic Graph Neural Explanations

no code implementations29 Sep 2021 Yaochen Xie, Sumeet Katariya, Xianfeng Tang, Edward W Huang, Nikhil Rao, Karthik Subbian, Shuiwang Ji

TAGE enables the explanation of GNN embedding models without downstream tasks and allows efficient explanation of multitask models.

Scalable Feature Selection for (Multitask) Gradient Boosted Trees

no code implementations5 Sep 2021 Cuize Han, Nikhil Rao, Daria Sorokina, Karthik Subbian

Gradient Boosted Decision Trees (GBDTs) are widely used for building ranking and relevance models in search and recommendation.

feature selection

Self-Supervised Hyperboloid Representations from Logical Queries over Knowledge Graphs

1 code implementation23 Dec 2020 Nurendra Choudhary, Nikhil Rao, Sumeet Katariya, Karthik Subbian, Chandan K. Reddy

Promising approaches to tackle this problem include embedding the KG units (e. g., entities and relations) in a Euclidean space such that the query embedding contains the information relevant to its results.

Anomaly Detection Knowledge Graphs +2

Learning Robust Models for e-Commerce Product Search

no code implementations ACL 2020 Thanh V. Nguyen, Nikhil Rao, Karthik Subbian

Showing items that do not match search query intent degrades customer experience in e-commerce.

counterfactual

Learning Diffusion using Hyperparameters

no code implementations ICML 2018 Dimitris Kalimeris, Yaron Singer, Karthik Subbian, Udi Weinsberg

Despite this obstacle, we can shrink the best-known sample complexity bound for learning IC by a factor of |E|/d where |E| is the number of edges in the graph and d is the dimension of the hyperparameter.

Cannot find the paper you are looking for? You can Submit a new open access paper.