Publications

Knowledge Augmented Complex Problem Solving with Large Language Models: A Survey

Da Zheng, Lun Du, Junwei Su, Yuchen Tian, Yuqi Zhu, Jintian Zhang, Lanning Wei, Ningyu Zhang, Huajun Chen

Preprint · Under review

View paper

Step-TP: A Grounded, Step-Level Dataset with Chain-of-Thought Reasoning for LLM-Guided Tensor Program Optimization

Mengfan Liu, Da Zheng, Junwei Su^†, Chuan Wu

KDD, 2026 · Conference paper

View paper

The Role of Social Support and Influencers in Content Markets

Junwei Su, Peter Marbach

ACM Transactions on Economics and Computation · Journal article

View paper

Structuring Semantic Embeddings for Principle Evaluation: A Prototype-Guided Contrastive Learning Approach

Che Shen, Junwei Su^†, Lingpeng Kong, Chuan Wu

TMLR · Journal paper

View paper

Learning under Quantization for High-Dimensional Linear Regression

Dechen Zhang, Junwei Su, Difan Zou

ICLR, 2026 · Conference paper

View paper

HetAuto: Cross-Cluster Auto-Parallelism for Heterogeneous Distributed Training

Guicheng Qi, Junwei Su^†, Liqi Yang, Tao Li, Tingwen Xie, Yerui Sun, Yuchen Xie, Chuan Wu

EuroSys, 2026 · Conference paper

Full-Graph vs. Mini-Batch Training: Comprehensive Analysis from a Batch Size and Fan-Out Size Perspective

Mengfan Liu, Da Zheng, Junwei Su^†, Chuan Wu

ICLR, 2026 · Conference paper

BG-HGNN: Toward Scalable and Efficient Heterogeneous Graph Neural Network

Junwei Su^*†, Lingjun Mao^*, Da Zheng, Chuan Wu

TMLR · Journal paper

View paper

Temporal-Aware Evaluation and Learning for Temporal Graph Neural Networks

Junwei Su, Shan Wu

AAAI, 2025 · Conference paper

View paper

SBGD: Efficient and Scalable Graph Diffusion Generative Model via Block Decomposition

Junwei Su, Shan Wu

ICML, 2025 · Conference paper

View paper

A Non-Asymptotic Convergence Analysis for Scored-Based Graph Generative Models via a System of Stochastic Differential Equations

Junwei Su, Chuan Wu

ICML, 2025 · Conference paper

View paper

On the Interplay between Graph Structure and Learning Algorithms in Graph Neural Networks

Junwei Su, Chuan Wu

ICML, 2025 · Conference paper

View paper

Heta: Distributed Training of Heterogeneous Graph Neural Networks

Yuchen Zhong, Junwei Su^†, Chuan Wu, Minjie Wang

VLDB, 2025 · Conference paper

View paper

On the Topology Awareness and Generalization Performance of Graph Neural Networks

Junwei Su, Chuan Wu

ECCV, 2024 · Conference paper

Note: Oral presentation; Best Paper Candidate; perfect reviewer scores

View paper

Structural Properties of Core-Periphery Communities

Junwei Su, Peter Marbach

Advances in Complex Systems, 2024 · Journal article

View paper

PRES: Toward Scalable Memory-Based Dynamic Graph Neural Networks

Junwei Su, Difan Zou, Chuan Wu

ICLR, 2024 · Conference paper

View paper

MTRGL: Effective Temporal Correlation Discerning through Multi-modal Temporal Relational Graph Learning

Junwei Su, Shan Wu, Jinhui Li

ICASSP, 2024 · Conference paper

View paper

MSPipe: Efficient Temporal GNN Training via Staleness-aware Pipeline

Guangming Sheng, Junwei Su^†, Huang Chao, Chuan Wu

KDD, 2024 · Conference paper

View paper

Improving Implicit Regularization of SGD with Preconditioning for Least Square Problems

Junwei Su, Difan Zou, Chuan Wu

ICLR BGPT Workshop / SIAM Journal on Mathematics of Data Science, 2024 · Workshop and journal paper

View paper

HEGformer: Representation Learning on Heterogeneous Graph Neural Network with Efficient Transformer

Tianzuo Qin, Junwei Su, Chuan Wu

ICKG, 2024 · Conference paper

View paper

On the Limitation and Experience Replay for GNNs in Continual Learning

Junwei Su, Difan Zou, Chuan Wu

CoLLAs, 2024 · Conference paper

View paper

CDMPP: A Device-Model Agnostic Framework for Tensor Program Latency Prediction

Hanpeng Hu, Junwei Su^†, Juntao Zhao, Chuan Wu, Yanghua Peng, Yibo Zhu, Haibin Lin

EuroSys, 2024 · Conference paper

View paper

Towards Robust Graph Incremental Learning on Evolving Graphs

Junwei Su, Difan Zou, Zhang Zijun, Chuan Wu

ICML, 2023 · Conference paper

View paper