← Search

Chuan Shi

50 accepted papers

2026

From Denoising to De-Channeling: Integrating Physical Channel Priors into Diffusion Models for Radio Signal Understanding

ICML 2026spotlight

In recent years, wireless signal recognition (WSR), which leverages artificial intelligence (AI) to identify properties of passively received radio signals, has garnered significant attention due to its broad applications, such as spectrum management. Existing WSR methods typically learn directly fr…

Cited by 0SourceScholar
2026

Graph4LLM: A Systematic Survey of Graph-Enhanced Large Language Models

IJCAI 2026

Large Language Models (LLMs) excel in natural language processing (NLP) tasks. However, they suffer from inherent limitations due to their sequence-based nature, such as structural information loss and factual unreliability. Graphs, with the ability to explicitly model entities and relations, offer

Cited by 0Scholar
2026

ParaTool: Shifting Tool Representations from Context to Parameters

ICML 2026poster

Tool calling extends large language models (LLMs) by enabling grounded interaction with external executable interfaces, thereby supporting environment-coupled problem solving. However, mainstream in-context learning (ICL) approaches typically incorporate detailed tool documentation and usage example…

Cited by 0SourceScholar
2026

PathRAG: Pruning Graph-based Retrieval Augmented Generation with Relational Paths

AAAI 2026technical

Retrieval-augmented generation (RAG) improves the response quality of large language models (LLMs) by retrieving knowledge from external databases. Typical RAG approaches split the text database into chunks, organizing them in a flat structure for efficient searches. To better capture the inherent d

Cited by 0SourcePDFScholar
2026

Relational In-Context Learning via Synthetic Pre-training with Structural Prior

ICML 2026poster

Relational Databases (RDBs) are the backbone of modern business, yet they have missed the Foundation Model revolution. Unlike text or images, high-quality RDB data is private and scarce, rendering the standard approach of ``pre-training on the internet'' infeasible. Consequently, existing solutions …

Cited by 0SourceScholar
2026

TS-DDAE: A novel Temporal-Spectral Denoising Diffusion AutoEncoder for Wireless Signal Recognition Model Pre-training

ICLR 2026poster

Wireless Signal Recognition (WSR) aims to identify the property of received signals using Artificial Intelligence (AI) without any prior knowledge, which has been widely used in civil and military radios. The current AI trend of pre-training and fine-tuning has shown great performance, and the exist…

Cited by 0SourcecodeScholar
2026

Topology Matters in RTL Circuit Representation Learning

ICLR 2026poster

Representation learning for register transfer level (RTL) circuits is fundamental to enabling accurate performance, power, and area (PPA) prediction, efficient circuit generation, and retrieval in automated chip design. Unlike general programming languages, RTL is inherently a structured dataflow gr…

Cited by 0SourcecodeScholar
2025

A Weighted Cross-entropy Loss for Mitigating LLM Hallucinations in Cross-lingual Continual Pretraining

ICASSP 2025accepted

Recently, due to the explosive advances of large language models (LLMs) on English, cross-lingual continual pretraining has been widely applied in obtaining Chinese LLMs. However, previous studies showed that these LLMs have suffered severe hallucinations, mainly caused by noisy tokens. To this aim,…

Cited by 0SourceScholar
2025

Between Circuits and Chomsky: Pre-pretraining on Formal Languages Imparts Linguistic Biases

ACL 2025long

Pretraining language models on formal language can improve their acquisition of natural language. Which features of the formal language impart an inductive bias that leads to effective transfer? Drawing on insights from linguistics and complexity theory, we hypothesize that effective transfer occurs…

Cited by 0SourcePDFScholar
2025

Blend the Separated: Mixture of Synergistic Experts for Data-Scarcity Drug-Target Interaction Prediction

AAAI 2025technical

Drug-target interaction prediction (DTI) is essential in various applications including drug discovery and clinical application. There are two perspectives of input data widely used in DTI prediction: Intrinsic data represents how drugs or targets are constructed, and extrinsic data represents how d…

2025

Exploring the Potential of Large Language Models for Heterophilic Graphs

NAACL 2025long

Large language models (LLMs) have presented significant opportunities to enhance various machine learning applications, including graph neural networks (GNNs). By leveraging the vast open-world knowledge within LLMs, we can more effectively interpret and utilize textual data to better characterize h…

Cited by 2SourcePDFScholar
2025

Federated Graph Condensation with Information Bottleneck Principles

AAAI 2025technical

Graph condensation (GC), which reduces the size of a large-scale graph by synthesizing a small-scale condensed graph as its substitution, has benefited various graph learning tasks. However, existing GC methods rely on centralized data storage, which is unfeasible for real-world decentralized data d…

Cited by 3SourcePDFScholar
2025

Harnessing Language Model for Cross-Heterogeneity Graph Knowledge Transfer

AAAI 2025technical

Heterogeneous graphs (HGs) that contain various node and edge types are ubiquitous in real-world scenarios. Considering the common label sparsity problem in HGs, some researchers propose to pretrain on source HGs to extract general knowledge and then fine-tune on a target HG for knowledge transfer.…

2025

Rethinking Byzantine Robustness in Federated Recommendation from Sparse Aggregation Perspective

AAAI 2025technical

To preserve user privacy in recommender systems, federated recommendation (FR) based on federated learning (FL) emerges, keeping the personal data on the local client and updating a model collaboratively. Unlike FL, FR has a unique sparse aggregation mechanism, where the embedding of each item is up…

2025

Seq1F1B: Efficient Sequence-Level Pipeline Parallelism for Large Language Model Training

NAACL 2025long

Training large language models (LLMs) heavily relies on distributed training strategies, among which pipeline parallelism (PP) plays a crucial role. As training sequences extend to 32k or even 128k tokens, current PP methods face severe bottlenecks, including substantial pipeline bubbles and high me…

2025

Unifying and Enhancing Graph Transformers via a Hierarchical Mask Framework

NeurIPS 2025poster

Graph Transformers (GTs) have emerged as a powerful paradigm for graph representation learning due to their ability to model diverse node interactions. However, existing GTs often rely on intricate architectural designs tailored to specific interactions, limiting their flexibly. To address this, we…

Cited by 0SourceScholar
2024

Can Large Language Models Analyze Graphs like Professionals? A Benchmark, Datasets and Models

NeurIPS 2024poster

The need to analyze graphs is ubiquitous across various fields, from social networks to biological research and recommendation systems. Therefore, enabling the ability of large language models (LLMs) to process graphs is an important step toward more advanced general intelligence. However, current L…

2024

FairSIN: Achieving Fairness in Graph Neural Networks through Sensitive Information Neutralization

AAAI 2024technical

Despite the remarkable success of graph neural networks (GNNs) in modeling graph-structured data, like other machine learning models, GNNs are also susceptible to making biased predictions based on sensitive attributes, such as race and gender. For fairness consideration, recent state-of-the-art (SO…

2024

Graph Contrastive Invariant Learning from the Causal Perspective

AAAI 2024technical

Graph contrastive learning (GCL), learning the node representation by contrasting two augmented graphs in a self-supervised way, has attracted considerable attention. GCL is usually believed to learn the invariant representation. However, does this understanding always hold in practice? In this pape…

2024

Graph Invariant Learning with Subgraph Co-mixup for Out-of-Distribution Generalization

AAAI 2024technical

Graph neural networks (GNNs) have been demonstrated to perform well in graph representation learning, but always lacking in generalization capability when tackling out-of-distribution (OOD) data. Graph invariant learning methods, backed by the invariance principle among defined multiple environments…

2024

Less is More: on the Over-Globalizing Problem in Graph Transformers

ICML 2024oral

Graph Transformer, due to its global attention mechanism, has emerged as a new tool in dealing with graph-structured data. It is well recognized that the global attention mechanism considers a wider receptive field in a fully connected graph, leading many to believe that useful information can be ex…

2023

Clustering-Based Supervised Contrastive Learning for Identifying Risk Items on Heterogeneous Graph

ICASSP 2023accepted

Risk item identification is vital for protecting the health of ecommerce trades. Existing solutions prefer to model structure information besides item attributes and optimize parameters in cross-entropy (CE) manners. However, the few labeled and imbalanced supervision in real-world scenarios usually…

Cited by 0SourceScholar
2023

Directed Acyclic Graph Structure Learning from Dynamic Graphs

AAAI 2023technical

Estimating the structure of directed acyclic graphs (DAGs) of features (variables) plays a vital role in revealing the latent data generation process and providing causal insights in various applications. Although there have been many studies on structure learning with various types of data, the str…

2023

Graph Contrastive Learning with Stable and Scalable Spectral Encoding

NeurIPS 2023poster

Graph contrastive learning (GCL) aims to learn representations by capturing the agreements between different graph views. Traditional GCL methods generate views in the spatial domain, but it has been recently discovered that the spectral domain also plays a vital role in complementing spatial views.…

Cited by 22SourcePDFScholar
2023

Injecting Multimodal Information into Rigid Protein Docking via Bi-level Optimization

NeurIPS 2023poster

The structure of protein-protein complexes is critical for understanding binding dynamics, biological mechanisms, and intervention strategies. Rigid protein docking, a fundamental problem in this field, aims to predict the 3D structure of complexes from their unbound states without conformational ch…

Cited by 6SourcePDFScholar
2023

Learning Invariant Representations of Graph Neural Networks via Cluster Generalization

NeurIPS 2023poster

Graph neural networks (GNNs) have become increasingly popular in modeling graph-structured data due to their ability to learn node representations by aggregating local structure information. However, it is widely acknowledged that the test graph structure may differ from the training graph structure…

2023

Provable Training for Graph Contrastive Learning

NeurIPS 2023spotlight

Graph Contrastive Learning (GCL) has emerged as a popular training approach for learning node embeddings from augmented graphs without labels. Despite the key principle that maximizing the similarity between positive node pairs while minimizing it between negative node pairs is well established, som…

2023

Specformer: Spectral Graph Neural Networks Meet Transformers

ICLR 2023poster

Spectral graph neural networks (GNNs) learn graph representations via spectral-domain graph convolutions. However, most existing spectral graph filters are scalar-to-scalar functions, i.e., mapping a single eigenvalue to a single filtered value, thus ignoring the global pattern of the spectrum. Furt…

2022

Data-Free Adversarial Knowledge Distillation for Graph Neural Networks

IJCAI 2022poster

Graph neural networks (GNNs) have been widely used in modeling graph structured data, owing to its impressive performance in a wide range of practical applications. Recently, knowledge distillation (KD) for GNNs has enabled remarkable progress in graph model compression and knowledge transfer. Howev…

Cited by 21SourcePDFScholar
2022

Debiasing Graph Neural Networks via Learning Disentangled Causal Substructure

NeurIPS 2022accept

Most Graph Neural Networks (GNNs) predict the labels of unseen graphs by learning the correlation between the input graphs and labels. However, by presenting a graph classification investigation on the training graphs with severe bias, surprisingly, we discover that GNNs always tend to explore the s…

2022

Regularizing Graph Neural Networks via Consistency-Diversity Graph Augmentations

AAAI 2022technical

Despite the remarkable performance of graph neural networks (GNNs) in semi-supervised learning, it is criticized for not making full use of unlabeled data and suffering from over-fitting. Recently, graph data augmentation, used to improve both accuracy and generalization of GNNs, has received consid…

Cited by 30SourcePDFScholar
2022

Revisiting Graph Contrastive Learning from the Perspective of Graph Spectrum

NeurIPS 2022accept

Graph Contrastive Learning (GCL), learning the node representations by augmenting graphs, has attracted considerable attentions. Despite the proliferation of various graph augmentation strategies, there are still some fundamental questions unclear: what information is essentially learned by GCL? Are…

2022

Robust Heterogeneous Graph Neural Networks against Adversarial Attacks

AAAI 2022technical

Heterogeneous Graph Neural Networks (HGNNs) have drawn increasing attention in recent years and achieved outstanding performance in many tasks. However, despite their wide use, there is currently no understanding of their robustness to adversarial attacks. In this work, we first systematically study…

Cited by 64SourcePDFScholar
2022

Self-supervised Graph Neural Networks for Multi-behavior Recommendation

IJCAI 2022poster

Traditional recommendation usually focuses on utilizing only one target user behavior (e.g., purchase) but ignoring other auxiliary behaviors (e.g., click, add to cart). Early efforts of multi-behavior recommendation often emphasize the differences between multiple behaviors, i.e., they aim to extra…

Cited by 98SourcePDFScholar
2021

Be Confident! Towards Trustworthy Graph Neural Networks via Confidence Calibration

NeurIPS 2021poster

Despite Graph Neural Networks (GNNs) have achieved remarkable accuracy, whether the results are trustworthy is still unexplored. Previous studies suggest that many modern neural networks are over-confident on the predictions, however, surprisingly, we discover that GNNs are primarily in the opposite…

Cited by 140SourcePDFScholar
2021

Beyond Low-frequency Information in Graph Convolutional Networks

AAAI 2021technical

Graph neural networks (GNNs) have been proven to be effective in various network-related tasks. Most existing GNNs usually exploit the low-frequency signals of node features, which gives rise to one fundamental question: is the low-frequency information all we need in the real world applications? In…

2021

Compare to The Knowledge: Graph Neural Fake News Detection with External Knowledge

ACL 2021long

Nowadays, fake news detection, which aims to verify whether a news document is trusted or fake, has become urgent and important. Most existing methods rely heavily on linguistic and semantic features from the news content, and fail to effectively exploit external knowledge which could help determine…

2021

CuCo: Graph Representation with Curriculum Contrastive Learning

IJCAI 2021poster

Graph-level representation learning is to learn low-dimensional representation for the entire graph, which has shown a large impact on real-world applications. Recently, limited by expensive labeled data, contrastive learning based graph-level representation learning attracts considerable attention.…

Cited by 115SourcePDFScholar
2021

GraphMSE: Efficient Meta-path Selection in Semantically Aligned Feature Space for Graph Neural Networks

AAAI 2021technical

Heterogeneous information networks (HINs) are ideal for describing real-world data with different types of entities and relationships. To carry out machine learning on HINs, meta-paths are widely utilized to extract semantics with pre-defined patterns, and models such as graph convolutional networks…

2021

Heterogeneous Graph Structure Learning for Graph Neural Networks

AAAI 2021technical

Heterogeneous Graph Neural Networks (HGNNs) have drawn increasing attention in recent years and achieved outstanding performance in many tasks. The success of the existing HGNNs relies on one fundamental assumption, i.e., the original heterogeneous graph structure is reliable. However, this assumpti…

2021

Who You Would Like to Share With? A Study of Share Recommendation in Social E-commerce

AAAI 2021technical

The prosperous development of social e-commerce has spawned diverse recommendation demands, and accompanied a new recommendation paradigm, share recommendation. Significantly different from traditional binary recommendations (e.g., item recommendation and friend recommendation), share recommendation…

Cited by 44SourcePDFScholar
2020

Decorrelated Clustering with Data Selection Bias

IJCAI 2020poster

Most of existing clustering algorithms are proposed without considering the selection bias in data. In many real applications, however, one cannot guarantee the data is unbiased. Selection bias might bring the unexpected correlation between features and ignoring those unexpected correlations will hu…

2020

Network Schema Preserving Heterogeneous Information Network Embedding

IJCAI 2020poster

As heterogeneous networks have become increasingly ubiquitous, Heterogeneous Information Network (HIN) embedding, aiming to project nodes into a low-dimensional space while preserving the heterogeneous structure, has drawn increasing attention in recent years. Many of the existing HIN embedding meth…