← Search

J Keith Rush

7 accepted papers

2025

Communication-Efficient Language Model Training Scales Reliably and Robustly: Scaling Laws for DiLoCo

NeurIPS 2025spotlight

As we scale to more massive machine learning models, the frequent synchronization demands inherent in data-parallel approaches create significant slowdowns, posing a critical challenge to further scaling. Recent work develops an approach (DiLoCo) that relaxes synchronization demands without compromi…

Cited by 0SourceScholar
2023

(Amplified) Banded Matrix Factorization: A unified approach to private training

NeurIPS 2023poster

Matrix factorization (MF) mechanisms for differential privacy (DP) have substantially improved the state-of-the-art in privacy-utility-computation tradeoffs for ML applications in a variety of scenarios, but in both the centralized and federated settings there remain instances where either MF cannot…

Cited by 35SourcePDFScholar
2023

Gradient Descent with Linearly Correlated Noise: Theory and Applications to Differential Privacy

NeurIPS 2023poster

We study gradient descent under linearly correlated noise. Our work is motivated by recent practical methods for optimization with differential privacy (DP), such as DP-FTRL, which achieve strong performance in settings where privacy amplification techniques are infeasible (such as in federated lear…

Cited by 19SourcePDFScholar
2023

Multi-Epoch Matrix Factorization Mechanisms for Private Machine Learning

ICML 2023oral

We introduce new differentially private (DP) mechanisms for gradient-based machine learning (ML) with multiple passes (epochs) over a dataset, substantially improving the achievable privacy-utility-computation tradeoffs. We formalize the problem of DP mechanisms for adaptive streams with multiple pa…

2022

Improved Differential Privacy for SGD via Optimal Private Linear Operators on Adaptive Streams

NeurIPS 2022accept

Motivated by recent applications requiring differential privacy in the setting of adaptive streams, we investigate the question of optimal instantiations of the matrix mechanism in this setting. We prove fundamental theoretical results on the applicability of matrix factorizations to the adaptive s…

2021

Differentially Private Model Personalization

NeurIPS 2021spotlight

We study personalization of supervised learning with user-level differential privacy. Consider a setting with many users, each of whom has a training data set drawn from their own distribution $P_i$. Assuming some shared structure among the problems $P_i$, can users collectively learn the shared str…

Cited by 41SourcePDFScholar
2021

Federated Reconstruction: Partially Local Federated Learning

NeurIPS 2021poster

Personalization methods in federated learning aim to balance the benefits of federated and local training for data availability, communication cost, and robustness to client heterogeneity. Approaches that require clients to communicate all model parameters can be undesirable due to privacy and commu…