Zitao Shuai

About Me

Hi! I’m Zitao Shuai [帅子滔], I’m currently a PhD student at UCLA Computer Science Department advised by Prof.Yuzhe Yang. I’m fortunate to work with Prof.Liyue Shen at the University of Michigan during my master’s studies, and with Prof.Kun Kuang at Zhejiang University during my undergraduate’s studies. And I’m also a member of CKC Honors College during my undergraduate’s studies.

I have broad interests in Research Interests in AI for Healthcare, Trustworthy AI, and Multimodal representation learning.

Research Interests

AI for Healthcare
Multi-modal learning
Trustworthy AI

My research interests are in multi-modal learning and AI for healthcare, especially in leveraging multi-source data to deliver trustworthy solutions for real-world healthcare problems. These interests have drawn me to explore two main research questions:

Foundation Model for Wearable Data: How can we build multi-modal foundation models for heterogeneous and long-context wearable signal data?
Scientific Finding with Wearable Data: How can we uncover correclations between diseases and signal data collected from wearable devices with modern AI techniques?

News

[Jul. 2026] Our Inertia-1 wearable motion foundation model is out, check here [https://yang-ai-lab.github.io/Inertia-1/]!
[Jul. 2026] Our SensorGen study on real-world sensor signal generation is out, check here [https://yang-ai-lab.github.io/sensor-gen/]!
[Mar. 2026] 2 paper accepted in ICML2026 (1 * regular, 1 * spotlight)
[Mar. 2026] Our Open Sleep Foundation model is out, check here [https://yang-ai-lab.github.io/osf/]!
[Mar. 2026] Our Open SleepLM is out, check here [https://yang-ai-lab.github.io/SleepLM/]!
[Mar. 2026] 1 paper accepted in CVPR2026 Findings!
[Oct. 2025] 1 paper accepted in JBI!
[Jul. 2025] 1 paper accepted in TMLR!
[May. 2025] 1 paper accepted in ACL2025!
[Apr. 2025] Joined Health Intelligence Lab!
[Feb. 2025] 1 paper accepted in Nature Scientific Data!
[Jan. 2025] 1 paper accepted in ICLR2025!
[Dec. 2024] 1 paper accepted in AAAI2025!
[Jun. 2024] 1 paper accepted in MICCAI2024!
[Dec. 2023] 1 paper accepted in Cell Patterns!
[Sep. 2023] Joined Biomedical AI Lab!
[Jun. 2022] Joined Trustworthy AI Lab!

Selected Publications

denotes equal contribution.

[Preprint]

Inertia-1: An Open Exploration of Wearable Motion Foundation Models

Zongzhe Xu*, Aakarsh Anand*, Sarah Jiang*, Chuntung Zhuang, Zitao Shuai, Sriram Sankararaman, Yuzhe Yang

PDF Code Project Page Foundation Models
[Preprint]

Signal or Noise? Understanding Generative Models for Real-World Sensor Time Series

Zitao Shuai*, Zongzhe Xu*, Yuntian Wu*, Sirui Li, Tianhong Li, Yuzhe Yang

PDF Code Project Page Generative Models
[ICML'26]

SleepLM: Natural-Language Intelligence for Human Sleep

Zongzhe Xu, Zitao Shuai, Eideen Mozaffari, Ravi S. Aysola, Rajesh Kumar, Yuzhe Yang

Spotlight

PDF Code Project Page Foundation Models
[ICML'26]

OSF: On Pre-training and Scaling of Sleep Foundation Models

Zitao Shuai, Zongzhe Xu, David Yang, Wei Wang, Yuzhe Yang

PDF Code Project Page Foundation Models
[CVPR'26]

Rethinking Medical High-Modality Learning under Missingness: A Long-Tailed Distribution Perspective

Chenwei Wu*, Zitao Shuai*, Liyue Shen

Findings

PDF Multi-modal Learning
[JBI]

Enhancing AI-based Diabetic Retinopathy Screening in Low- and Middle-Income Countries with Synthetic Data

Zitao Shuai*, Chenwei Wu*, Zhengxu Tang, David Restrepo, Michael Morley, Luis Filipe Nakayama

PDF AI for Healthcare
[TMLR]

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

Zitao Shuai, Chenwei Wu, Zhengxu Tang, Liyue Shen

PDF Trustworthy AI
[ICLR'25]

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

Chenwei Wu*, Zitao Shuai*, Zhengxu Tang*, Luning Wang, Liyue Shen

PDF Multi-modal Learning
[Cell Patterns]

Unified Fair Federated Learning for Digital Healthcare

Fengda Zhang*, Zitao Shuai*, Kun Kuang, Fei Wu, Yueting Zhuang, Jun Xiao

PDF Trustworthy AI

Outside the Office

I’m a tennis enthusiast and have been playing since I was 9. My favorite player is Rafael Nadal, not only for his exceptional athleticism but also for his remarkable sportsmanship - give his best on every shot until the very end of the match, which inspires me to persevere in my research endeavors and relentlessly pursue my scientific aspirations.