Yinwei Dai

I am a 1st year 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 the University of Michigan and B.S.E in Electrical and Compter Engineering from Shanghai Jiao Tong University

Previously, I worked with Prof. Mosharaf Chowdhury and Prof. Harsha V. Madhyastha on projects related to ML systems and Big data analytics. I was also supervised by Prof. David Fouhey on Computer Vision related projects.

Email  /  CV  /  Google Scholar  /  Twitter  /  Github

profile photo
Research

My research interests are at the intersection of networked system and machine learning in the context of data-intensive computing, including big data systems for scale-out ML and federated data analytics.

Preprints
profile photo
Auxo: Heterogeneity-Mitigating Federated Learning via Scalable Client Clustering
Jiachen Liu, Fan Lai, Yinwei Dai, Aditya Akella, Harsha Madhyastha, Mosharaf Chowdhury
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.

Publications
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.

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

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

Teaching
Graduate Student Instructor EECS 442 Computer Vision, Winter 2022

Graduate Student Instructor EECS 489 Computer Network , Fall 2021

Grader EECS 545 Introduction to Machine Learning, Winter 2020

Grader EECS 504, Introduction to Computer Vision, Winter 2020

Service
Program Commitee: NeurIPS (Datasets and Benchmarks) 2022

Artifact Evaluation Commitee: SIGCOMM 2022

Projects

Interesting projects I have done.

profile photo
2-way superscalar P6 processor
Code / Report / Slide

The project has built a P6-structure pipeline to handle RISC-V instructions. Some advanced features such as superscalar execution are added to enhance processor performance.

profile photo
Compiler for Decaf Programming Language

Implementation of a full stack compiler including lexical scanning, parsing, abstract syntax trees, semantic analysis, intermediate code generation, MIPS code generation and optimization

profile photo
The Gold Rush: Multiplayer Online Game Made with Unity
Jiaxing Yang, Siyuan Peng, Weiliu Liu, Yinwei Dai, Zhiqi Chen
DevLog / Trailer / Github

The capstone project made by Godzilli Studio

profile photo
Black and White

Demo / Github / Download

A small puzzel game I made with unity. The player needs to change color space accordingly to pass each level

profile photo
The Legend of Zelda 1986: Remade with Unity
Yinwei Dai, Zhiqi Chen
Demo / Github / Download

Reimplementation of classic The Legend of Zelda 1986 with Unity

Misc
My name in Chinese: name photo

If you want chat with me, you can find a slot here. Please also add the meeting link as well. :)