Zuobai Zhang 「张作柏」

I am a second-year PhD student at Mila - Québec AI Institute supervised by Prof. Jian Tang. Before that, I received my B.S. in Computer Science from Fudan University advised by Prof. Zhongzhi Zhang. I was also a machine learning engineering intern at ByteDance Inc., where I worked on recommender systems under the supervision of Yitan Li.

Email: zuobai.zhang [at] umontreal.ca  /  zuobai.zhang [at] mila.quebec  /  zuobai.zhang [at] gmail.com

Github   /   Google Scholar   /   CV     

Research

My research interests include machine learning for drug discovery, especially for protein science.

Publications
PEER: A Comprehensive and Multi-Task Benchmark for Protein Sequence Understanding
Minghao Xu*, Zuobai Zhang*, Jiarui Lu, Zhaocheng Zhu, Yangtian Zhang, Chang Ma, Runcheng Liu, Jian Tang
NeurIPS 2022 Datasets and Benchmarks Track
Effects of Stubbornness on Opinion Dynamics
Wanyue Xu, Liwang Zhu, Jiale Guan, Zuobai Zhang, Zhongzhi Zhang
CIKM 2022
Protein Representation Learning by Geometric Structure Pretraining
Zuobai Zhang, Minghao Xu, Arian Jamasb, Vijil Chenthamarakshan, Aurelie Lozano, Payel Das, Jian Tang
ICML 2022 Pre-Training Workshop
Neural-Symbolic Models for Logical Queries on Knowledge Graphs
Zhaocheng Zhu, Mikhail Galkin, Zuobai Zhang, Jian Tang
ICML 2022
Structured Multi-task Learning for Molecular Property Prediction
Shengchao Liu, Meng Qu, Zuobai Zhang, Huiyu Cai, Jian Tang
AISTATS 2022
NeurIPS AI for Science Workshop 2021
Biharmonic Distance-Based Performance Metric for Second-Order Noisy Consensus Networks
Yuhao Yi, Bingjia Yang, Zuobai Zhang, Wanyue Xu, Zhongzhi Zhang, Stacy Patterson
IEEE Transactions on Information Theory
Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Prediction
Zhaocheng Zhu, Zuobai Zhang, Louis-Pascal Xhonneux, Jian Tang
NeurIPS 2021
Learning Interpretable Cellular and Gene Signature Embeddings from Single-Cell Transcriptomic Data
Yifan Zhao*, Huiyu Cai*, Zuobai Zhang, Jian Tang, Yue Li
Nature Communications
Minimizing Spectral Radius of Non-Backtracking Matrix by Edge Removal
Zuobai Zhang, Zhongzhi Zhang, Guanrong Chen
CIKM 2021
Coherence Scaling of Noisy Second-Order Scale-Free Consensus Network
Wanyue Xu, Bin Wu, Zuobai Zhang, Zhongzhi Zhang, Haibin Kan, Guanrong Chen
IEEE Transactions on Cybernetics
Opinion Dynamics Incorporating High-Order Interactions
Zuobai Zhang, Wanyue Xu, Zhongzhi Zhang, Guanrong Chen
ICDM 2020 (short paper)
Power-Law Graphs Have Minimal Scaling of Kemeny Constant for Random Walks
Wanyue Xu, Yibin Sheng, Zuobai Zhang, Haibin Kan, Zhongzhi Zhang
WWW 2020
Fast Approximation of Coherence for Second-Order Noisy Consensus Networks
Zuobai Zhang, Wanyue Xu, Yuhao Yi, Zhongzhi Zhang
IEEE Transactions on Cybernetics
Nearly Linear Time Algorithm for Mean Hitting Times of Random Walks on a Graph
Zuobai Zhang, Wanyue Xu, Zhongzhi Zhang
WSDM 2020
Preprints
TorchDrug: A Powerful and Flexible Machine Learning Platform for Drug Discovery
Zhaocheng Zhu, Chence Shi, Zuobai Zhang, Shengchao Liu, Minghao Xu, Xinyu Yuan, Yangtian Zhang, Junkun Chen, Huiyu Cai, Jiarui Lu, Chang Ma, Runcheng Liu, Louis-Pascal Xhonneux, Meng Qu, Jian Tang
arXiv
A Roadmap for Big Model
I'm a co-first author of this paper.
arXiv
Projects
TorchProtein: A flexible machine learning library for protein science.

A machine learning library for protein science based on TorchDrug, providing representation learning models for both sequences and structures, and fundamental protein tasks like function prediction, structure prediction.

[Homepage] [Github] [Twitter]
TorchDrug: A powerful and flexible machine learning platform for drug discovery.

A machine learning platform designed for drug discovery, covering techniques from graph machine learning, deep generative models to reinforcement learning.

[Homepage] [Github] [Twitter]
Awards

Outstanding Undergraduates Graduates of Shanghai, Jun. 2021

Scholarships for Outstanding Academic Individuals at Fudan University, Dec. 2020

Shanghai Scholarship (Top 1%), Oct. 2020

SenseTime Scholarship (29 undergrads from China), Jan. 2020

Chinese National Scholarship (Top 1%), Oct. 2019

Wish Scholarship, May 2019

Chun Tsung Scholar Program, Apr. 2019

Chinese National Scholarship (Top 1%), Oct. 2018

Gold Medal, ACM-ICPC Asia Regional Contest EC-Final, Dec. 2017

Gold Medal, ACM-ICPC Asia Regional Contest Qingdao Site, Nov. 2017

Silver Medal, National Olympiad in Informatics, National Finals, Jul. 2016



Updated at Sept. 2022