A. Avestimehr

Most-cited papers

A Field Guide to Federated Optimization
Jianyu Wang, Zachary B. Charles, Zheng Xu, Gauri Joshi +49 more
2021arXiv.org497 citations
Semantic Scholar
FairFed: Enabling Group Fairness in Federated Learning
Yahya H. Ezzeldin, Shen Yan, Chaoyang He, Emilio Ferrara +1 more
2021AAAI Conference on Artificial Intelligence360 citations Open access
Semantic Scholar
Federated Learning for Internet of Things: Applications, Challenges, and Opportunities
Tuo Zhang, Lei Gao, Chao-Yang He, Mi Zhang +2 more
2021arXiv.org284 citations
Semantic Scholar
FLamby: Datasets and Benchmarks for Cross-Silo Federated Learning in Realistic Healthcare Settings
Jean Ogier du Terrail, Samy Ayed, Edwige Cyffers, Felix Grimberg +20 more
2022Neural Information Processing Systems276 citations Open access
Semantic Scholar
FedGraphNN: A Federated Learning System and Benchmark for Graph Neural Networks
Chaoyang He, Keshav Balasubramanian, Emir Ceyani, Yu Rong +4 more
2021arXiv.org245 citations
Semantic Scholar
FedNLP: Benchmarking Federated Learning Methods for Natural Language Processing Tasks
Bill Yuchen Lin, Chao-Yang He, Zihang Ze, Hu-Lin Wang +6 more
2021NAACL-HLT179 citations Open access
Semantic Scholar
SLoRA: Federated Parameter Efficient Fine-Tuning of Language Models
Sara Babakniya, A. Elkordy, Yahya H. Ezzeldin, Qingfeng Liu +3 more
2023arXiv.org156 citations Open access
Semantic Scholar
Securing Secure Aggregation: Mitigating Multi-Round Privacy Leakage in Federated Learning
Jinhyun So, Ramy E. Ali, Basak Guler, Jian-Tao Jiao +1 more
2021IACR Cryptology ePrint Archive120 citations Open access
Semantic Scholar
FedSpace: An Efficient Federated Learning Framework at Satellites and Ground Stations
Jinhyun So, Kevin Hsieh, Behnaz Arzani, S. Noghabi +2 more
2022arXiv.org106 citations
Semantic Scholar
Federated Learning for the Internet of Things: Applications, Challenges, and Opportunities
Tuo Zhang, Lei Gao, Chaoyang He, Mi Zhang +2 more
2022IEEE Internet of Things Magazine105 citations Open access
Semantic Scholar
Layer-wise Adaptive Model Aggregation for Scalable Federated Learning
Sun-Woo Lee, Tuo Zhang, Chaoyang He, A. Avestimehr
2021AAAI Conference on Artificial Intelligence105 citations Open access
Semantic Scholar
FedCV: A Federated Learning Framework for Diverse Computer Vision Tasks
Chaoyang He, A. Shah, Zhen-Heng Tang, Dian Fan +7 more
2021arXiv.org89 citations
Semantic Scholar
SpreadGNN: Serverless Multi-task Federated Learning for Graph Neural Networks
Chaoyang He, Emir Ceyani, Keshav Balasubramanian, M. Annavaram +1 more
2021arXiv.org68 citations
Semantic Scholar
MARS: Meaning-Aware Response Scoring for Uncertainty Estimation in Generative LLMs
Yavuz Faruk Bakman, D. Yaldiz, Baturalp Buyukates, Chenyang Tao +2 more
2024Annual Meeting of the Association for Computational Linguistics66 citations
Semantic Scholar
SpreadGNN: Decentralized Multi-Task Federated Learning for Graph Neural Networks on Molecular Data
Chaoyang He, Emir Ceyani, Keshav Balasubramanian, M. Annavaram +1 more
2022AAAI Conference on Artificial Intelligence63 citations Open access
Semantic Scholar
FedNLP: A Research Platform for Federated Learning in Natural Language Processing
Bill Yuchen Lin, Chaoyang He, ZiHang Zeng, Hu-Lin Wang +4 more
2021arXiv.org54 citations
Semantic Scholar
How Much Privacy Does Federated Learning with Secure Aggregation Guarantee?
A. Elkordy, Jiang Zhang, Yahya H. Ezzeldin, K. Psounis +1 more
2022Proceedings on Privacy Enhancing Technologies53 citations Open access
Semantic Scholar
Federated Analytics: A survey
A. Elkordy, Yahya H. Ezzeldin, Shan-Shan Han, Sharad Sharma +3 more
2023APSIPA Transactions on Signal and Information Processing47 citations Open access
Semantic Scholar
SSFL: Tackling Label Deficiency in Federated Learning via Personalized Self-Supervision
Chaoyang He, Zhengyu Yang, Erum Mushtaq, Sun-Woo Lee +2 more
2021arXiv.org43 citations
Semantic Scholar
PipeTransformer: Automated Elastic Pipelining for Distributed Training of Transformers
Chaoyang He, Shen Li, M. Soltanolkotabi, A. Avestimehr
2021arXiv.org41 citations
Semantic Scholar
PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models
Chaoyang He, Shen Li, M. Soltanolkotabi, A. Avestimehr
2021International Conference on Machine Learning41 citations
Semantic Scholar
Self-Aware Personalized Federated Learning
Huili Chen, Jie Ding, Eric W. Tramel, Shuang Wu +3 more
2022Neural Information Processing Systems37 citations Open access
Semantic Scholar
Coded Computing for Resilient, Secure, and Privacy-Preserving Distributed Matrix Multiplication
Qian Yu, A. Avestimehr
2021IEEE Transactions on Communications37 citations
Semantic Scholar
Federated Orthogonal Training: Mitigating Global Catastrophic Forgetting in Continual Federated Learning
Yavuz Faruk Bakman, D. Yaldiz, Yahya H. Ezzeldin, A. Avestimehr
2023International Conference on Learning Representations35 citations Open access
Semantic Scholar
LightSecAgg: Rethinking Secure Aggregation in Federated Learning
Chien-Sheng Yang, Jinhyun So, Chaoyang He, Song-Ze Li +2 more
2021arXiv.org34 citations
Semantic Scholar
Secure Aggregation for Buffered Asynchronous Federated Learning
Jinhyun So, Ramy E. Ali, Başak Güler, A. Avestimehr
2021arXiv.org32 citations
Semantic Scholar
Do Not Design, Learn: A Trainable Scoring Function for Uncertainty Estimation in Generative LLMs
D. Yaldiz, Yavuz Faruk Bakman, Baturalp Buyukates, Chenyang Tao +4 more
2024North American Chapter of the Association for Computational Linguistics30 citations
Semantic Scholar
Secure Federated Learning against Model Poisoning Attacks via Client Filtering
D. Yaldiz, Tuo Zhang, A. Avestimehr
2023arXiv.org30 citations Open access
Semantic Scholar
List-Decodable Coded Computing: Breaking the Adversarial Toleration Barrier
M. Soleymani, Ramy E. Ali, Hessam Mahdavifar, A. Avestimehr
2021IEEE Journal on Selected Areas in Information Theory30 citations Open access
Semantic Scholar
Ethos: Rectifying Language Models in Orthogonal Parameter Space
Lei Gao, Yue Niu, Tingting Tang, A. Avestimehr +1 more
2024NAACL-HLT27 citations
Semantic Scholar
Revisiting OPRO: The Limitations of Small-Scale LLMs as Optimizers
Tuo Zhang, Jinyu Yuan, A. Avestimehr
2024Annual Meeting of the Association for Computational Linguistics27 citations
Semantic Scholar
Achieving small-batch accuracy with large-batch scalability via Hessian-aware learning rate adjustment
Sun-Woo Lee, Chaoyang He, A. Avestimehr
2022Neural Networks27 citations Open access
Semantic Scholar
Leveraging Uncertainty Estimation for Efficient LLM Routing
Tuo Zhang, Asal Mehradfar, Dimitrios Dimitriadis, A. Avestimehr
2025arXiv.org26 citations
Semantic Scholar
AICircuit: A Multi-Level Dataset and Benchmark for AI-Driven Analog Integrated Circuit Design
Asal Mehradfar, Xu-Zhe Zhao, Yue Niu, Sara Babakniya +3 more
2024arXiv.org25 citations
Semantic Scholar
Partial Model Averaging in Federated Learning: Performance Guarantees and Benefits
Sun-Woo Lee, Anit Kumar Sahu, Chaoyang He, A. Avestimehr
2022Neurocomputing25 citations Open access
Semantic Scholar
3LegRace: Privacy-Preserving DNN Training over TEEs and GPUs
Yue Niu, Ramy E. Ali, A. Avestimehr
2021Proceedings on Privacy Enhancing Technologies24 citations Open access
Semantic Scholar
SPIDER: Searching Personalized Neural Architecture for Federated Learning
Erum Mushtaq, Chaoyang He, Jie Ding, A. Avestimehr
2021arXiv.org24 citations
Semantic Scholar
FedAIoT: A Federated Learning Benchmark for Artificial Intelligence of Things
Samiul Alam, Tuo Zhang, Tiantian Feng, Hui Shen +7 more
2023J. Data-centric Mach. Learn. Res.22 citations Open access
Semantic Scholar
Revisiting Sparsity Hunting in Federated Learning: Why does Sparsity Consensus Matter?
Sara Babakniya, Souvik Kundu, Saurav Prakash, Yue Niu +1 more
2022Trans. Mach. Learn. Res.22 citations
Semantic Scholar
Federated Learning of Large Models at the Edge via Principal Sub-Model Training
Yue Niu, Saurav Prakash, Souvik Kundu, Sun-Woo Lee +1 more
2022arXiv.org21 citations Open access
Semantic Scholar
Supervised Learning for Analog and RF Circuit Design: Benchmarks and Comparative Insights
Asal Mehradfar, Xu-Zhe Zhao, Yue Niu, Sara Babakniya +3 more
2025arXiv.org20 citations
Semantic Scholar
PolyShard: Coded Sharding Achieves Linearly Scaling Efficiency and Security Simultaneously
Song-Ze Li, Mingchao Yu, Chien-Sheng Yang, A. Avestimehr +2 more
2021IEEE Transactions on Information Forensics and Security20 citations Open access
Semantic Scholar
Reconsidering LLM Uncertainty Estimation Methods in the Wild
Yavuz Faruk Bakman, D. Yaldiz, Sungmin Kang, Tuo Zhang +3 more
2025Annual Meeting of the Association for Computational Linguistics19 citations
Semantic Scholar
Toward a Geometrical Understanding of Self-supervised Contrastive Learning
Romain Cosentino, Anirvan M. Sengupta, A. Avestimehr, M. Soltanolkotabi +3 more
2022arXiv.org19 citations Open access
Semantic Scholar
Secure Aggregation in Federated Learning is not Private: Leaking User Data at Large Scale through Model Modification
Joshua C. Zhao, Atul Sharma, A. Elkordy, Yahya H. Ezzeldin +2 more
2023arXiv.org18 citations Open access
Semantic Scholar
Coded Computing for Secure Boolean Computations
Chien-Sheng Yang, A. Avestimehr
2021IEEE Journal on Selected Areas in Information Theory16 citations Open access
Semantic Scholar
Creating a Lens of Chinese Culture: A Multimodal Dataset for Chinese Pun Rebus Art Understanding
Tuo Zhang, Tiantian Feng, Yibin Ni, Meng Cao +7 more
2024Annual Meeting of the Association for Computational Linguistics15 citations
Semantic Scholar
Secure Federated Clustering
Song-Ze Li, Si-Zai Hou, Baturalp Buyukates, A. Avestimehr
2022IACR Cryptology ePrint Archive15 citations Open access
Semantic Scholar
FALCON: An ML Framework for Fully Automated Layout-Constrained Analog Circuit Design
Asal Mehradfar, Xu-Zhe Zhao, Yi-Lun Huang, Emir Ceyani +4 more
2025Neural Information Processing Systems14 citations
Semantic Scholar
Analog Secret Sharing With Applications to Private Distributed Learning
M. Soleymani, Hessam Mahdavifar, A. Avestimehr
2022IEEE Transactions on Information Forensics and Security13 citations Open access
Semantic Scholar