← Search

Yin Tat Lee

18 accepted papers

2024

Differentially Private Synthetic Data via Foundation Model APIs 2: Text

ICML 2024spotlight

Text data has become extremely valuable due to the emergence of machine learning algorithms that learn from it. A lot of high-quality text data generated in the real world is private and therefore cannot be shared or used freely due to privacy concerns. Generating synthetic replicas of private text…

2023

Automatic Prompt Optimization with "Gradient Descent" and Beam Search

EMNLP 2023long main

Large Language Models (LLMs) have shown impressive performance as general purpose agents, but their abilities remain highly dependent on prompts which are hand written with onerous trial-and-error effort. We propose a simple and nonparametric solution to this problem, Prompt Optimization with Textua…

Cited by 0SourcecodeScholar
2023

Exploring the Limits of Differentially Private Deep Learning with Group-wise Clipping

ICLR 2023poster

Differentially private deep learning has recently witnessed advances in computational efficiency and privacy-utility trade-off. We explore whether further improvements along the two axes are possible and provide affirmative answers leveraging two instantiations of \emph{group-wise clipping}. To red…

Cited by 52SourcePDFScholar
2023

Learning threshold neurons via edge of stability

NeurIPS 2023poster

Existing analyses of neural network training often operate under the unrealistic assumption of an extremely small learning rate. This lies in stark contrast to practical wisdom and empirical studies, such as the work of J. Cohen et al. (ICLR 2021), which exhibit startling new phenomena (the "edge of…

Cited by 47SourcePDFScholar
2022

A gradient sampling method with complexity guarantees for Lipschitz functions in high and low dimensions

NeurIPS 2022accept

Zhang et al. (ICML 2020) introduced a novel modification of Goldstein's classical subgradient method, with an efficiency guarantee of $O(\varepsilon^{-4})$ for minimizing Lipschitz functions. Their work, however, makes use of an oracle that is not efficiently implementable. In this paper, we obtain…

Cited by 53SourcePDFScholar
2022

Decomposable Non-Smooth Convex Optimization with Nearly-Linear Gradient Oracle Complexity

NeurIPS 2022accept

Many fundamental problems in machine learning can be formulated by the convex program \[ \min_{\theta\in \mathbb{R}^d}\ \sum_{i=1}^{n}f_{i}(\theta), \] where each $f_i$ is a convex, Lipschitz function supported on a subset of $d_i$ coordinates of $\theta$. One common approach to this problem, exemp…

Cited by 3SourcePDFScholar
2022

Differentially Private Fine-tuning of Language Models

ICLR 2022poster

We give simpler, sparser, and faster algorithms for differentially private fine-tuning of large-scale pre-trained language models, which achieve the state-of-the-art privacy versus utility tradeoffs on many standard NLP tasks. We propose a meta-framework for this problem, inspired by the recent succ…

Cited by 403SourcePDFScholar
2021

Fast and Memory Efficient Differentially Private-SGD via JL Projections

NeurIPS 2021poster

Differentially Private-SGD (DP-SGD) of Abadi et al. and its variations are the only known algorithms for private training of large scale neural networks. This algorithm requires computation of per-sample gradients norms which is extremely slow and memory intensive in practice. In this paper, we pres…

Cited by 48SourcePDFScholar
2021

Lower Bounds on Metropolized Sampling Methods for Well-Conditioned Distributions

NeurIPS 2021oral

We give lower bounds on the performance of two of the most popular sampling methods in practice, the Metropolis-adjusted Langevin algorithm (MALA) and multi-step Hamiltonian Monte Carlo (HMC) with a leapfrog integrator, when applied to well-conditioned distributions. Our main result is a nearly-tigh…

Cited by 35SourcePDFScholar
2020

Network size and size of the weights in memorization with two-layers neural networks

NeurIPS 2020poster

In 1988, Eric B. Baum showed that two-layers neural networks with threshold activation function can perfectly memorize the binary labels of $n$ points in general position in $\R^d$ using only $\ulcorner n/d \urcorner$ neurons. We observe that with ReLU networks, using four times as many neurons one…

Cited by 64SourcePDFScholar
2019

Adversarial examples from computational constraints

ICML 2019oral

Why are classifiers in high dimension vulnerable to “adversarial” perturbations? We show that it is likely not due to information theoretic limitations, but rather it could be due to computational constraints. First we prove that, for a broad set of classification tasks, the mere existence of a robu…

Cited by 262SourcePDFScholar
2018

Optimal Algorithms for Non-Smooth Distributed Optimization in Networks

NeurIPS 2018oral

In this work, we consider the distributed optimization of non-smooth convex functions using a network of computing units. We investigate this problem under two regularity assumptions: (1) the Lipschitz continuity of the global objective function, and (2) the Lipschitz continuity of local individual…

Cited by 194SourcePDFScholar
2017

Optimal Algorithms for Smooth and Strongly Convex Distributed Optimization in Networks

ICML 2017poster

In this paper, we determine the optimal convergence rates for strongly convex and smooth distributed optimization in two settings: centralized and decentralized communications over a network. For centralized (i.e. master/slave) algorithms, we show that distributing Nesterov’s accelerated gradient de…

Cited by 389SourcePDFScholar