Yuhang Yao | Homepage

prof_pic.png

Palo Alto, California

[email protected]

I am an incoming Research Engineer at Zoom Agent.

I spent two wonderful years as a funding research engineer at TensorOpera AI & ChainOpera AI. I got my Ph.D. of Electrical Computer Engineering at Carnegie Mellon University in 2024, working with Prof. Carlee Joe-Wong who leads the LIONS research group. I earned my Bachelor of Science degree at the IEEE Honor Class from Shanghai Jiao Tong University, supervised by Prof. Xinbing Wang and Luoyi Fu, in 2019.

My research interests include Super AI Agent System and Federated Graph Learning. Feel free to contact me if you have some research ideas or industry collaborations.

News

Aug 04, 2025 Gave an 1-min talk at Berkeley Agentic AI Summit! Finished 10K (6.2 miles) at San Francisco Marathon!
May 09, 2025 Got the A.G. Milnes Award 2025 (Best PhD Thesis Award at CMU ECE Department)!!
May 07, 2025 Check my interview from MIT Technology Review and DeepTech Wechat Blog.
Mar 01, 2025 Deployed the Super Agent System CoCo at ChainOpera Platform.

Selected Publications

  1. super_agent.png
    Toward super agent system with hybrid ai routers
    Yuhang Yao, Haixin Wang , Yibo Chen , and 5 more authors
    arXiv preprint arXiv:2504.10519, 2025
    PDF Video Blog Slides Website
  2. router.png
    Tensoropera router: A multi-model router for efficient llm inference
    Dimitris Stripelis , Zijian Hu , Jipeng Zhang , and 6 more authors
    EMNLP (Industry Track), 2024
    PDF Slides
  3. scalellm.png
    Scalellm: A resource-frugal LLM serving framework by optimizing end-to-end efficiency
    Yuhang Yao, Han Jin , Alay Dilipbhai Shah , and 7 more authors
    EMNLP (Industry Track), 2024
    PDF Slides
  4. fox.png
    Fox-1: Open Small Language Model for Cloud and Edge
    Zijian Hu , Jipeng Zhang , Rui Pan , and 8 more authors
    arXiv preprint arXiv:2411.05281, 2024
    PDF Blog Website
  5. fedgraph.png
    FedGraph: A Research Library and Benchmark for Federated Graph Learning
    Yuhang Yao, Yuan Li , Xinyi Fan , and 7 more authors
    arXiv preprint arXiv:2410.06340, 2024
    PDF Code
  6. fedgat.png
    FedGAT: A Privacy-Preserving Federated Approximation Algorithm for Graph Attention Networks
    Siddharth Ambekar , Yuhang Yao, Ryan Li , and 1 more author
    arXiv preprint arXiv:2412.16144, 2024
    PDF Code
  7. fedllm.png
    Federated large language models: Current progress and future directions
    Yuhang Yao, Jianyi Zhang , Junda Wu , and 8 more authors
    arXiv preprint arXiv:2409.15723, 2024
    PDF
  8. fedgcn.png
    FedGCN: Convergence-Communication Tradeoffs in Federated Training of Graph Convolutional Networks
    Yuhang Yao, Weizhao Jin , Srivatsan Ravi , and 1 more author
    Advances in Neural Information Processing Systems, 2024
    PDF Video Code Slides
  9. wyzerule.png
    Wyze Rule: Federated Rule Dataset for Rule Recommendation Benchmarking
    Mohammad Mahdi Kamani , Yuhang Yao, Hanjia Lyu , and 5 more authors
    Advances in Neural Information Processing Systems, 2024
    PDF Code Slides Website
  10. fedsecurity.png
    Fedmlsecurity: A benchmark for attacks and defenses in federated learning and llms
    Shanshan Han , Baturalp Buyukates , Zijian Hu , and 8 more authors
    Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, 2024
    PDF Code
  11. multi_agent.png
    LLM multi-agent systems: Challenges and open problems
    Shanshan Han , Qifan Zhang , Yuhang Yao, and 2 more authors
    arXiv preprint arXiv:2402.03578, 2024
    PDF
  12. thesis.png
    Efficient Federated Graph Learning: Formulation, Algorithms and Applications
    Yuhang Yao
    Carnegie Mellon University , 2024
    🏆 A.G. Milnes Award (Best Thesis Award)
    PDF
  13. fedrule.png
    FedRule: Federated Rule Recommendation System with Graph Neural Networks
    Yuhang Yao, Mohammad Mahdi Kamani , Zhongwei Cheng , and 3 more authors
    In Proceedings of the 8th ACM/IEEE Conference on Internet of Things Design and Implementation , 2023
    PDF Code Slides
  14. fedhe.png
    FedML-HE: An Efficient Homomorphic-Encryption-Based Privacy-Preserving Federated Learning System
    Weizhao Jin , Yuhang Yao, Shanshan Han , and 4 more authors
    FL-NeurIPS Workshop, 2023
    PDF Code
  15. neuralmab.png
    A neural-based bandit approach to mobile crowdsourcing
    Shouxu Lin , Yuhang Yao, Pei Zhang , and 2 more authors
    In Proceedings of the 23rd Annual International Workshop on Mobile Computing Systems and Applications , 2022
    PDF
  16. gcnse.png
    Gcn-se: Attention as explainability for node classification in dynamic graphs
    Yucai Fan , Yuhang Yao, and Carlee Joe-Wong
    In 2021 IEEE International Conference on Data Mining (ICDM) , 2021
    PDF Code
  17. rnngcn.png
    Interpretable clustering on dynamic graphs with recurrent graph neural networks
    Yuhang Yao, and Carlee Joe-Wong
    In Proceedings of the AAAI Conference on Artificial Intelligence , 2021
    PDF Code