Avatar

Zhihao Shi

PhD Student


zhihaoshi@miralab.ai

Interests

  • Scalable Graph Neural Networks/Graph Transformer
  • Large Pretrained Model

Education

  • B.S., Electronic Engineering and Information Science, 2020

    University of Science and Technology of China

About me

I am currently a master student in USTC. My advidor is Prof. Jie Wang, who leads the MIRA Lab. My research interests include scalable graph neural networks/graph Transformer and large pretrained model.

Experience

  • Intern at Alibaba DAMO Academy, 2023/07 - 2024/10

Publications

Yue Shen, Jie Wang, Zhe Wang, Zhihao Shi, Hanzhu Chen, Zheng Wang, Yukang Jiang, Xiaopu Wang, Chuandong Cheng, Xueqin Wang, Hongtu Zhu, Jieping Ye. CATI: A medical context-enhanced framework for diagnosis code assignment in the UK Biobank study. AI in Medicine.

Zhihao Shi, Jie Wang, Zhiwei Zhuang, Xize Liang, Bin Li, Feng Wu. Accurate and Scalable Graph Neural Networks via Message Invariance. ICLR 2025.

Zhihao Shi, Jie Wang, Fanghua Lu, Hanzhu Chen, Defu Lian, Zheng Wang, Jieping Ye, Feng Wu. Label Deconvolution for Node Representation Learning on Large-scale Attributed Graphs against Learning Bias. TPAMI.

Jie Wang, Zhihai Wang, Xijun Li, Yufei Kuang, Zhihao Shi, Fangzhou Zhu, Mingxuan Yuan, Jia Zeng, Yongdong Zhang, Feng Wu. Learning to Cut via Hierarchical Sequence/Set Model for Efficient Mixed-Integer Programming. TPAMI.

Zhihao Shi, Xize Liang, Jie Wang. LMC: Fast Training of GNNs via Subgraph Sampling with Provable Convergence (Spotlight). ICLR 2023.

PDF Code

Jie Wang, Zhanqiu Zhang, Zhihao Shi, Jianyu Cai, Shuiwang Ji, Feng Wu. Duality-Induced Regularizer for Semantic Matching Knowledge Graph Embeddings. TPAMI.