Yuankai Luo (罗元凯)

Assistant Professor at Nanjing University

luoyuankai.jpg

Email: yuankailuo@nju.edu.cn

[Google Scholar] [Github]

Now I am an Assistant Professor at the School of Artificial Intelligence, Nanjing University (NJU). I received my Ph.D. degree from the School of Computer Science and Engineering at Beihang University, where I was supervised by Prof. Lei Shi and jointly trained at The Hong Kong Polytechnic University under the supervision of Prof. Xiao-Ming Wu. Before that, I did research supervised by Veronika Thost.

My research interests span Graph Neural Network (GNN) research, including architecture design, pre-training strategies, and compression/acceleration:

GNN Architecture and Analysis:

  • Unified Framework GNN+ for Reassessing Classic GNNs in General Graph Tasks: developed GNN+, a framework that integrates message passing and well-known regularization techniques like dropout. GNN+ demonstrates that the true potential of classic GNNs has been previously underestimated in both node-level and graph-level tasks, challenging the belief that complex mechanisms are necessary for superior performance in graph models [NeurIPS 2024, ICML 2025, ICLR 2025].
  • Graph Transformers for Specialized Small-Scale Graph Tasks: designed specialized Graph Transformers architecture tailored for small-scale graph tasks, particularly directed acyclic graphs [NeurIPS 2023, KDD 2023], and graphs with multi-level structures [NeurIPS 2024].

GNN Pre-training: proposed a graph self-supervised learning framework based on persistent homology theory, which effectively captures the multi-scale topological features of graph data [NeurIPS 2023].

GNN Compression: introduced vector quantization to compress continuous node embeddings into highly compact (typically 6-15 dimensions), discrete (int4 type), and interpretable node representations—termed Node IDs [ICLR 2025].

Recent Publications

  1. ICML 2025
    GNNplus.jpg
    Can Classic GNNs Be Strong Baselines for Graph-level Tasks?
    Simple Architectures Meet Excellence
    Yuankai Luo, Lei Shi, and Xiao-Ming Wu
    In The Forty-second International Conference on Machine Learning, 2025
  2. ICLR 2025
    NodeID.png
    Node Identifiers: Compact, Discrete Representations for Efficient Graph Learning
    Yuankai Luo, Hongkang Li, Qijiong Liu, Lei Shi, and Xiao-Ming Wu
    In The Thirteenth International Conference on Learning Representations, 2025
  3. ICLR 2025
    dropout.jpg
    Beyond Random Masking: When Dropout meets Graph Convolutional Networks
    Yuankai Luo, Xiao-Ming Wu, and Hao Zhu
    In The Thirteenth International Conference on Learning Representations, 2025
  4. NeurIPS 2024
    GNN2.png
    Classic GNNs are Strong Baselines: Reassessing GNNs for Node Classification
    Yuankai Luo, Lei Shi, and Xiao-Ming Wu
    In Thirty-eighth Conference on Neural Information Processing Systems, 2024
  5. NeurIPS 2024
    HDSE.png
    Enhancing Graph Transformers with Hierarchical Distance Structural Encoding
    Yuankai Luo, Hongkang Li, Lei Shi, and Xiao-Ming Wu
    In Thirty-eighth Conference on Neural Information Processing Systems, 2024
  6. NeurIPS 2023
    DAG.png
    Transformers over Directed Acyclic Graphs
    Yuankai Luo, Veronika Thost, and Lei Shi
    In Thirty-seventh Conference on Neural Information Processing Systems, 2023
  7. NeurIPS 2023
    PH.png
    Improving Self-supervised Molecular Representation Learning using Persistent Homology
    Yuankai Luo, Lei Shi, and Veronika Thost
    In Thirty-seventh Conference on Neural Information Processing Systems, 2023
  8. SIGKDD 2023
    KDD.png
    Impact-Oriented Contextual Scholar Profiling Using Self-Citation Graphs
    Yuankai Luo, Lei Shi, Mufan Xu, Yuwen Ji, Fengli Xiao, and 2 more authors
    In Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, 2023

Academic Services

Conference Reviewer:

  • WSDM 2023/2024, ICML 2024/2025, NeurIPS 2024(Top Reviewer Award)/2025, ACL ARR 2024/2025, ICLR 2025, AAAI 2025

Journal Reviewer:

  • IEEE Transactions on Knowledge and Data Engineering (TKDE)
  • IEEE Transactions on Neural Networks and Learning Systems (TNNLS)
  • IEEE Transactions on Intelligent Transportation Systems (TITS)