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 with Kurt Keutzer and Michael Mahoney at UC Berkeley during my undergrad visit. My reserach mainly focuses on machine learning for system design and efficient machine 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] arXiv 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] arXiv Preprint

QuIP: 2-Bit Quantization of Large Language Models With Guarantees

Jerry Chee, Yaohui Cai, Volodymyr Kuleshov, Chris De Sa.

[paper] [code] Conference on Neural Information Processing Systems (NeurIPS), 2023. Spotlight

A Comprehensive Evaluation of FPGA-Based Spatial Acceleration of LLMs

Hongzheng Chen, Niansong Zhang, Yixiao Du, Zichao Yue, Yaohui Cai, Zhiru Zhang.

[paper] ACM Transactions on Reconfigurable Technology and Systems (TRETS), 2024.

SmoothE: Differentiable E-Graph Extraction

Yaohui Cai, Kaixin Yang, Chenhui Deng, Cunxi Yu, Zhiru Zhang.

International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS), 2025.

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 - Feb. 2024

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.