|
Yitao YANG, Ph.D. student |
I am currently a third-year Ph.D. student at the Department of Computer Science and Engineering of The Chinese University of Hong Kong under the supervision of Prof. Hong Xu and co-supervised by Prof. Baochun Li. Previously, I received my B.Eng. from The Northwestern Polytechnical University (NWPU) in 2023. My research interests lie primarily in the area of AI Agent for Network and AI Infrastructure, and Machine Learning Systems (MLSys).
AI Agent for Network and AI Infrastructure
Efficient Distributed Training & Inference Systems
Ph.D. Computer Science and Engineering, The Chinese University of Hong Kong, Aug. 2023 - Present
B.Eng. Computer Science and Engineering, The Northwestern Polytechnical University, Sep. 2019 - Jul. 2023
TSGuard: Automated User-Centric Incident Diagnosis for AI Workloads in the Cloud
Yitao Yang, Yangtao Deng, Yifan Xiong, Baochun Li, Hong Xu, Peng Cheng
ACM FSE, 2026
Towards End-to-End Optimization of LLM-based Applications with Ayo
Xin Tan, Yimin Jiang, Yitao Yang, Hong Xu
ACM ASPLOS, 2025
Arlo: Serving Transformer-based Language Models with Dynamic Input Lengths
Xin Tan, Jiamin Li, Yitao Yang, Jingzong Li, Hong Xu
ACM ICPP, 2024
Adaptive Gating in Mixture-of-Experts based Language Models
Jiamin Li, Qiang Su, Yitao Yang, Yimin Jiang, Cong Wang, Hong Xu
Empirical Methods in Natural Language Processing (EMNLP), 2023
Intern, ByteDance Network Observability, Mar. 2025 - Nov. 2025
Mentor: Shixian Guo, Zhuo Jiang
Outcome: OneNet Agent, a unified observability platform for ByteDance Network Group (deployed in ByteDance)
Research Intern, MSRA Network Infrastructure Group, Apr. 2024 - Jan. 2025
Mentor: Yifan Xiong
Topic: AI agent for incident diagnosis in AI infrastructure
Outcome: TSGuard@FSE'26
Visiting Student, The Chinese University of Hong Kong, Hong Kong SAR, Oct. 2022 - Jun. 2023
Topic: Efficient MoE Training Systems
Full Postgraduate Scholarship, The Chinese University of Hong Kong, 2023 - 2027
SAMSUNG Scholarship, SAMSUNG, 2020 - 2021
National Scholarship, Ministry of Education, 2019 - 2020
Pacemaker to Merit Student, The Northwestern Polytechnical University, 2019 - 2020