| SLoRA: Federated parameter efficient fine-tuning of language models S Babakniya, AR Elkordy, YH Ezzeldin, Q Liu, KB Song, M El-Khamy, ... FL@FM-NeurIPS workshop 2023, 2023 | 145 | 2023 |
| A data-free approach to mitigate catastrophic forgetting in federated class incremental learning for vision tasks S Babakniya, Z Fabian, C He, M Soltanolkotabi, S Avestimehr Advances in Neural Information Processing Systems 36, 66408-66425, 2023 | 80 | 2023 |
| Revisiting sparsity hunting in federated learning: Why does sparsity consensus matter? S Babakniya, S Kundu, S Prakash, Y Niu, S Avestimehr Transactions on Machine Learning Research, 2023 | 22 | 2023 |
| Federated sparse training: Lottery aware model compression for resource constrained edge S Babakniya, S Kundu, S Prakash, Y Niu, S Avestimehr NeurIPS 2022 Federated Learning Workshop, 2022 | 21 | 2022 |
| AICircuit: A multi-level dataset and benchmark for AI-driven analog integrated circuit design A Mehradfar, X Zhao, Y Niu, S Babakniya, M Alesheikh, H Aghasi, ... AI4Science@NeurIPS 2024, 2024 | 14 | 2024 |
| Defending against poisoning backdoor attacks on federated meta-learning CL Chen, S Babakniya, M Paolieri, L Golubchik ACM Transactions on Intelligent Systems and Technology (TIST) 13 (5), 1-25, 2022 | 10 | 2022 |
| Supervised learning for analog and rf circuit design: Benchmarks and comparative insights A Mehradfar, X Zhao, Y Niu, S Babakniya, M Alesheikh, H Aghasi, ... arXiv preprint arXiv:2501.11839, 2025 | 7 | 2025 |
| Don't Memorize; Mimic The Past: Federated Class Incremental Learning Without Episodic Memory S Babakniya, Z Fabian, C He, M Soltanolkotabi, S Avestimehr ICML 2023 Federated Learning Workshop, 2023 | 7 | 2023 |
| Escaping Collapse: The Strength of Weak Data for Large Language Model Training K Amin, S Babakniya, A Bie, W Kong, U Syed, S Vassilvitskii 39th Annual Conference on Neural Information Processing Systems (NeurIPS 2025), 2025 | 6 | 2025 |
| Clustering and Median Aggregation Improve Differentially Private Inference K Amin, S Avestimehr, S Babakniya, A Bie, W Kong, N Ponomareva, ... arXiv preprint arXiv:2506.04566, 2025 | 4 | 2025 |
| Deep-n-Cheap: An Automated Efficient and Extensible Search Framework for Cost-Effective Deep Learning S Dey, S Babakniya, SC Kanala, M Paolieri, L Golubchik, PA Beerel, ... SN Computer Science 2 (4), 265, 2021 | 2 | 2021 |
| Systems and methods for tuning parameters of a machine learning model for federated learning AR ElKordy, S Babakniya, Q Liu, M El-Khamy, YHE Essa, ... US Patent App. 18/646,660, 2024 | | 2024 |