Yihao Zhao｜赵怡浩

I am a final-year Ph.D. candidate in the Department of Computer Science at Peking University, co-advised by Prof. Xuanzhe Liu and Prof. Xin Jin. I received my bachelor degree from Turing Class of Peking University. I was a research intern in the Hyperplane Lab led by Prof. Hao Dong from July 2019 to Sept. 2020.

Email  /  CV  /  GitHub

Research

My research interest lies in AI Infrastructure scheduling and management, e.g., GPU cluster scheduling and datacenter network management.

ARL-Tangram: Unleash the Resource Efficiency in Agentic Reinforcement Learning
Bangjun Xiao*, Yihao Zhao*, Xiangwei Deng, Shihua Yu, Yuxing Xiang, Huaqiu Liu, Qiying Wang, Liang Zhao, Hailin Zhang, Xuanzhe Liu, Xin Jin, Fuli Luo.
In Preprint, 2026
paper

SeaLLM: Service-Aware and Latency-Optimized Resource Sharing for Large Language Model Inference
Yihao Zhao, Jiadun Chen, Peng Sun, Lei Li, Jingxin Li, Xuanzhe Liu, Xin Jin.
In Preprint, 2025
paper

MuxFlow: Efficient GPU Sharing in Production-Level Clusters with More Than 10,000 GPUs
Xuanzhe Liu (Advisor), Yihao Zhao, Shufan Liu, Xiang Li, Yibo Zhu, Xin Liu, Xin Jin.
SCIS, 2024
paper

GreenFlow: A Carbon Efficient Scheduler for Deep Learning Workloads
Diandian Gu, Yihao Zhao, Peng Sun, Xuanzhe Liu, Xin Jin.
TPDS, 2024
paper

How Useful is Communication Scheduling for Distributed Training?
Yihao Zhao, Xuanzhe Liu, Xin Jin.
MoNeTec, 2024, Best Paper Award
paper

Klotski: Efficient and Safe Network Migration of Large Production Datacenters
Yihao Zhao*, Xiaoxiang Zhang*, Hang Zhu, Ying Zhang, Zhaodong Wang, Yuandong Tian, Alex Nikulkov, Joao Ferreira, Xuanzhe Liu, Xin Jin.
SIGCOMM, 2023
paper

ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning
Diandian Gu*, Yihao Zhao*, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, Xuanzhe Liu.
ASPLOS, 2023
paper / code

Multi-Resource Interleaving for Deep Learning Training
Yihao Zhao, Yuanqiang Liu, Yanghua Peng, Yibo Zhu, Xuanzhe Liu, Xin Jin.
SIGCOMM, 2022
paper / code

Unpaired Image-to-Image Translation using Adversarial Consistency Loss
Yihao Zhao, Ruihai Wu, Hao Dong.
ECCV, 2020
paper / code / bibtex

Experiences

Intern

  • Research Intern @ Xiaomi MiMo, Beijing
    • Aug. 2025 - Feb. 2026
    • Mentor: Fuli Luo
  • Research Intern @ Shanghai AI Lab, Beijing
    • June 2024 - Feb. 2025
    • Mentor: Peng Sun
  • Research Intern @ ByteDance, Beijing
    • Jan. 2021 - Oct. 2021, May 2022 - Oct. 2023
    • Mentor: Yanghua Peng, Xin Liu, Yibo Zhu

    Teaching Experience

  • TA, Distributed Maching Learning, PKU (Fall 2022, Fall 2023)
  • TA, Introduction to Computing (A), PKU (Fall 2021)
  • TA, Introduction to Computer Systems, PKU (Fall 2019)