Zhuokai Zhao

Zhuokai Zhao

Staff AI Research Scientist, Meta AI
PhD in Computer Science, University of Chicago
Biography

I am a Staff AI Research Scientist at Meta AI. I received my PhD in Computer Science from the University of Chicago, where I was fortunate to be advised by Prof. Yuxin Chen. Before that, I received my Master's degree in Robotics from Johns Hopkins University under the supervision of Prof. Nassir Navab, and my Bachelor's degree with Honors in Electrical Engineering from the University of Illinois at Urbana-Champaign, where I worked on robotics with Prof. Seth Hutchinson.

My research focuses on developing novel training recipes, both data and algorithms, that enhance reasoning in (multimodal) large language models, as well as multi-agent systems that leverage them.

News
Selected Publications * co-first author  ·  † joint last author
2026
Scaling Multi-Agent Intelligence with Substrate-Centric Coordination
Zhenting Qi, Zhuokai Zhao, Rulin Shao, Kerem Oktar, Zhiyuan Yao, Himabindu Lakkaraju, Wen-tau Yih, Ansong Ni, Bhargavi Paranjape, and Yilun Du
Preprint 2026 In submission
SplitJEPA: Learning Invariant and Variant Latent Worlds without Reconstruction
Ruijin Hua, Zichuan Liu, Zhuokai Zhao, and Yujia Zheng
Preprint 2026 In submission
Discovered, Not Designed: Population Evolution for Collaborative and Compute-Intensive Model Discovery
Bo Peng, Lizhu Zhang, Yuhang Zhou, Mingyi Wang, Yifan Wu, Serena Li, Xiangjun Fan, and Zhuokai Zhao
Preprint 2026 In submission
Mixture of Self-Improving Branches For Agent Harness Optimization
Haoyu Dong, Yuhang Zhou, Zihao Lin, Yifan Wu, Bo Peng, Mingyi Wang, Xiangjun Fan, Lizhu Zhang, and Zhuokai Zhao
Preprint 2026 In submission
Rethinking Transfer in Continual Learning: A Replay-Based Realisation
Yang Meng, Zhenya Liu, Zhuokai Zhao†, Yuxin Chen†
Preprint 2026 In submission
Remember When It Matters: Proactive Memory Agent for Long-Horizon Agents
Yifan Wu, Lizhu Zhang, Yuhang Zhou, Mingyi Wang, Bo Peng, Serena Li, Xiangjun Fan, and Zhuokai Zhao
Preprint 2026 In submission
SWE-Together: Evaluating Coding Agents in Interactive User Sessions
Yifan Wu, Zhuokai Zhao, Songlin Li, Ho Hin Lee, Jiacheng Zhu, Shirley Wu, Tianhe Yu, Serena Li, Lizhu Zhang, Xiangjun Fan, and Shengzhi Li
Preprint 2026 In submission
SAGE-OPD: Selective Agent-Guided Intervention for Multi-Turn On-Policy Distillation
Yuhang Zhou, Lizhu Zhang, Yifan Wu, Mingyi Wang, Bo Peng, Jiayi Liu, Xiangjun Fan, and Zhuokai Zhao
Preprint 2026 In submission
OmniOPD: Logit-Free On-Policy Distillation via Speculative Verification
Yuhang Zhou, Lizhu Zhang, Yifan Wu, Mingyi Wang, Bo Peng, Jiayi Liu, Xiangjun Fan, and Zhuokai Zhao
Preprint 2026 In submission
Self-Policy Distillation via Capability-Selective Subspace Projection
Guangya Hao, Yitong Shang, Yunbo Long, and Zhuokai Zhao†, Hanxue Liang†
Preprint 2026 In submission
Self-Evolving Multi-Agent Systems via Decentralized Memory
Guangya Hao, Yunbo Long, and Zhuokai Zhao
Preprint 2026 In submission
TARo: Token-level Adaptive Routing for LLM Test-time Alignment
Arushi Rai, Qiang Zhang, Hanqing Zeng, Yunkai Zhang, Dipesh Tamboli, Xiangjun Fan, and Zhuokai Zhao†, Lizhu Zhang†
ACL 2026 Findings of The 64th Annual Meeting of the Association for Computational Linguistics (ACL)
Mixture-of-Minds: Multi-Agent Reinforcement Learning for Table Understanding
Yuhang Zhou, Mingrui Zhang, Ke Li, Mingyi Wang, Qiao Liu, Qifei Wang, Jiayi Liu, Fei Liu, Serena Li, Weiwei Li, Mingze Gao, Abhishek Kumar, Xiangjun Fan, and Zhuokai Zhao†, Lizhu Zhang†
ACL 2026 The 64th Annual Meeting of the Association for Computational Linguistics (ACL)
Synthetic Sandbox for Training Machine Learning Engineering Agents
Yuhang Zhou*, Lizhu Zhang*, Yifan Wu, Jiayi Liu, Xiangjun Fan, and Zhuokai Zhao†, Hong Yan†
COLM 2026 Conference on Language Modeling (COLM)
Scaling Agent Learning via Experience Synthesis
Zhaorun Chen, Zhuokai Zhao, Kai Zhang, Bo Liu, Qi Qi, Yifan Wu, Tarun Kalluri, Sara Cao, Yuanhao Xiong, Haibo Tong, Huaxiu Yao, Hengduo Li, Jiacheng Zhu, Xian Li, Dawn Song, Bo Li, and Jason Weston†, Dat Huynh†
ICLR 2026 The Fourteenth International Conference on Learning Representations (ICLR)
2025
Exploring System 1 and 2 Communication for Latent Reasoning in LLMs
Julian Coda-Forno, Zhuokai Zhao, Qiang Zhang, Dipesh Tamboli, Weiwei Li, Xiangjun Fan, Lizhu Zhang, Eric Schulz, and Hsiao-Ping Tseng
NeurIPS-W 2025 39th NeurIPS Workshop on Foundations of Reasoning in Language Models
RankCLIP: Ranking-Consistent Language-Image Pretraining
Yiming Zhang*, Zhuokai Zhao*, Zhaorun Chen, Zhili Feng, Zenghui Ding, and Yining Sun
ICCV 2025 International Conference on Computer Vision (ICCV)
Beyond Training: Dynamic Token Merging for Zero-Shot Video Understanding
Yiming Zhang, Zhuokai Zhao, Zhaorun Chen, Zenghui Ding, Xianjun Yang, and Yining Sun
ICCV 2025 International Conference on Computer Vision (ICCV)
Autonomous Multimodal Reasoning via Implicit Chain-of-Vision
Yiqiao Huang*, He Qi*, Zhaorun Chen, Haopeng Zhang, Hanchao Yu, and Zhuokai Zhao
CVPR-W 2025 Oral Presentation CVPR Workshop on Multimodal Algorithmic Reasoning
2024
Direct Acquisition Optimization for Low-Budget Active Learning
Zhuokai Zhao, Yibo Jiang, and Yuxin Chen
NeurIPS-W 2024 Spotlight Talk 38th NeurIPS Workshop on Bayesian Decision-making and Uncertainty
Multimodal Guidance Network for Missing-Modality Inference in Content Moderation
Zhuokai Zhao, Harish Palani, Tianyi Liu, Lena Evans, and Ruth Toner
ICME 2024 IEEE International Conference on Multimedia and Expo (ICME)
AutoPRM: Automating Procedural Supervision for Multi-Step Reasoning via Controllable Question Decomposition
Zhaorun Chen*, Zhuokai Zhao*, Zhihong Zhu*, Ruiqi Zhang, Xiang Li, Bhiksha Raj, and Huaxiu Yao
NAACL 2024 Annual Conference of the North American Chapter of the Association for Computational Linguistics (NAACL) · Preliminary version appeared in ICLR Workshop on Reliable and Responsible Foundation Models
2023
RELAX: Reinforcement Learning Enabled 2D-LiDAR Autonomous System for Parsimonious UAVs
Guanlin Wu, Zhuokai Zhao, and Yutao He
AAAI-W 2023 39th AAAI Workshop on Planning and Reinforcement Learning (PRL)
2020
2026
Scaling Multi-Agent Intelligence with Substrate-Centric Coordination
Zhenting Qi, Zhuokai Zhao, Rulin Shao, Kerem Oktar, Zhiyuan Yao, Himabindu Lakkaraju, Wen-tau Yih, Ansong Ni, Bhargavi Paranjape, and Yilun Du
Preprint 2026 In submission
SplitJEPA: Learning Invariant and Variant Latent Worlds without Reconstruction
Ruijin Hua, Zichuan Liu, Zhuokai Zhao, and Yujia Zheng
Preprint 2026 In submission
Verify Less, Evolve More: Training Idea-Level Critics for Verification-Efficient ML Evolving Agents
Jiamu Bai, Lizhu Zhang, Xin Yu, Yanhong Wu, Zellux Wang, Serena Li, Weiwei Li, Zhuokai Zhao, Lingzhou Xue, Kiwan Maeng, Xiangjun Fan, and Bo Peng
Preprint 2026 In submission
Discovered, Not Designed: Population Evolution for Collaborative and Compute-Intensive Model Discovery
Bo Peng, Lizhu Zhang, Yuhang Zhou, Mingyi Wang, Yifan Wu, Serena Li, Xiangjun Fan, and Zhuokai Zhao
Preprint 2026 In submission
Mixture of Self-Improving Branches For Agent Harness Optimization
Haoyu Dong, Yuhang Zhou, Zihao Lin, Yifan Wu, Bo Peng, Mingyi Wang, Xiangjun Fan, Lizhu Zhang, and Zhuokai Zhao
Preprint 2026 In submission
Knowledge Distillation During Mid-Training Favors Reasoning over Factual Recall
Jacqueline He, Howard Yen, Shuyue Stella Li, Margaret Li, Hanqing Zeng, Yinglong Xia, Benyu Zhang, Zhuokai Zhao, Qiang Zhang, Pang Wei Koh, Luke Zettlemoyer, and Wen-tau Yih
Preprint 2026 In submission
Shared Actors Need Not Share Critics: Effects of Value Mismatch in Parallel Reinforcement Learning
Zhenya Liu, Yang Meng, Zhuokai Zhao, Xuefeng Liu†, Yuxin Chen†
Preprint 2026 In submission
Rethinking Transfer in Continual Learning: A Replay-Based Realisation
Yang Meng, Zhenya Liu, Zhuokai Zhao†, Yuxin Chen†
Preprint 2026 In submission
SVR-R1: Bootstrapping Multi-modal Reasoning with Self-verification in Reinforcement Learning
Mingyuan Wu, Jingcheng Yang, Shengyi Qian, Xudong Wang, Jize Jiang, Qifan Wang, Aashu Singh, Khoi Pham, Fei Liu, Zhaolun Su, Zhuokai Zhao, Klara Nahrstedt, Jianyu Wang, and Hanchao Yu
Preprint 2026 In submission
Remember When It Matters: Proactive Memory Agent for Long-Horizon Agents
Yifan Wu, Lizhu Zhang, Yuhang Zhou, Mingyi Wang, Bo Peng, Serena Li, Xiangjun Fan, and Zhuokai Zhao
Preprint 2026 In submission
SWE-Together: Evaluating Coding Agents in Interactive User Sessions
Yifan Wu, Zhuokai Zhao, Songlin Li, Ho Hin Lee, Jiacheng Zhu, Shirley Wu, Tianhe Yu, Serena Li, Lizhu Zhang, Xiangjun Fan, and Shengzhi Li
Preprint 2026 In submission
SAGE-OPD: Selective Agent-Guided Intervention for Multi-Turn On-Policy Distillation
Yuhang Zhou, Lizhu Zhang, Yifan Wu, Mingyi Wang, Bo Peng, Jiayi Liu, Xiangjun Fan, and Zhuokai Zhao
Preprint 2026 In submission
OmniOPD: Logit-Free On-Policy Distillation via Speculative Verification
Yuhang Zhou, Lizhu Zhang, Yifan Wu, Mingyi Wang, Bo Peng, Jiayi Liu, Xiangjun Fan, and Zhuokai Zhao
Preprint 2026 In submission
Self-Policy Distillation via Capability-Selective Subspace Projection
Guangya Hao, Yitong Shang, Yunbo Long, and Zhuokai Zhao†, Hanxue Liang†
Preprint 2026 In submission
Self-Evolving Multi-Agent Systems via Decentralized Memory
Guangya Hao, Yunbo Long, and Zhuokai Zhao
Preprint 2026 In submission
DAG-MoE: From Simple Mixture to Structural Aggregation in Mixture-of-Experts
Jiarui Feng, Hanqing Zeng, Karish Grover, Ruizhong Qiu, Yinglong Xia, Qiang Zhang, Qifan Wang, Ren Chen, Dongqi Fu, Jiayi Liu, Zhuokai Zhao, Xiangjun Fan, Benyu Zhang, and Yixin Chen
ICML 2026 43rd International Conference on Machine Learning (ICML)
TARo: Token-level Adaptive Routing for LLM Test-time Alignment
Arushi Rai, Qiang Zhang, Hanqing Zeng, Yunkai Zhang, Dipesh Tamboli, Xiangjun Fan, and Zhuokai Zhao†, Lizhu Zhang†
ACL 2026 Findings of The 64th Annual Meeting of the Association for Computational Linguistics (ACL)
Mixture-of-Minds: Multi-Agent Reinforcement Learning for Table Understanding
Yuhang Zhou, Mingrui Zhang, Ke Li, Mingyi Wang, Qiao Liu, Qifei Wang, Jiayi Liu, Fei Liu, Serena Li, Weiwei Li, Mingze Gao, Abhishek Kumar, Xiangjun Fan, and Zhuokai Zhao†, Lizhu Zhang†
ACL 2026 The 64th Annual Meeting of the Association for Computational Linguistics (ACL)
Synthetic Sandbox for Training Machine Learning Engineering Agents
Yuhang Zhou*, Lizhu Zhang*, Yifan Wu, Jiayi Liu, Xiangjun Fan, and Zhuokai Zhao†, Hong Yan†
COLM 2026 Conference on Language Modeling (COLM)
LLM-Driven Reasoning for Constraint-Aware Feature Selection in Industrial Systems
Yuhang Zhou, Zhuokai Zhao, Ke Li, Spilios Evmorfos, Gökalp Demirci, Mingyi Wang, Qiao Liu, Qifei Wang, Serena Li, Weiwei Li, Tingting Wang, Mingze Gao, Gedi Zhou, Abhishek Kumar, Xiangjun Fan, Lizhu Zhang, and Jiayi Liu
Preprint 2026 In submission
Scaling Agent Learning via Experience Synthesis
Zhaorun Chen, Zhuokai Zhao, Kai Zhang, Bo Liu, Qi Qi, Yifan Wu, Tarun Kalluri, Sara Cao, Yuanhao Xiong, Haibo Tong, Huaxiu Yao, Hengduo Li, Jiacheng Zhu, Xian Li, Dawn Song, Bo Li, and Jason Weston†, Dat Huynh†
ICLR 2026 The Fourteenth International Conference on Learning Representations (ICLR)
2025
Exploring System 1 and 2 Communication for Latent Reasoning in LLMs
Julian Coda-Forno, Zhuokai Zhao, Qiang Zhang, Dipesh Tamboli, Weiwei Li, Xiangjun Fan, Lizhu Zhang, Eric Schulz, and Hsiao-Ping Tseng
NeurIPS-W 2025 39th NeurIPS Workshop on Foundations of Reasoning in Language Models
S'MoRE: Structural Mixture of Residual Experts for LLM Fine-tuning
Hanqing Zeng, Yinglong Xia, Zhuokai Zhao, Gilbert Jiang, Qiang Zhang, Jiayi Liu, Lizhu Zhang, Xiangjun Fan, and Benyu Zhang
NeurIPS 2025 Advances in Neural Information Processing Systems (NeurIPS)
MJ-BENCH: Is Your Multimodal Reward Model Really a Good Judge for Text-to-Image Generation?
Zhaorun Chen*, Yichao Du*, Zichen Wen*, Yiyang Zhou*, Chenhang Cui, Zhenzhen Weng, Haoqin Tu, Chaoqi Wang, Zhengwei Tong, Qinglan Huang, Canyu Chen, Qinghao Ye, Zhihong Zhu, Yuqing Zhang, Jiawei Zhou, Zhuokai Zhao, Rafael Rafailov, Chelsea Finn, and Huaxiu Yao
NeurIPS 2025 Advances in Neural Information Processing Systems (NeurIPS)
RecoWorld: Building Simulated Environments for Agentic Recommender Systems
Fei Liu, Xinyu Lin, Hanchao Yu, Mingyuan Wu, Jianyu Wang, Qiang Zhang, Zhuokai Zhao, Yinglong Xia, Yao Zhang, Weiwei Li, Mingze Gao, Qifan Wang, Lizhu Zhang, Benyu Zhang, and Xiangjun Fan
WWW-W 2025 Best Paper Award WWW Workshop on Human-Centric Recommender Systems (HCRS)
RankCLIP: Ranking-Consistent Language-Image Pretraining
Yiming Zhang*, Zhuokai Zhao*, Zhaorun Chen, Zhili Feng, Zenghui Ding, and Yining Sun
ICCV 2025 International Conference on Computer Vision (ICCV)
Beyond Training: Dynamic Token Merging for Zero-Shot Video Understanding
Yiming Zhang, Zhuokai Zhao, Zhaorun Chen, Zenghui Ding, Xianjun Yang, and Yining Sun
ICCV 2025 International Conference on Computer Vision (ICCV)
Autonomous Multimodal Reasoning via Implicit Chain-of-Vision
Yiqiao Huang*, He Qi*, Zhaorun Chen, Haopeng Zhang, Hanchao Yu, and Zhuokai Zhao
CVPR-W 2025 Oral Presentation CVPR Workshop on Multimodal Algorithmic Reasoning
2024
Direct Acquisition Optimization for Low-Budget Active Learning
Zhuokai Zhao, Yibo Jiang, and Yuxin Chen
NeurIPS-W 2024 Spotlight Talk 38th NeurIPS Workshop on Bayesian Decision-making and Uncertainty
EscIRL: Evolving Self-Contrastive IRL for Trajectory Prediction in Autonomous Driving
Siyue Wang*, Zhaorun Chen*, Zhuokai Zhao, Chaoli Mao, Yiyang Zhou, Jiayu He, and Albert Sibo Hu
CoRL 2024 8th Annual Conference on Robot Learning (CoRL)
Multimodal Guidance Network for Missing-Modality Inference in Content Moderation
Zhuokai Zhao, Harish Palani, Tianyi Liu, Lena Evans, and Ruth Toner
ICME 2024 IEEE International Conference on Multimedia and Expo (ICME)
AutoPRM: Automating Procedural Supervision for Multi-Step Reasoning via Controllable Question Decomposition
Zhaorun Chen*, Zhuokai Zhao*, Zhihong Zhu*, Ruiqi Zhang, Xiang Li, Bhiksha Raj, and Huaxiu Yao
NAACL 2024 Annual Conference of the North American Chapter of the Association for Computational Linguistics (NAACL) · Preliminary version appeared in ICLR Workshop on Reliable and Responsible Foundation Models
2023
RELAX: Reinforcement Learning Enabled 2D-LiDAR Autonomous System for Parsimonious UAVs
Guanlin Wu, Zhuokai Zhao, and Yutao He
AAAI-W 2023 39th AAAI Workshop on Planning and Reinforcement Learning (PRL)
2020
Dissertations
Enhanced Data Utilization for Efficient and Trustworthy Deep Learning
Ph.D. in Computer Science, University of Chicago, 2024
Utilizing Both Past and Future: Multi-Frame Memory Based Network in Solving Particle Image Velocimetry
MS in Computer Science, University of Chicago, 2021
Other Projects
Trajectory Planning and Control for Nonholonomic Robot Among Onstacles
Zhuokai Zhao, Mengdi Xu, and Changxin Yan
Nonlinear Control and Planning in Robotics, 2018
Service
Area Chair
Conference Reviewer
Journal Reviewer