Sara Babakniya

Researcher, Google

Joined

May 2022

Names

Sara Babakniya (Preferred)

Emails

****@usc.edu (Confirmed)

****@google.com (Confirmed)

Personal Links

Career & Education History

Researcher

Google (google.com)

2024 – Present

PhD student

University of Southern California (usc.edu)

2019 – 2025

Advisors, Relations & Conflicts

PhD Advisor

Prof. Salman Avestimehr

2022 – Present

Expertise

Efficient Machine Learning

2021 – Present

Federated Learning

2020 – Present

Publications

Clustering Improves Differentially Private Inference
Kareem Amin, Salman Avestimehr, Sara Babakniya, Alex Bie, Weiwei Kong, Natalia Ponomareva, Umar Syed
- Submitted to ICLR 2026
- Readers: Everyone
Escaping Collapse: The Strength of Weak Data for Large Language Model Training
Kareem Amin, Sara Babakniya, Alex Bie, Weiwei Kong, Umar Syed, Sergei Vassilvitskii
- NeurIPS 2025 poster
- Readers: Everyone
Escaping Collapse: The Strength of Weak Data for Large Language Model Training
Kareem Amin, Sara Babakniya, Alex Bie, Weiwei Kong, Umar Syed, Sergei Vassilvitskii
- SSI-FM Poster
- Readers: Everyone
Escaping Collapse: The Strength of Weak Data for Large Language Model Training
Kareem Amin, Sara Babakniya, Alex Bie, Weiwei Kong, Umar Syed, Sergei Vassilvitskii
- CoRR 2025
- Readers: Everyone
Supervised Learning for Analog and RF Circuit Design: Benchmarks and Comparative Insights
Asal Mehradfar, Xuzhe Zhao, Yue Niu, Sara Babakniya, Mahdi Alesheikh, Hamidreza Aghasi, Salman Avestimehr
- CoRR 2025
- Readers: Everyone
AICircuit: A Multi-Level Dataset and Benchmark for AI-Driven Analog Integrated Circuit Design
Asal Mehradfar, Xuzhe Zhao, Yue Niu, Sara Babakniya, Mahdi Alesheikh, Hamidreza Aghasi, Salman Avestimehr
- CoRR 2024
- Readers: Everyone
SLoRA: Federated Parameter Efficient Fine-Tuning of Language Models
Sara Babakniya, Ahmed Roushdy Elkordy, Yahya H. Ezzeldin, Qingfeng Liu, Kee-Bong Song, MOSTAFA EL-Khamy, Salman Avestimehr
- FL@FM-NeurIPS’23 Oral
- Readers: Everyone
Revisiting Sparsity Hunting in Federated Learning: Why does Sparsity Consensus Matter?
Sara Babakniya, Souvik Kundu, Saurav Prakash, Yue Niu, Salman Avestimehr
- Accepted by TMLR
- Readers: Everyone
Don’t Memorize; Mimic The Past: Federated Class Incremental Learning Without Episodic Memory
Sara Babakniya, Zalan Fabian, Chaoyang He, Mahdi Soltanolkotabi, Salman Avestimehr
- FL-ICML 2023
- Readers: Everyone
A Data-Free Approach to Mitigate Catastrophic Forgetting in Federated Class Incremental Learning for Vision Tasks
Sara Babakniya, Zalan Fabian, Chaoyang He, Mahdi Soltanolkotabi, Salman Avestimehr
- NeurIPS 2023 poster
- Readers: Everyone

View all 20 publications

Co-Authors

View all 30 co-authors