activity
20212026
most citedEDITS: Modeling and Mitigating Data Bias for Graph Neural Networks

128 citations · 405 across the 50 of their papers we have counts for

collaborators
Showing cs.LGShow all

35 papers · 1 filter

cs.LG2026

A Mechanistic Study of Automated Failure Attribution in LLM-based Agentic Systems

Wendy Zheng, Liang Wu, Yinhan He +3

Agentic systems have demonstrated strong capabilities on complex tasks by enabling multiple interacting large language model (LLM)-based agents to collaborate and execute subtasks…

cs.LG2025

Interpretable Neuropsychiatric Diagnosis via Concept-Guided Graph Neural Networks

Song Wang, Zhenyu Lei, Zhen Tan +4

Nearly one in five adolescents currently live with a diagnosed mental or behavioral health condition, such as anxiety, depression, or conduct disorder, underscoring the urgency of…

cs.LG2025

Are We Merely Justifying Results ex Post Facto? Quantifying Explanatory Inversion in Post-Hoc Model Explanations

Zhen Tan, Song Wang, Yifan Li +4

Post-hoc explanation methods provide interpretation by attributing predictions to input features. Natural explanations are expected to interpret how the inputs lead to the predicti…

cs.LG2025

Reasoning of Large Language Models over Knowledge Graphs with Super-Relations

Song Wang, Junhong Lin, Xiaojie Guo +3

While large language models (LLMs) have made significant progress in processing and reasoning over knowledge graphs, current methods suffer from a high non-retrieval rate. This lim…

cs.LG2025

Enhancing Distribution and Label Consistency for Graph Out-of-Distribution Generalization

Song Wang, Xiaodong Yang, Rashidul Islam +4

To deal with distribution shifts in graph data, various graph out-of-distribution (OOD) generalization techniques have been recently proposed. These methods often employ a two-step…

cs.LG2024

ST-FiT: Inductive Spatial-Temporal Forecasting with Limited Training Data

Zhenyu Lei, Yushun Dong, Jundong Li +1

Spatial-temporal graphs are widely used in a variety of real-world applications. Spatial-Temporal Graph Neural Networks (STGNNs) have emerged as a powerful tool to extract meaningf…

cs.LG2024★ 1 cited

Graph Neural Networks Are More Than Filters: Revisiting and Benchmarking from A Spectral Perspective

Yushun Dong, Patrick Soga, Yinhan He +2

Graph Neural Networks (GNNs) have achieved remarkable success in various graph-based learning tasks. While their performance is often attributed to the powerful neighborhood aggreg…

cs.LG2024

Virtual Nodes Can Help: Tackling Distribution Shifts in Federated Graph Learning

Xingbo Fu, Zihan Chen, Yinhan He +4

Federated Graph Learning (FGL) enables multiple clients to jointly train powerful graph learning models, e.g., Graph Neural Networks (GNNs), without sharing their local graph data…

cs.LG2024

Federated Graph Learning with Graphless Clients

Xingbo Fu, Song Wang, Yushun Dong +3

Federated Graph Learning (FGL) is tasked with training machine learning models, such as Graph Neural Networks (GNNs), for multiple clients, each with its own graph data. Existing m…

cs.LG2024

Explaining Graph Neural Networks with Large Language Models: A Counterfactual Perspective for Molecular Property Prediction

Yinhan He, Zaiyi Zheng, Patrick Soga +3

In recent years, Graph Neural Networks (GNNs) have become successful in molecular property prediction tasks such as toxicity analysis. However, due to the black-box nature of GNNs,…

cs.LG2024

A Survey of Deep Graph Learning under Distribution Shifts: from Graph Out-of-Distribution Generalization to Adaptation

Kexin Zhang, Shuhan Liu, Song Wang +6

Distribution shifts on graphs -- the discrepancies in data distribution between training and employing a graph machine learning model -- are ubiquitous and often unavoidable in rea…

cs.LG2024

Federated Graph Learning with Structure Proxy Alignment

Xingbo Fu, Zihan Chen, Binchi Zhang +2

Federated Graph Learning (FGL) aims to learn graph learning models over graph data distributed in multiple data owners, which has been applied in various applications such as socia…

cs.LG2024

Towards Certified Unlearning for Deep Neural Networks

Binchi Zhang, Yushun Dong, Tianhao Wang +1

In the field of machine unlearning, certified unlearning has been extensively studied in convex machine learning models due to its high efficiency and strong theoretical guarantees…

cs.LG2024

IDEA: A Flexible Framework of Certified Unlearning for Graph Neural Networks

Yushun Dong, Binchi Zhang, Zhenyu Lei +2

Graph Neural Networks (GNNs) have been increasingly deployed in a plethora of applications. However, the graph data used for training may contain sensitive personal information of…

cs.LG2024★ 1 cited

A Benchmark for Fairness-Aware Graph Learning

Yushun Dong, Song Wang, Zhenyu Lei +4

Fairness-aware graph learning has gained increasing attention in recent years. Nevertheless, there lacks a comprehensive benchmark to evaluate and compare different fairness-aware…

cs.LG2024★ 7 cited

Causal Inference with Latent Variables: Recent Advances and Future Prospectives

Yaochen Zhu, Yinhan He, Jing Ma +3

Causality lays the foundation for the trajectory of our world. Causal inference (CI), which aims to infer intrinsic causal relations among variables of interest, has emerged as a c…

cs.LG2024

Spectral Greedy Coresets for Graph Neural Networks

Mucong Ding, Yinhan He, Jundong Li +1

The ubiquity of large-scale graphs in node-classification tasks significantly hinders the real-world applications of Graph Neural Networks (GNNs). Node sampling, graph coarsening,…

cs.LG2024

Safety in Graph Machine Learning: Threats and Safeguards

Song Wang, Yushun Dong, Binchi Zhang +7

Graph Machine Learning (Graph ML) has witnessed substantial advancements in recent years. With their remarkable ability to process graph-structured data, Graph ML techniques have b…

cs.LG2024

GraphRCG: Self-Conditioned Graph Generation

Song Wang, Zhen Tan, Xinyu Zhao +3

Graph generation generally aims to create new graphs that closely align with a specific graph distribution. Existing works often implicitly capture this distribution through the op…

cs.LG2023★ 1 cited

Adversarial Attacks on Fairness of Graph Neural Networks

Binchi Zhang, Yushun Dong, Chen Chen +3

Fairness-aware graph neural networks (GNNs) have gained a surge of attention as they can reduce the bias of predictions on any demographic group (e.g., female) in graph-based appli…

cs.LG2023

Fair Few-shot Learning with Auxiliary Sets

Song Wang, Jing Ma, Lu Cheng +1

Recently, there has been a growing interest in developing machine learning (ML) models that can promote fairness, i.e., eliminating biased predictions towards certain populations (…

cs.LG2023★ 1 cited

Federated Few-shot Learning

Song Wang, Xingbo Fu, Kaize Ding +3

Federated Learning (FL) enables multiple clients to collaboratively learn a machine learning model without exchanging their own local data. In this way, the server can exploit the…

cs.LG2023

Spatial-Temporal Networks for Antibiogram Pattern Prediction

Xingbo Fu, Chen Chen, Yushun Dong +4

An antibiogram is a periodic summary of antibiotic resistance results of organisms from infected patients to selected antimicrobial drugs. Antibiograms help clinicians to understan…

cs.LG2023★ 18 cited

Few-shot Node Classification with Extremely Weak Supervision

Song Wang, Yushun Dong, Kaize Ding +2

Few-shot node classification aims at classifying nodes with limited labeled nodes as references. Recent few-shot node classification methods typically learn from classes with abund…