ADMSv2: A Modern Architecture for Transportation Data Management and Analysis. - Chrysovalantis Anastasiou; Jianfa Lin; Chaoyang He 001; Yao-Yi Chiang; Cyrus Shahabi (ARIC@SIGSPATIAL, 2019)

Collecting Indicators of Compromise from Unstructured Text of Cybersecurity Articles using Neural-Based Sequence Labelling. - Zi Long; Lianzhi Tan; Shengping Zhou; Chaoyang He 001; Xin Liu (IJCNN, 2019)

Adversarial Representation Learning on Large-Scale Bipartite Graphs. - Chaoyang He 001; Tian Xie 005; Yu Rong 001; Wen-bing Huang 001; Junzhou Huang; Xiang Ren 001; Cyrus Shahabi (CoRR, 2019)

Efficient Spatial Anti-Aliasing Rendering for Line Joins on Vector Maps. - Chaoyang He 001; Ming Li (CoRR, 2019)

Central Server Free Federated Learning over Single-sided Trust Social Networks. - Chaoyang He 001; Conghui Tan; Hanlin Tang; Shuang Qiu; Ji Liu 002 (CoRR, 2019)

Advances and Open Problems in Federated Learning. - Peter Kairouz; H. Brendan McMahan; Brendan Avent; Aurélien Bellet; Mehdi Bennis; Arjun Nitin Bhagoji; Kallista A. Bonawitz; Zachary Charles; Graham Cormode; Rachel Cummings; Rafael G. L. D'Oliveira; Salim El Rouayheb; et al. (CoRR, 2019)

MiLeNAS: Efficient Neural Architecture Search via Mixed-Level Reformulation. - Chaoyang He 001; Haishan Ye; Li Shen 005; Tong Zhang 001 (CVPR, 2020)

Group Knowledge Transfer: Federated Learning of Large CNNs at the Edge. - Chaoyang He 001; Murali Annavaram; Salman Avestimehr (NeurIPS, 2020)

FedNAS: Federated Deep Learning via Neural Architecture Search. - Chaoyang He 001; Murali Annavaram; Salman Avestimehr (CoRR, 2020)

FedML: A Research Library and Benchmark for Federated Machine Learning. - Chaoyang He 001; Songze Li; Jinhyun So; Mi Zhang 002; Hongyi Wang 001; Xiaoyang Wang; Praneeth Vepakomma; Abhishek Singh 005; Hang Qiu 001; Li Shen 005; Peilin Zhao; Yan Kang 001; et al. (CoRR, 2020)

Group Knowledge Transfer: Collaborative Training of Large CNNs on the Edge. - Chaoyang He 001; Murali Annavaram; Salman Avestimehr (CoRR, 2020)

PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. - Chaoyang He 001; Shen Li; Mahdi Soltanolkotabi; Salman Avestimehr (ICML, 2021)

MEST: Accurate and Fast Memory-Economic Sparse Training Framework on the Edge. - Geng Yuan; Xiaolong Ma; Wei Niu 002; Zhengang Li; Zhenglun Kong; Ning Liu 007; Yifan Gong 004; Zheng Zhan 001; Chaoyang He 001; Qing Jin; Siyue Wang; Minghai Qin; et al. (NeurIPS, 2021)

Federated Learning for Internet of Things. - Tuo Zhang; Chaoyang He 001; Tianhao Ma; Lei Gao; Mark Ma; Salman Avestimehr (SenSys, 2021)

PipeTransformer: Automated Elastic Pipelining for Distributed Training of Transformers. - Chaoyang He 001; Shen Li; Mahdi Soltanolkotabi; Salman Avestimehr (CoRR, 2021)

FedGraphNN: A Federated Learning System and Benchmark for Graph Neural Networks. - Chaoyang He 001; Keshav Balasubramanian; Emir Ceyani; Yu Rong 001; Peilin Zhao; Junzhou Huang; Murali Annavaram; Salman Avestimehr (CoRR, 2021)

FedNLP: A Research Platform for Federated Learning in Natural Language Processing. - Bill Yuchen Lin; Chaoyang He 001; Zihang Zeng; Hulin Wang; Yufen Huang; Mahdi Soltanolkotabi; Xiang Ren 001; Salman Avestimehr (CoRR, 2021)

Lightweight Image Super-Resolution with Hierarchical and Differentiable Neural Architecture Search. - Han Huang; Li Shen 005; Chaoyang He 001; Weisheng Dong; Haozhi Huang 001; Guangming Shi (CoRR, 2021)

SpreadGNN: Serverless Multi-task Federated Learning for Graph Neural Networks. - Chaoyang He 001; Emir Ceyani; Keshav Balasubramanian; Murali Annavaram; Salman Avestimehr (CoRR, 2021)

Federated Learning for Internet of Things: A Federated Learning Framework for On-device Anomaly Data Detection. - Tuo Zhang; Chaoyang He 001; Tianhao Ma; Mark Ma; Salman Avestimehr (CoRR, 2021)

A Field Guide to Federated Optimization. - Jianyu Wang; Zachary Charles; Zheng Xu 002; Gauri Joshi; H. Brendan McMahan; Blaise Agüera y Arcas; Maruan Al-Shedivat; Galen Andrew; Salman Avestimehr; Katharine Daly; Deepesh Data; Suhas N. Diggavi; et al. (CoRR, 2021)

LightSecAgg: Rethinking Secure Aggregation in Federated Learning. - Chien-Sheng Yang; Jinhyun So; Chaoyang He 001; Songze Li; Qian Yu 001; Salman Avestimehr (CoRR, 2021)

SSFL: Tackling Label Deficiency in Federated Learning via Personalized Self-Supervision. - Chaoyang He 001; Zhengyu Yang 003; Erum Mushtaq; Sunwoo Lee 001; Mahdi Soltanolkotabi; Salman Avestimehr (CoRR, 2021)

Layer-wise Adaptive Model Aggregation for Scalable Federated Learning. - Sunwoo Lee 001; Tuo Zhang; Chaoyang He 001; Salman Avestimehr (CoRR, 2021)

FedCV: A Federated Learning Framework for Diverse Computer Vision Tasks. - Chaoyang He 001; Alay Dilipbhai Shah; Zhenheng Tang; Di Fan; Adarshan Naiynar Sivashunmugam; Keerti Bhogaraju; Mita Shimpi; Li Shen 005; Xiaowen Chu 001; Mahdi Soltanolkotabi; Salman Avestimehr (CoRR, 2021)

AutoCTS: Automated Correlated Time Series Forecasting - Extended Version. - Xinle Wu; Dalin Zhang 001; Chenjuan Guo; Chaoyang He 001; Bin Yang 002; Christian S. Jensen (CoRR, 2021)

SPIDER: Searching Personalized Neural Architecture for Federated Learning. - Erum Mushtaq; Chaoyang He 001; Jie Ding 002; Salman Avestimehr (CoRR, 2021)

Advances and Open Problems in Federated Learning. - Peter Kairouz; H. Brendan McMahan; Brendan Avent; Aurélien Bellet; Mehdi Bennis; Arjun Nitin Bhagoji; Kallista A. Bonawitz; Zachary Charles; Graham Cormode; Rachel Cummings; Rafael G. L. D'Oliveira; Hubert Eichner; et al. (Found. Trends Mach. Learn., 2021)

AutoCTS: Automated Correlated Time Series Forecasting. - Xinle Wu; Dalin Zhang 001; Chenjuan Guo; Chaoyang He 001; Bin Yang 002; Christian S. Jensen (Proc. VLDB Endow., 2021)

SpreadGNN: Decentralized Multi-Task Federated Learning for Graph Neural Networks on Molecular Data. - Chaoyang He 001; Emir Ceyani; Keshav Balasubramanian; Murali Annavaram; Salman Avestimehr (AAAI, 2022)

The 1st International Workshop on Federated Learning with Graph Data (FedGraph). - Carl Yang 001; Xiaoxiao Li; Nathalie Baracaldo; Neil Shah; Chaoyang He 001; Lingjuan Lyu; Lichao Sun 001; Salman Avestimehr (CIKM, 2022)

FedNLP: Benchmarking Federated Learning Methods for Natural Language Processing Tasks. - Bill Yuchen Lin; Chaoyang He 001; Zihang Ze; Hulin Wang; Yufen Hua; Christophe Dupuy; Rahul Gupta 001; Mahdi Soltanolkotabi; Xiang Ren 001; Salman Avestimehr (NAACL-HLT (Findings), 2022)

FLamby: Datasets and Benchmarks for Cross-Silo Federated Learning in Realistic Healthcare Settings. - Jean Ogier du Terrail; Samy-Safwan Ayed; Edwige Cyffers; Felix Grimberg; Chaoyang He 001; Regis Loeb; Paul Mangold; Tanguy Marchand; Othmane Marfoq; Erum Mushtaq; Boris Muzellec; Constantin Philippenko; et al. (NeurIPS, 2022)

SMILE: Scaling Mixture-of-Experts with Efficient Bi-level Routing. - Chaoyang He 001; Shuai Zheng 004; Aston Zhang; George Karypis; Trishul Chilimbi; Mahdi Soltanolkotabi; Salman Avestimehr (CoRR, 2022)

Federated Learning for the Internet of Things: Applications, Challenges, and Opportunities. - Tuo Zhang; Lei Gao; Chaoyang He 001; Mi Zhang 002; Bhaskar Krishnamachari; Amir Salman Avestimehr (IEEE Internet Things Mag., 2022)

FairFed: Enabling Group Fairness in Federated Learning. - Yahya H. Ezzeldin; Shen Yan; Chaoyang He 001; Emilio Ferrara; Amir Salman Avestimehr (AAAI, 2023)

Proof-of-Contribution-Based Design for Collaborative Machine Learning on Blockchain. - Baturalp Buyukates; Chaoyang He 001; Shanshan Han; Zhiyong Fang; Yupeng Zhang 001; Jieyi Long; Ali Farahanchi; Salman Avestimehr (DAPPS, 2023)

PyTorch RPC: Distributed Deep Learning Built on Tensor-Optimized Remote Procedure Calls. - Shen Li; Pritam Damania; Luca Wehrstedt; Rohan Varma; Omkar Salpekar; Pavel Belevich; Howard Huang; Yanli Zhao; Lucas Hosseini; Wanchao Liang; Hongyi Jia; Shihao Xu; et al. (MLSys, 2023)

A Data-Free Approach to Mitigate Catastrophic Forgetting in Federated Class Incremental Learning for Vision Tasks. - Sara Babakniya; Zalan Fabian; Chaoyang He 001; Mahdi Soltanolkotabi; Salman Avestimehr (NeurIPS, 2023)

Federated Analytics: A survey. - Ahmed Roushdy Elkordy; Yahya H. Ezzeldin; Shanshan Han; Shantanu Sharma 001; Chaoyang He 001; Sharad Mehrotra; Salman Avestimehr (CoRR, 2023)

FedML Parrot: A Scalable Federated Learning System via Heterogeneity-aware Scheduling on Sequential and Hierarchical Training. - Zhenheng Tang; Xiaowen Chu 001; Ryan Yide Ran; Sunwoo Lee 001; Shaohuai Shi; Yonggang Zhang 003; Yuxin Wang 003; Alex Qiaozhong Liang; Salman Avestimehr; Chaoyang He 001 (CoRR, 2023)

FedML-HE: An Efficient Homomorphic-Encryption-Based Privacy-Preserving Federated Learning System. - Weizhao Jin; Yuhang Yao; Shanshan Han; Carlee Joe-Wong; Srivatsan Ravi; Salman Avestimehr; Chaoyang He 001 (CoRR, 2023)

FedMLSecurity: A Benchmark for Attacks and Defenses in Federated Learning and LLMs. - Shanshan Han; Baturalp Buyukates; Zijian Hu 001; Han Jin; Weizhao Jin; Lichao Sun 001; Xiaoyang Wang; Chulin Xie; Kai Zhang 039; Qifan Zhang 002; Yuhui Zhang; Chaoyang He 001; et al. (CoRR, 2023)

Don't Memorize; Mimic The Past: Federated Class Incremental Learning Without Episodic Memory. - Sara Babakniya; Zalan Fabian; Chaoyang He 001; Mahdi Soltanolkotabi; Salman Avestimehr (CoRR, 2023)

Kick Bad Guys Out! Zero-Knowledge-Proof-Based Anomaly Detection in Federated Learning. - Shanshan Han; Wenxuan Wu; Baturalp Buyukates; Weizhao Jin; Yuhang Yao; Qifan Zhang 002; Salman Avestimehr; Chaoyang He 001 (CoRR, 2023)

Distributed Architecture Search Over Heterogeneous Distributions. - Erum Mushtaq; Chaoyang He 001; Jie Ding 002; Salman Avestimehr (Trans. Mach. Learn. Res., 2023)

Achieving small-batch accuracy with large-batch scalability via Hessian-aware learning rate adjustment. - Sunwoo Lee 001; Chaoyang He 001; Salman Avestimehr (Neural Networks, 2023)

Differentiable Neural Architecture Search for Extremely Lightweight Image Super-Resolution. - Han Huang; Li Shen 008; Chaoyang He 001; Weisheng Dong; Wei Liu 005 (IEEE Trans. Circuits Syst. Video Technol., 2023)

Partial model averaging in Federated Learning: Performance guarantees and benefits. - Sunwoo Lee 001; Anit Kumar Sahu; Chaoyang He 001; Salman Avestimehr (Neurocomputing, 2023)

Accelerated Distributed Approximate Newton Method. - Haishan Ye; Chaoyang He 001; Xiangyu Chang (IEEE Trans. Neural Networks Learn. Syst., 2023)

L-BGNN: Layerwise Trained Bipartite Graph Neural Networks. - Tian Xie 005; Chaoyang He 001; Xiang Ren 001; Cyrus Shahabi; C.-C. Jay Kuo (IEEE Trans. Neural Networks Learn. Syst., 2023)

TensorOpera Router: A Multi-Model Router for Efficient LLM Inference. - Dimitris Stripelis; Zhaozhuo Xu; Zijian Hu 001; Alay Dilipbhai Shah; Han Jin; Yuhang Yao; Jipeng Zhang; Tong Zhang 001; Salman Avestimehr; Chaoyang He 001 (EMNLP (Industry Track), 2024)

ScaleLLM: A Resource-Frugal LLM Serving Framework by Optimizing End-to-End Efficiency. - Yuhang Yao; Han Jin; Alay Dilipbhai Shah; Shanshan Han; Zijian Hu 001; Dimitris Stripelis; Yide Ran; Zhaozhuo Xu; Salman Avestimehr; Chaoyang He 001 (EMNLP (Industry Track), 2024)

FedSecurity: A Benchmark for Attacks and Defenses in Federated Learning and Federated LLMs. - Shanshan Han; Baturalp Buyukates; Zijian Hu 001; Han Jin; Weizhao Jin; Lichao Sun 001; Xiaoyang Wang; Wenxuan Wu; Chulin Xie; Yuhang Yao; Kai Zhang 039; Qifan Zhang 002; et al. (KDD, 2024)

LLM Multi-Agent Systems: Challenges and Open Problems. - Shanshan Han; Qifan Zhang 002; Yuhang Yao; Weizhao Jin; Zhaozhuo Xu; Chaoyang He 001 (CoRR, 2024)

TorchOpera: A Compound AI System for LLM Safety. - Shanshan Han; Yuhang Yao; Zijian Hu 001; Dimitris Stripelis; Zhaozhuo Xu; Chaoyang He 001 (CoRR, 2024)

PolyRouter: A Multi-LLM Querying System. - Dimitris Stripelis; Zijian Hu 001; Jipeng Zhang; Zhaozhuo Xu; Alay Dilipbhai Shah; Han Jin; Yuhang Yao; Salman Avestimehr; Chaoyang He 001 (CoRR, 2024)

Alopex: A Computational Framework for Enabling On-Device Function Calls with LLMs. - Yide Ran; Zhaozhuo Xu; Yuhang Yao; Zijian Hu 001; Shanshan Han; Han Jin; Alay Dilipbhai Shah; Jipeng Zhang; Dimitris Stripelis; Tong Zhang 001; Salman Avestimehr; Chaoyang He 001 (CoRR, 2024)

Fox-1 Technical Report. - Zijian Hu 001; Jipeng Zhang; Rui Pan 002; Zhaozhuo Xu; Shanshan Han; Han Jin; Alay Dilipbhai Shah; Dimitris Stripelis; Yuhang Yao; Salman Avestimehr; Chaoyang He 001; Tong Zhang 001 (CoRR, 2024)

Bridging the Safety Gap: A Guardrail Pipeline for Trustworthy LLM Inferences. - Shanshan Han; Salman Avestimehr; Chaoyang He 001 (CoRR, 2025)

Toward Super Agent System with Hybrid AI Routers. - Yuhang Yao; Haixin Wang; Yibo Chen; Jiawen Wang; Min Chang Jordan Ren; Bosheng Ding; Salman Avestimehr; Chaoyang He 001 (CoRR, 2025)