← Search

Ziteng Sun

22 accepted papers

2026

LAPRAS : Learning-Augmented PRivate Answering for linear query Streams.

ICML 2026poster

Modern database workloads are highly predictable: query streams are dominated by recurring jobs and templates, even when their arrival order is not known in advance. This motivates a learning-augmented view of online differentially private (DP) analytics: can algorithms utilize predictions about *wh…

Cited by 0SourceScholar
2025

Block Verification Accelerates Speculative Decoding

ICLR 2025poster

Speculative decoding is an effective method for lossless acceleration of large language models during inference. It uses a fast model to draft a block of tokens which are then verified in parallel by the target model, and provides a guarantee that the output is distributed identically to a sample f…

Cited by 3SourcePDFScholar
2025

InfAlign: Inference-aware language model alignment

ICML 2025poster

Language model alignment is a critical step in training modern generative language models. Alignment targets to improve win rate of a sample from the aligned model against the base model. Today, we are increasingly using inference-time algorithms (e.g., Best-of-$N$ , controlled decoding, tree search…

Cited by 3SourcePDFScholar
2025

Private Set Union with Multiple Contributions

NeurIPS 2025spotlight

In the private set union problem each user owns a bag of at most $k$ items (from some large universe of items), and we are interested in computing the union of the items in the bags of all of the users. This is trivial without privacy, but a differentially private algorithm must be careful about rep…

Cited by 0SourceScholar
2024

The importance of feature preprocessing for differentially private linear optimization

ICLR 2024poster

Training machine learning models with differential privacy (DP) has received increasing interest in recent years. One of the most popular algorithms for training differentially private models is differentially private stochastic gradient descent (DPSGD) and its variants, where at each step gradients…

Cited by 4SourcePDFScholar
2023

Discrete Distribution Estimation under User-level Local Differential Privacy

AISTATS 2023poster

We study discrete distribution estimation under user-level local differential privacy (LDP). In user-level $\varepsilon$-LDP, each user has a $m\ge1$ samples and the privacy of all $m$ samples must be preserved simultaneously. We resolve the following dilemma: While on the one hand having more sampl…

2023

Federated Heavy Hitter Recovery under Linear Sketching

ICML 2023poster

Motivated by real-life deployments of multi-round federated analytics with secure aggregation, we investigate the fundamental communication-accuracy tradeoffs of the heavy hitter discovery and approximate (open-domain) histogram problems under a linear sketching constraint. We propose efficient algo…

Cited by 2SourcePDFScholar
2023

Sample Complexity of Distinguishing Cause from Effect

AISTATS 2023poster

We study the sample complexity of causal structure learning on a two-variable system with observational and experimental data. Specifically, for two variables $X$ and $Y$, we consider the classical scenario where either $X$ causes $Y$, $Y$ causes $X$, or there is an unmeasured confounder between $X$…

Cited by 4SourcePDFScholar
2023

SpecTr: Fast Speculative Decoding via Optimal Transport

NeurIPS 2023poster

Autoregressive sampling from large language models has led to state-of-the-art results in several natural language tasks. However, autoregressive sampling generates tokens one at a time making it slow, and even prohibitive in certain tasks. One way to speed up sampling is *speculative decoding*: use…

Cited by 86SourcePDFScholar
2023

Subset-Based Instance Optimality in Private Estimation

ICML 2023poster

We propose a new definition of instance optimality for differentially private estimation algorithms. Our definition requires an optimal algorithm to compete, simultaneously for every dataset $D$, with the best private benchmark algorithm that (a) knows $D$ in advance and (b) is evaluated by its wors…

Cited by 9SourcePDFScholar
2023

Unified Lower Bounds for Interactive High-dimensional Estimation under Information Constraints

NeurIPS 2023poster

We consider distributed parameter estimation using interactive protocols subject to local information constraints such as bandwidth limitations, local differential privacy, and restricted measurements. We provide a unified framework enabling us to derive a variety of (tight) minimax lower bounds for…

Cited by 44SourcePDFScholar
2022

Correlated Quantization for Distributed Mean Estimation and Optimization

ICML 2022spotlight

We study the problem of distributed mean estimation and optimization under communication constraints. We propose a correlated quantization protocol whose error guarantee depends on the deviation of data points instead of their absolute range. The design doesn’t need any prior knowledge on the concen…

2021

Distributed Estimation with Multiple Samples per User: Sharp Rates and Phase Transition

NeurIPS 2021poster

We obtain tight minimax rates for the problem of distributed estimation of discrete distributions under communication constraints, where $n$ users observing $m $ samples each can broadcast only $\ell$ bits. Our main result is a tight characterization (up to logarithmic factors) of the error rate as…

Cited by 13SourcePDFScholar
2021

Learning with User-Level Privacy

NeurIPS 2021poster

We propose and analyze algorithms to solve a range of learning tasks under user-level differential privacy constraints. Rather than guaranteeing only the privacy of individual samples, user-level DP protects a user's entire contribution ($m \ge 1$ samples), providing more stringent but more realisti…

Cited by 105SourcePDFScholar
2020

Context Aware Local Differential Privacy

ICML 2020poster

Local differential privacy (LDP) is a strong notion of privacy that often leads to a significant drop in utility. The original definition of LDP assumes that all the elements in the data domain are equally sensitive. However, in many real-life applications, some elements are more sensitive than othe…

Cited by 54SourcePDFScholar
2019

Estimating Entropy of Distributions in Constant Space

NeurIPS 2019poster

We consider the task of estimating the entropy of $k$-ary distributions from samples in the streaming model, where space is limited. Our main contribution is an algorithm that requires $O\left(\frac{k \log (1/\varepsilon)^2}{\varepsilon^3}\right)$ samples and a constant $O(1)$ memory words of space…

Cited by 18SourcePDFScholar
2019

Hadamard Response: Estimating Distributions Privately, Efficiently, and with Little Communication

AISTATS 2019poster

We study the problem of estimating $k$-ary distributions under $\eps$-local differential privacy. $n$ samples are distributed across users who send privatized versions of their sample to a central server. All previously known sample optimal algorithms require linear (in $k$) communication from each…

2018

Differentially Private Testing of Identity and Closeness of Discrete Distributions

NeurIPS 2018spotlight

We study the fundamental problems of identity testing (goodness of fit), and closeness testing (two sample test) of distributions over $k$ elements, under differential privacy. While the problems have a long history in statistics, finite sample bounds for these problems have only been established r…

Cited by 103SourcePDFScholar