Yinwei Dai

I am a Computer Science Ph.D. student at Princeton University, working with Prof. Ravi Netravali.

I obtained my M.S.E. and B.S.E. in Computer Science at University of Michigan , where I worked with Prof. Mosharaf Chowdhury and Prof. Harsha V. Madhyastha on projects related to networked systems, I also collaborated Prof. David Fouhey and B.S.E in Electrical and Compter Engineering from Shanghai Jiao Tong University

Email  /  Google Scholar  /  Twitter  /  Github

profile photo
Research

My research interests are at the intersection of networked system and data-intensive computing.

Publications
profile photo
Auxo: Heterogeneity-Mitigating Federated Learning via Scalable Client Clustering
Jiachen Liu, Fan Lai, Yinwei Dai, Aditya Akella, Harsha Madhyastha, Mosharaf Chowdhury
SoCC, 2023 / Github / Paper

We propose Auxo, a scalable FL system that enable the server to decompose the large-scale FL task into groups with smaller intra-cohort heterogeneity.

profile photo
ModelKeeper: Accelerating DNN Training via Automated Training Warmup
Fan Lai, Yinwei Dai, Harsha Madhyastha, Mosharaf Chowdhury
NSDI, 2023 Acceptance Rate: 18.38% / Github / Paper / Talk

We introduce ModelKeeper, a cluster-scale model service framework to accelerate DNN training, by reducing the computation needed for achieving the same model performance via automated model transformation

profile photo
FedScale: Benchmarking Model and System Performance of Federated Learning
Fan Lai, Yinwei Dai, Sanjay Singapuram, Jiachen Liu, Xiangfeng Zhu, Harsha Madhyastha, Mosharaf Chowdhury
ICML, 2022 Acceptance Rate: 21.94% / Website / Github
Short Version: Best Paper Award @ ResilientFL at SOSP, 2021 Paper / Talk

We present FedScale, a diverse set of challenging and realistic benchmark datasets to facilitate scalable, comprehensive, and reproducible federated learning (FL) research.

Teaching
Graduate Student Instructor EECS 442 Computer Vision, Winter 2022

Graduate Student Instructor EECS 489 Computer Network , Fall 2021

Graduate Student Instructor COS 316: Principles of Computer System Design, Fall 2023

Service
Conference Reviewer: NeurIPS (Datasets and Benchmarks) 2022, 2023

Journal Reviewer: Transactions on Mobile Computing 2022

Artifact Evaluation Committee: SIGCOMM 2022, MLSys 2023

Misc
My name in Chinese: name photo

If you want to chat with me, please send me an email. :)