Yaohui Cai

I am Yaohui Cai (蔡耀辉), a Ph.D. candidate at Cornell University advised by Zhiru Zhang and Chris De Sa. I received my bachelor degree from Peking University. My reserach mainly focuses on machine learning for system design and efficient machine learning.

Email: yc2632 [at] cornell [dot] edu

Google Scholar / GitHub

Selected Publications

HeuriGym: An Agentic Benchmark for LLM-Crafted Heuristics in Combinatorial Optimization

Hongzheng Chen* , Yingheng Wang*, Yaohui Cai*, Hins Hu*, Jiajie Li* , Shirley Huang, Chenhui Deng, Rongjian Liang, Shufeng Kong, Haoxing Ren, Samitha Samaranayake, Carla P. Gomes, Zhiru Zhang.

[paper] [code] arXiv Preprint

SmoothE: Differentiable E-Graph Extraction

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

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

(Best Paper Award)

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)

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

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.

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.