Publications

Full publication list. See also my Google Scholar profile.

2026

  1. queenbee.png
    QueenBee Planner: Skill-Evolving Communication Topologies for Token-Efficient LLM Multi-Agent Systems
    Congjia Tian , Yuhang Yao, and Jiaming Cui
    arXiv preprint arXiv:2606.27492, 2026
    PDF
  2. trustworthy_agent_network.png
    Trustworthy Agent Network: Trust in Agent Networks Must Be Baked In, Not Bolted On
    Yixiang Yao , Yuhang Yao, Xinyi Fan , Jiechao Gao , Jie Wang , Minjia Zhang , Srivatsan Ravi , and Carlee Joe-Wong
    KDD Blue Sky Ideas Track, 2026
    PDF Website
  3. twinrouterbench.png
    TwinRouterBench: Fast Static and Live Dynamic Evaluation for Realistic Agentic LLM Routing
    Pei Yang , Wanyi Chen , Tongyun Yang , Pengbin Feng , Jiarong Xing , Wentao Guo , Yuhang Yao, Yuhang Han , Hanchen Li , Xu Wang , Zeyu Wang , Jie Xiao , Anjie Yang , Liang Tian , Lynn Ai , Eric Yang , and Tianyu Shi
    RL-Eval Workshop in ACM Conference on AI and Agentic Systems, 2026
    PDF
  4. vista.png
    VISTA: An End-to-End Benchmark for Visual Spec-to-Web-App Coding Agents
    JunJia Guo , Yuhang Yao, Jiawei Zhou , and Jingdi Chen
    arXiv preprint arXiv:2605.26144, 2026
    PDF Website
  5. dig_to_heal.png
    DIG to Heal: Scaling General-purpose Agent Collaboration via Explainable Dynamic Decision Paths
    Hanqing Yang , Hyungwoo Lee , Yuhang Yao, Zhiwei Liu , Kay Liu , Jingdi Chen , and Carlee Joe-Wong
    arXiv preprint arXiv:2603.00309, 2026
    PDF
  6. rooflinebench.png
    RooflineBench: A Benchmarking Framework for On-Device LLMs via Roofline Analysis
    Zhen Bi , Xueshu Chen , Luoyang Sun , Yuhang Yao, Qing Shen , Jungang Lou , and Cheng Deng
    arXiv preprint arXiv:2602.11506, 2026
    PDF
  7. fedllm.png
    Federated large language models: Current progress and future directions
    Yuhang Yao, Jianyi Zhang , Junda Wu , Chengkai Huang , Yu Xia , Tong Yu , Ruiyi Zhang , Sungchul Kim , Ryan Rossi , Ang Li , Lina Yao , Julian McAuley , Yiran Chen , and Carlee Joe-Wong
    Pacific-Asia Conference on Knowledge Discovery and Data Mining, 2026
    PDF
  8. RL-Eval-CAIS
    MERA: Model Evolution and Routing with Skill Adaptation for Agentic Systems at Scale
    Yuhang Yao, Zeyu Wang , Tongyun Yang , Wanyi Chen , Yuhang Han , Jie Xiao , Chengke Bao , and Tianyu Shi
    RLEval: Methods and Reinforcement Learning Environments for Evaluating AI Agents, 2026
    PDF

2025

  1. gap.png
    GAP: Graph-Based Agent Planning with Parallel Tool Use and Reinforcement Learning
    Jiaqi Wu , Qinlao Zhao , Zefeng Chen , Kai Qin , Yifei Zhao , Xueqian Wang , and Yuhang Yao
    NORA Workshop in NeurIPS, 2025
    PDF Website
  2. delta_sampling.png
    Delta Sampling: Data-Free Knowledge Transfer Across Diffusion Models
    Zhidong Gao , Zimeng Pan , Yuhang Yao, Chenyue Xie , and Wei Wei
    arXiv preprint arXiv:2512.03056, 2025
    PDF
  3. super_agent.png
    Toward super agent system with hybrid ai routers
    Yuhang Yao, Haixin Wang , Yibo Chen , Jiawen Wang , Min Chang Jordan Ren , Bosheng Ding , Salman Avestimehr , and Chaoyang He
    arXiv preprint arXiv:2504.10519, 2025
    PDF Video Blog Slides Website
  4. flammable.png
    FLAMMABLE: A Multi-Model Federated Learning Framework with Multi-Model Engagement and Adaptive Batch Sizes
    Shouxu Lin , Zimeng Pan , Yuhang Yao, Haeyoung Noh , Pei Zhang , and Carlee Joe-Wong
    arXiv preprint arXiv:2510.10380, 2025
    PDF
  5. selective_encryption.png
    Evaluating Selective Encryption Against Gradient Inversion Attacks
    Jiajun Gu , Yuhang Yao, Shuaiqi Wang , and Carlee Joe-Wong
    arXiv preprint arXiv:2508.04155, 2025
    PDF
  6. NPGML-NeurIPS
  7. FL
    Efficient Privacy-Preserving Federated Learning With Selective Parameter Encryption
    Weizhao Jin , Yuhang Yao, Shanshan Han , Jiajun Gu , Carlee Joe-Wong , Srivatsan Ravi , Salman Avestimehr , and Chaoyang He
    2025
    PDF
  8. Security
    Centroid-Based Learning for Malware Detection and Novel Family Identification
    Saranya Vijayakumar , Zifan Wang , Yuhang Yao, and Matt Fredrikson
    2025
    PDF

2024

  1. router.png
    Tensoropera router: A multi-model router for efficient llm inference
    Dimitris Stripelis , Zhaozhuo Xu , Zijian Hu , Alay Dilipbhai Shah , Han Jin , Yuhang Yao, Jipeng Zhang , Tong Zhang , Salman Avestimehr , and Chaoyang He
    EMNLP (Industry Track), 2024
    PDF Slides
  2. scalellm.png
    Scalellm: A resource-frugal LLM serving framework by optimizing end-to-end efficiency
    Yuhang Yao, Han Jin , Alay Dilipbhai Shah , Shanshan Han , Zijian Hu , Dimitris Stripelis , Yide Ran , Zhaozhuo Xu , Salman Avestimehr , and Chaoyang He
    EMNLP (Industry Track), 2024
    PDF Slides
  3. fox.png
    Fox-1: Open Small Language Model for Cloud and Edge
    Zijian Hu , Jipeng Zhang , Rui Pan , Zhaozhuo Xu , Shanshan Han , Han Jin , Alay Dilipbhai Shah , Dimitris Stripelis , Yuhang Yao, Salman Avestimehr , Tong Zhang , and Chaoyang He
    arXiv preprint arXiv:2411.05281, 2024
    PDF Blog Website
  4. alopex.png
    Alopex: A computational framework for enabling on-device function calls with llms
    Yide Ran , Zhaozhuo Xu , Yuhang Yao, Zijian Hu , Shanshan Han , Han Jin , Alay Dilipbhai Shah , Jipeng Zhang , Dimitris Stripelis , Tong Zhang , Salman Avestimehr , and Chaoyang He
    arXiv preprint arXiv:2411.05209, 2024
    PDF
  5. fedgraph.png
    FedGraph: A Research Library and Benchmark for Federated Graph Learning
    Yuhang Yao, Yuan Li , Xinyi Fan , Junhao Li , Kay Liu , Weizhao Jin , Yu Yang , Srivatsan Ravi , Philip S Yu , and Carlee Joe-Wong
    New Perspectives in Graph Machine Learning, NeurIPS, 2024
    PDF Code
  6. fedgat.png
    FedGAT: A Privacy-Preserving Federated Approximation Algorithm for Graph Attention Networks
    Siddharth Ambekar , Yuhang Yao, Ryan Li , and Carlee Joe-Wong
    AAAI Workshop, 2024
    PDF Code
  7. fedgcn.png
    FedGCN: Convergence-Communication Tradeoffs in Federated Training of Graph Convolutional Networks
    Yuhang Yao, Weizhao Jin , Srivatsan Ravi , and Carlee Joe-Wong
    Advances in Neural Information Processing Systems, 2024
    PDF Code Slides
  8. wyzerule.png
    Wyze Rule: Federated Rule Dataset for Rule Recommendation Benchmarking
    Mohammad Mahdi Kamani , Yuhang Yao, Hanjia Lyu , Zhongwei Cheng , Lin Chen , Liangju Li , Carlee Joe-Wong , and Jiebo Luo
    Advances in Neural Information Processing Systems, 2024
    PDF Code Slides Website
  9. fedsecurity.png
    Fedmlsecurity: A benchmark for attacks and defenses in federated learning and llms
    Shanshan Han , Baturalp Buyukates , Zijian Hu , Han Jin , Weizhao Jin , Lichao Sun , Xiaoyang Wang , Wenxuan Wu , Chulin Xie , Yuhang Yao, Kai Zhang , Qifan Zhang , Yuhui Zhang , Carlee Joe-Wong , Salman Avestimehr , and Chaoyang He
    Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, 2024
    PDF Code
  10. torchopera.png
    Torchopera: A compound ai system for llm safety
    Shanshan Han , Zijian Hu , Alay Dilipbhai Shah , Han Jin , Yuhang Yao, Dimitris Stripelis , Zhaozhuo Xu , and Chaoyang He
    arXiv preprint arXiv:2406.10847, 2024
    PDF
  11. multi_agent.png
    LLM multi-agent systems: Challenges and open problems
    Shanshan Han , Qifan Zhang , Yuhang Yao, Weizhao Jin , and Zhaozhuo Xu
    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

2023

  1. fedrule.png
    FedRule: Federated Rule Recommendation System with Graph Neural Networks
    Yuhang Yao, Mohammad Mahdi Kamani , Zhongwei Cheng , Lin Chen , Carlee Joe-Wong , and Tianqiang Liu
    In Proceedings of the 8th ACM/IEEE Conference on Internet of Things Design and Implementation , 2023
    PDF Code Slides
  2. fedhe.png
    FedML-HE: An Efficient Homomorphic-Encryption-Based Privacy-Preserving Federated Learning System
    Weizhao Jin , Yuhang Yao, Shanshan Han , Jiajun Gu , Carlee Joe-Wong , Srivatsan Ravi , Salman Avestimehr , and Chaoyang He
    FL-NeurIPS Workshop, 2023
    PDF Code
  3. kick_bad_guys.png
    Kick bad guys out! zero-knowledge-proof-based anomaly detection in federated learning
    Shanshan Han , Wenxuan Wu , Baturalp Buyukates , Weizhao Jin , Qifan Zhang , Yuhang Yao, Salman Avestimehr , and Chaoyang He
    2023
    PDF

2022

  1. faithful_explanations.png
    Faithful explanations for deep graph models
    Zifan Wang , Yuhang Yao, Chaoran Zhang , Han Zhang , Youjie Kang , Carlee Joe-Wong , Matt Fredrikson , and Anupam Datta
    arXiv preprint arXiv:2205.11850, 2022
    PDF
  2. neuralmab.png
    A neural-based bandit approach to mobile crowdsourcing
    Shouxu Lin , Yuhang Yao, Pei Zhang , Hae Young Noh , and Carlee Joe-Wong
    In Proceedings of the 23rd Annual International Workshop on Mobile Computing Systems and Applications , 2022
    PDF
  3. personalized_fgl.png
    Personalized Federated Graph Learning
    Kiran Pansetty , Yuhang Yao, Mohammad Mahdi Kamani , and Carlee Joe-Wong
    CrossFL-MLSys, 2022

2021

  1. 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
  2. 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

2019

  1. aprp.png
    APRP: An Anonymous Propagation Method in Bitcoin Network
    Yuhang Yao, Xiang Zeng , Tao Cao , Luoyi Fu , and Xinbing Wang
    In Proceedings of the AAAI Conference on Artificial Intelligence , 2019
    PDF

2018

  1. glp.png
    GLP: A novel framework for group-level location promotion in geo-social networks
    Xudong Wu , Luoyi Fu , Yuhang Yao, Xinzhe Fu , Xinbing Wang , and Guihai Chen
    IEEE/ACM Transactions on Networking, 2018
    PDF
  2. evolving_networks.png
    Modeling, analysis and validation of evolving networks with hybrid interactions
    Jiaqi Liu , Luoyi Fu , Yuhang Yao, Xinzhe Fu , Xinbing Wang , and Guihai Chen
    IEEE/ACM Transactions on Networking, 2018
    PDF
  3. TNSE
    Core percolation in interdependent networks
    Jiapeng Zhang , Luoyi Fu , Shuhao Li , Yuhang Yao, and Xinbing Wang
    IEEE Transactions on Network Science and Engineering, 2018

2017

  1. MobiHoc
    Evolving K-graph: Modeling hybrid interactions in networks
    Jiaqi Liu , Yuhang Yao, Xinzhe Fu , Luoyi Fu , Xinyu Liu , and Xinbing Wang
    In Proceedings of the 18th ACM International Symposium on Mobile Ad Hoc Networking and Computing , 2017
  2. MobiHoc
    Core percolation in coupled networks
    Jing Pan , Yuhang Yao, Luoyi Fu , and Xinbing Wang
    In Proceedings of the 18th ACM International Symposium on Mobile Ad Hoc Networking and Computing , 2017