Zitao Shuai | University of Michigan

About Me

Hi! I’m Zitao Shuai [帅子滔], I’m currently a master’s student in the EECS department at the University of Michigan. I’m advised by Prof.Liyue Shen in the University of Michigan and I’m fortunate to work with Prof.Kun Kuang during my undergraduate in Zhejiang University. I got my bachelor’s degree in Computer Science and my bachelor’s degree in Finance from Zhejiang University. And I’m also a member of CKC Honors College. During my undergraduate, I was advised by Prof.Kun Kuang in the Department of Computer Science and Technology of ZJU.

Research Interests

Bio-Medical AI
Trustworthy AI
Multimodal representation learning

News

[Sep. 2023] Joined Biomedical AI Lab!

Publications

[Cell Patterns]

Unified fair federated learning for digital healthcare

Zhang, Fengda and Shuai, Zitao[co-first author] and Kuang, Kun and Wu, Fei and Zhuang, Yueting and Xiao, Jun

PDF ML fairness for healthcare

[Submitted to TMLR]

Latent Space Disentanglement in Diffusion Transformers Enables Zero-shot Fine-grained Semantic Editing

Shuai, Zitao and Wu, Chenwei and Tang, Zhengxu and Song, Bowen and Shen, Liyue

PDF Trustworthy in multi-modal learning

[Submitted to ICLR2025]

Dynamic Modeling of Patients, Modalities and Tasks via Multi-modal Multi-task Mixture of Experts

Wu, Chenwei and Shuai, Zitao[co-first author] and Tang, Zhengxu and Wang, Luning and Shen, Liyue

PDF multi-modal learning

[Submitted to TMLR]

Distributionally Robust Alignment for Medical Federated Vision-Language Pre-training Under Data Heterogeneity

Shuai, Zitao and Shen, Liyue

PDF Trustworthy in multi-modal learning

[AAAI2025]

Multi-OphthaLingua: A Multilingual Benchmark for Assessing and Debiasing LLM Ophthalmological QA in LMICs

Restrepo, David and Wu, Chenwei and Tang, Zhengxu, and Shuai, Zitao, and et al

PDF Foundation model in healthcare

[MICCAI2024]

Efficient In-Context Medical Segmentation with Meta-driven Visual Prompt Selection

Wu, Chenwei and Restrepo, David and Shuai, Zitao and Liu, Zhongming and Shen, Liyue

PDF Foundation model in healthcare

[IEEE Trans on Selected Signal Processing]

Multi-Sources Fusion Learning for Multi-Points NLOS Localization in OFDM System

Wang, Bohao and Shuai, Zitao and Huang, Chongwen, et al

PDF AI4Science application

Talks

MLLM

Semantic Disentanglemnt in Diffusion Transformers

Zitao Shuai, Chenwei Wu

PDF Lab Meeting

MLLM

Tutorial on Disentanglement in Diffusion Models

Zitao Shuai, Chenwei Wu

PDF Lab Meeting

Outside the Office

I’m a passionate tennis enthusiast and have been playing since primary school. I’m a baseline topspin player, right-handed (two-handed backhand). My favorite player is Rafael Nadal, not only for his exceptional athleticism but also for his remarkable sportsmanship, which inspires me to persevere in my research endeavors and relentlessly pursue my scientific aspirations.