← Search

Kunal Talwar

39 accepted papers

2025

Enabling Differentially Private Federated Learning for Speech Recognition: Benchmarks, Adaptive Optimizers, and Gradient Clipping

NeurIPS 2025poster

While federated learning (FL) and differential privacy (DP) have been extensively studied, their application to automatic speech recognition (ASR) remains largely unexplored due to the challenges in training large transformer models. Specifically, large models further exacerbate issues in FL as they…

Cited by 0SourcecodeScholar
2025

PREAMBLE: Private and Efficient Aggregation via Block Sparse Vectors

NeurIPS 2025poster

We revisit the problem of secure aggregation of high-dimensional vectors in a two-server system such as Prio. These systems are typically used to aggregate vectors such as gradients in private federated learning, where the aggregate itself is protected via noise addition to ensure differential priva…

Cited by 0SourceScholar
2024

Instance-Optimal Private Density Estimation in the Wasserstein Distance

NeurIPS 2024poster

Estimating the density of a distribution from samples is a fundamental problem in statistics. In many practical settings, the Wasserstein distance is an appropriate error metric for density estimation. For example, when estimating population densities in a geographic region, a small Wasserstein dist…

Cited by 1SourcePDFScholar
2024

Private Vector Mean Estimation in the Shuffle Model: Optimal Rates Require Many Messages

ICML 2024poster

We study the problem of private vector mean estimation in the shuffle model of privacy where $n$ users each have a unit vector $v^{(i)} \in \mathbb{R}^d$. We propose a new multi-message protocol that achieves the optimal error using $O(\min(n\varepsilon^2,d))$ messages per user. Moreover, we show th…

Cited by 3SourcePDFScholar
2024

Private and Personalized Frequency Estimation in a Federated Setting

NeurIPS 2024poster

Motivated by the problem of next word prediction on user devices we introduce and study the problem of personalized frequency histogram estimation in a federated setting. In this problem, over some domain, each user observes a number of samples from a distribution which is specific to that user. The…

Cited by 0SourcePDFScholar
2023

Fast Optimal Locally Private Mean Estimation via Random Projections

NeurIPS 2023poster

We study the problem of locally private mean estimation of high-dimensional vectors in the Euclidean ball. Existing algorithms for this problem either incur sub-optimal error or have high communication and/or run-time complexity. We propose a new algorithmic framework, namely ProjUnit, for private m…

2023

Near-Optimal Algorithms for Private Online Optimization in the Realizable Regime

ICML 2023poster

We consider online learning problems in the realizable setting, where there is a zero-loss solution, and propose new Differentially Private (DP) algorithms that obtain near-optimal regret bounds. For the problem of online prediction from experts, we design new algorithms that obtain near-optimal reg…

Cited by 12SourcePDFScholar
2022

Mean Estimation with User-level Privacy under Data Heterogeneity

NeurIPS 2022accept

A key challenge in many modern data analysis tasks is that user data is heterogeneous. Different users may possess vastly different numbers of data points. More importantly, it cannot be assumed that all users sample from the same underlying distribution. This is true, for example in language data,…

Cited by 30SourcePDFScholar
2022

Practical Almost-Linear-Time Approximation Algorithms for Hybrid and Overlapping Graph Clustering

ICML 2022spotlight

Detecting communities in real-world networks and clustering similarity graphs are major data mining tasks with a wide range of applications in graph mining, collaborative filtering, and bioinformatics. In many such applications, overwhelming empirical evidence suggests that communities and clusters…

Cited by 9SourcePDFScholar
2022

Privacy of Noisy Stochastic Gradient Descent: More Iterations without More Privacy Loss

NeurIPS 2022accept

A central issue in machine learning is how to train models on sensitive user data. Industry has widely adopted a simple algorithm: Stochastic Gradient Descent with noise (a.k.a. Stochastic Gradient Langevin Dynamics). However, foundational theoretical questions about this algorithm's privacy loss re…

Cited by 68SourcePDFScholar
2022

Private frequency estimation via projective geometry

ICML 2022spotlight

In this work, we propose a new algorithm ProjectiveGeometryResponse (PGR) for locally differentially private (LDP) frequency estimation. For universe size of k and with n users, our eps-LDP algorithm has communication cost ceil(log_2 k) and computation cost O(n + k\exp(eps) log k) for the server to…

2022

Subspace Recovery from Heterogeneous Data with Non-isotropic Noise

NeurIPS 2022accept

Recovering linear subspaces from data is a fundamental and important task in statistics and machine learning. Motivated by heterogeneity in Federated Learning settings, we study a basic formulation of this problem: the principal component analysis (PCA), with a focus on dealing with irregular noise…

Cited by 8SourcePDFScholar
2021

Characterizing Structural Regularities of Labeled Data in Overparameterized Models

ICML 2021oral

Humans are accustomed to environments that contain both regularities and exceptions. For example, at most gas stations, one pays prior to pumping, but the occasional rural station does not accept payment in advance. Likewise, deep neural networks can generalize across instances that share common pat…

Cited by 119SourcePDFScholar
2021

Private Adaptive Gradient Methods for Convex Optimization

ICML 2021spotlight

We study adaptive methods for differentially private convex optimization, proposing and analyzing differentially private variants of a Stochastic Gradient Descent (SGD) algorithm with adaptive stepsizes, as well as the AdaGrad algorithm. We provide upper bounds on the regret of both algorithms and s…

Cited by 69SourcePDFScholar
2021

Private Stochastic Convex Optimization: Optimal Rates in L1 Geometry

ICML 2021oral

Stochastic convex optimization over an $\ell_1$-bounded domain is ubiquitous in machine learning applications such as LASSO but remains poorly understood when learning with differential privacy. We show that, up to logarithmic factors the optimal excess population loss of any $(\epsilon,\delta)$-dif…

Cited by 114SourcePDFScholar
2020

Faster Differentially Private Samplers via Rényi Divergence Analysis of Discretized Langevin MCMC

NeurIPS 2020poster

Various differentially private algorithms instantiate the exponential mechanism, and require sampling from the distribution $\exp(-f)$ for a suitable function $f$. When the domain of the distribution is high-dimensional, this sampling can be challenging. Using heuristic sampling schemes such as Gibb…

Cited by 48SourcePDFScholar
2020

Stability of Stochastic Gradient Descent on Nonsmooth Convex Losses

NeurIPS 2020spotlight

Uniform stability is a notion of algorithmic stability that bounds the worst case change in the model output by the algorithm when a single data point in the dataset is replaced. An influential work of Hardt et al. [2016] provides strong upper bounds on the uniform stability of the stochastic gradie…

Cited by 223SourcePDFScholar
2020

Stochastic Optimization with Laggard Data Pipelines

NeurIPS 2020poster

State-of-the-art optimization is steadily shifting towards massively parallel pipelines with extremely large batch sizes. As a consequence, CPU-bound preprocessing and disk/memory/network operations have emerged as new performance bottlenecks, as opposed to hardware-accelerated gradient computations…

Cited by 17SourcePDFScholar
2019

Private Stochastic Convex Optimization with Optimal Rates

NeurIPS 2019spotlight

We study differentially private (DP) algorithms for stochastic convex optimization (SCO). In this problem the goal is to approximately minimize the population loss given i.i.d.~samples from a distribution over convex and Lipschitz loss functions. A long line of existing work on private convex optimi…

Cited by 294SourcePDFScholar
2019

Semi-Cyclic Stochastic Gradient Descent

ICML 2019oral

We consider convex SGD updates with a block-cyclic structure, i.e., where each cycle consists of a small number of blocks, each with many samples from a possibly different, block-specific, distribution. This situation arises, e.g., in Federated Learning where the mobile devices available for updates…

Cited by 134SourcePDFScholar
2018

Adversarially Robust Generalization Requires More Data

NeurIPS 2018spotlight

Machine learning models are often susceptible to adversarial perturbations of their inputs. Even small perturbations can cause state-of-the-art classifiers with high "standard" accuracy to produce an incorrect prediction with high confidence. To better understand this phenomenon, we study adversaria…

Cited by 962SourcePDFScholar
2018

Learning Differentially Private Recurrent Language Models

ICLR 2018poster

We demonstrate that it is possible to train large recurrent language models with user-level differential privacy guarantees with only a negligible cost in predictive accuracy. Our work builds on recent advances in the training of deep networks on user-partitioned data and privacy accounting for sto…

Cited by 1654SourcePDFScholar
2018

Online Linear Quadratic Control

ICML 2018oral

We study the problem of controlling linear time-invariant systems with known noisy dynamics and adversarially chosen quadratic losses. We present the first efficient online learning algorithms in this setting that guarantee $O(\sqrt{T})$ regret under mild assumptions, where $T$ is the time horizon.…

Cited by 169SourcePDFScholar
2018

Scalable Private Learning with PATE

ICLR 2018poster

The rapid adoption of machine learning has increased concerns about the privacy implications of machine learning models trained on sensitive data, such as medical records or other personal information. To address those concerns, one promising approach is Private Aggregation of Teacher Ensembles, or…

2017

Semi-supervised Knowledge Transfer for Deep Learning from Private Training Data

ICLR 2017oral

Some machine learning applications involve training data that is sensitive, such as the medical histories of patients in a clinical trial. A model may inadvertently and implicitly store some of its training data; careful analysis of the model may therefore reveal sensitive information. To address t…

Cited by 1306SourcecodeScholar