Yaohui Cai

I am Yaohui Cai (蔡耀辉), a Ph.D. student at Cornell University advised by Zhiru Zhang and Chris De Sa. I received my bachelor degree at Peking University. I worked closely with Kurt Keutzer and Michael Mahoney at UC Berkeley. I mainly focus on efficient inference and training for deep learning.

Email: yc2632 [at] cornell [dot] edu

Google Scholar / GitHub

Publications

ZeroQ: A Novel Zero Shot Quantization Framework.

Yaohui Cai*, Zhewei Yao*, Zhen Dong*, Amir Gholami, Michael Mahoney, and Kurt Keutzer.

[paper] [code] Conference on Computer Vision and Pattern Recognition (CVPR), 2020.

HAWQ-V2: Hessian Aware trace-Weighted Quantization of Neural Networks.

Zhen Dong, Zhewei Yao, Yaohui Cai*, Daiyaan Arfeen*, Amir Gholami, Michael Mahoney, and Kurt Keutzer.

[paper] [code] Preprint

CoDeNet: Efficient Deployment of Input-Adaptive Object Detection on Embedded FPGAs

Zhen Dong, Qijing Huang, Dequan Wang, Yizhao Gao, Yaohui Cai, Tian Li, Bichen Wu, Kurt Keutzer, and John Wawrzynek.

[paper] [code] International Symposium on Field-Programmable Gate Arrays (FPGA), 2021. Oral

SPADE: A Spectral Method for Black-Box Adversarial Robustness Evaluation

Wuxinlin Cheng, Chenhui Deng, Zhiqiang Zhao, Yaohui Cai, Zhiru Zhang, Zhuo Feng,

[paper] [code] International Conference on Machine Learning (ICML), 2021

Structured Pruning is All You Need for Pruning CNNs at Initialization

Yaohui Cai, Weizhe Hua, Hongzheng Chen, G. Edward Suh, Chris De Sa, Zhiru Zhang,

[paper] Preprint

Education

Cornell University, Ithaca, NY Ph.D. in Electrical and Computer Engineering • Jan. 2021 - Present
Peking University, Beijing, China B.Sc. in Data Science • Sep. 2016 to Jun. 2020
Massachusetts Institute of Technology, Cambridge, MA Undergraduate Special Student Program (Exchange Program) • Feb. 2019 - May 2019

Experiences

Google Research, Brain Team (Now Google DeepMind)
Student Researcher • May 2023 - Present

Meta, Meta Reality Labs
Research Scientist Intern • May 2022 - Dec. 2022

University of California, Berkeley, Berkeley Artificial Intelligence Research (BAIR) Lab
Visiting Student Researcher • May 2019 - Jan. 2020
Advisors: Professor Kurt Keutzer and Professor Michael Mahoney

Website source from Jon Barron and Xingyuan Sun.