← Search

Neil Shah

31 accepted papers

2026

Plain Transformers are Surprisingly Powerful Link Predictors

ICML 2026poster

Link prediction is a core challenge in graph machine learning, demanding models that capture rich and complex topological dependencies. While Graph Neural Networks (GNNs) are the standard solution, state-of-the-art pipelines often rely on explicit structural heuristics or memory-intensive node embed…

Cited by 0SourceScholar
2025

A Pre-training Framework for Relational Data with Information-theoretic Principles

NeurIPS 2025poster

Relational databases underpin critical infrastructure across a wide range of domains, yet the design of generalizable pre-training strategies for learning from relational databases remains an open challenge due to task heterogeneity. Specifically, there exist many possible downstream tasks, as tasks…

Cited by 0SourcecodeScholar
2025

Haste Makes Waste: A Simple Approach for Scaling Graph Neural Networks

ICML 2025poster

Graph neural networks (GNNs) have demonstrated remarkable success in graph representation learning and various sampling approaches have been proposed to scale GNNs to applications with large-scale graphs. A class of promising GNN training algorithms take advantage of historical embeddings to reduce…

2025

Learning Along the Arrow of Time: Hyperbolic Geometry for Backward-Compatible Representation Learning

ICML 2025poster

Backward compatible representation learning enables updated models to integrate seamlessly with existing ones, avoiding to reprocess stored data. Despite recent advances, existing compatibility approaches in Euclidean space neglect the uncertainty in the old embedding models and force the new model…

Cited by 0SourcePDFScholar
2025

Mosaic of Modalities: A Comprehensive Benchmark for Multimodal Graph Learning

CVPR 2025poster

Graph machine learning has made significant strides in recent years, yet the integration of visual information with graph structure and its potential for improving performance in downstream tasks remains an underexplored area. To address this critical gap, we introduce the Multimodal Graph Benchmark…

2024

A Topological Perspective on Demystifying GNN-Based Link Prediction Performance

ICLR 2024poster

Graph Neural Networks (GNNs) have shown great promise in learning node embeddings for link prediction (LP). While numerous studies improve the overall GNNs' LP performance, none have explored their varying performance across different nodes and the underlying reasons. To this end, we demystify which…

2024

Explainability and Hate Speech: Structured Explanations Make Social Media Moderators Faster

ACL 2024short

Content moderators play a key role in keeping the conversation on social media healthy. While the high volume of content they need to judge represents a bottleneck to the moderation pipeline, no studies have explored how models could support them to make faster decisions. There is, by now, a vast bo…

2024

How Does Message Passing Improve Collaborative Filtering?

NeurIPS 2024poster

Collaborative filtering (CF) has exhibited prominent results for recommender systems and been broadly utilized for real-world applications. A branch of research enhances CF methods by message passing (MP) used in graph neural networks, due to its strong capabilities of extracting knowledge from grap…

2024

Position: Graph Foundation Models Are Already Here

ICML 2024spotlight

Graph Foundation Models (GFMs) are emerging as a significant research topic in the graph domain, aiming to develop graph models trained on extensive and diverse data to enhance their applicability across various tasks and domains. Developing GFMs presents unique challenges over traditional Graph Neu…

2024

Revisiting Link Prediction: a data perspective

ICLR 2024poster

Link prediction, a fundamental task on graphs, has proven indispensable in various applications, e.g., friend recommendation, protein analysis, and drug interaction prediction. However, since datasets span a multitude of domains, they could have distinct underlying mechanisms of link formation. Evid…

2023

Are Message Passing Neural Networks Really Helpful for Knowledge Graph Completion?

ACL 2023long

Knowledge graphs (KGs) facilitate a wide variety of applications. Despite great efforts in creation and maintenance, even the largest KGs are far from complete. Hence, KG completion (KGC) has become one of the most crucial tasks for KG research. Recently, considerable literature in this space has ce…

2023

Demystifying Structural Disparity in Graph Neural Networks: Can One Size Fit All?

NeurIPS 2023poster

Recent studies on Graph Neural Networks(GNNs) provide both empirical and theoretical evidence supporting their effectiveness in capturing structural patterns on both homophilic and certain heterophilic graphs. Notably, most real-world homophilic and heterophilic graphs are comprised of a mixture of…

2023

Empowering Graph Representation Learning with Test-Time Graph Transformation

ICLR 2023poster

As powerful tools for representation learning on graphs, graph neural networks (GNNs) have facilitated various applications from drug discovery to recommender systems. Nevertheless, the effectiveness of GNNs is immensely challenged by issues related to data quality, such as distribution shift, abnor…

2023

Evaluating Graph Neural Networks for Link Prediction: Current Pitfalls and New Benchmarking

NeurIPS 2023poster

Link prediction attempts to predict whether an unseen edge exists based on only a portion of the graph. A flurry of methods has been created in recent years that attempt to make use of graph neural networks (GNNs) for this task. Furthermore, new and diverse datasets have also been created to better…

2023

GraphPatcher: Mitigating Degree Bias for Graph Neural Networks via Test-time Augmentation

NeurIPS 2023poster

Recent studies have shown that graph neural networks (GNNs) exhibit strong biases towards the node degree: they usually perform satisfactorily on high-degree nodes with rich neighbor information but struggle with low-degree nodes. Existing works tackle this problem by deriving either designated GNN…

2023

Link Prediction with Non-Contrastive Learning

ICLR 2023poster

Graph neural networks (GNNs) are prominent in the graph machine learning domain, owing to their strong performance across various tasks. A recent focal area is the space of graph self-supervised learning (SSL), which aims to derive useful node representations without labeled data. Notably, many stat…

2023

Linkless Link Prediction via Relational Distillation

ICML 2023poster

Graph Neural Networks (GNNs) have shown exceptional performance in the task of link prediction. Despite their effectiveness, the high latency brought by non-trivial neighborhood data dependency limits GNNs in practical deployments. Conversely, the known efficient MLPs are much less effective than GN…

Cited by 64SourcePDFScholar
2023

MLPInit: Embarrassingly Simple GNN Training Acceleration with MLP Initialization

ICLR 2023poster

Training graph neural networks (GNNs) on large graphs is complex and extremely time consuming. This is attributed to overheads caused by sparse matrix multiplication, which are sidestepped when training multi-layer perceptrons (MLPs) with only node features. MLPs, by ignoring graph context, are simp…

2023

Multi-task Self-supervised Graph Neural Networks Enable Stronger Task Generalization

ICLR 2023poster

Self-supervised learning (SSL) for graph neural networks (GNNs) has attracted increasing attention from the graph machine learning community in recent years, owing to its capability to learn performant node embeddings without costly label information. One weakness of conventional SSL frameworks for…

2022

Automated Self-Supervised Learning for Graphs

ICLR 2022poster

Graph self-supervised learning has gained increasing attention due to its capacity to learn expressive node representations. Many pretext tasks, or loss functions have been designed from distinct perspectives. However, we observe that different pretext tasks affect downstream tasks differently cross…

2022

From Stars to Subgraphs: Uplifting Any GNN with Local Structure Awareness

ICLR 2022poster

Message Passing Neural Networks (MPNNs) are a common type of Graph Neural Network (GNN), in which each node’s representation is computed recursively by aggregating representations (“messages”) from its immediate neighbors akin to a star-shaped pattern. MPNNs are appealing for being efficient and sca…

2022

GStarX: Explaining Graph Neural Networks with Structure-Aware Cooperative Games

NeurIPS 2022accept

Explaining machine learning models is an important and increasingly popular area of research interest. The Shapley value from game theory has been proposed as a prime approach to compute feature importance towards model predictions on images, text, tabular data, and recently graph neural networks (G…

2022

Graph Condensation for Graph Neural Networks

ICLR 2022poster

Given the prevalence of large-scale graphs in real-world applications, the storage and time for training neural models have raised increasing concerns. To alleviate the concerns, we propose and study the problem of graph condensation for graph neural networks (GNNs). Specifically, we aim to condens…

2022

Graph-less Neural Networks: Teaching Old MLPs New Tricks Via Distillation

ICLR 2022poster

Graph Neural Networks (GNNs) are popular for graph machine learning and have shown great results on wide node classification tasks. Yet, they are less popular for practical deployments in the industry owing to their scalability challenges incurred by data dependency. Namely, GNN inference depends on…

2021

Data Augmentation for Graph Neural Networks

AAAI 2021technical

Data augmentation has been widely used to improve generalizability of machine learning models. However, comparatively little work studies data augmentation for graphs. This is largely due to the complex, non-Euclidean structure of graphs, which limits possible manipulation operations. Augmentation…

2020

The Devil is in the Details: Evaluating Limitations of Transformer-based Methods for Granular Tasks

COLING 2020main

Contextual embeddings derived from transformer-based neural language models have shown state-of-the-art performance for various tasks such as question answering, sentiment analysis, and textual similarity in recent years. Extensive work shows how accurately such models can represent abstract, semant…

2018

Time-Frequency Masking-Based Speech Enhancement Using Generative Adversarial Network

ICASSP 2018accepted

The success of time-frequency (T-F) mask-based approaches is dependent on the accuracy of predicted mask given the noisy spectral features. The state-of-the-art methods in T- F masking-based enhancement employ Deep Neural Network (DNN) to predict mask. Recently, Generative Adversarial Networks (GAN)…

Cited by 0SourceScholar