← Search

Steven Wu

73 accepted papers

2026

All Roads Lead to Likelihood: The Value of Reinforcement Learning in Fine-Tuning

ICLR 2026poster

From a first-principles perspective, it may seem odd that the strongest results in foundation model fine-tuning (FT) are achieved via a relatively complex, two-stage training procedure. Specifically, one first trains a reward model (RM) on some dataset (e.g., human preferences) before using it to pr…

Cited by 62SourceScholar
2026

Doubly-Robust LLM-as-a-Judge: Externally Valid Estimation with Imperfect Personas

ICLR 2026poster

As Generative AI (GenAI) systems see growing adoption, a key concern involves the external validity of evaluations, or the extent to which they generalize from lab-based to real-world deployment conditions. Threats to the external validity of GenAI evaluations arise when the source sample of human r…

Cited by 0SourceScholar
2026

Nearly-Optimal Bandit Learning in Stackelberg Games with Side Information

ICLR 2026poster

We study the problem of online learning in Stackelberg games with side information between a leader and a sequence of followers. In every round the leader observes contextual information and commits to a mixed strategy, after which the follower best-responds. We provide learning algorithms for the l…

Cited by 0SourceScholar
2026

Optimizing Canaries for Privacy Auditing with Metagradient Descent

ICLR 2026poster

In this work we study black-box privacy auditing, where the goal is to lower bound the privacy parameter of a differentially private learning algorithm using only the algorithm’s outputs (i.e., final trained model). For DP-SGD (the most successful method for training differentially private deep lear…

Cited by 0SourceScholar
2025

Discretization-free Multicalibration through Loss Minimization over Tree Ensembles

NeurIPS 2025poster

In recent years, multicalibration has emerged as a desirable learning objective for ensuring that a predictor is calibrated across a rich collection of overlapping subpopulations. Existing approaches typically achieve multicalibration by discretizing the predictor's output space and iteratively adju…

Cited by 0SourcecodeScholar
2025

Kandinsky Conformal Prediction: Beyond Class- and Covariate-Conditional Coverage

ICML 2025poster

Conformal prediction is a powerful distribution-free framework for constructing prediction sets with coverage guarantees. Classical methods, such as split conformal prediction, provide marginal coverage, ensuring that the prediction set contains the label of a random test point with a target probabi…

Cited by 0SourcePDFScholar
2025

Leveraging Model Guidance to Extract Training Data from Personalized Diffusion Models

ICML 2025poster

Diffusion Models (DMs) have evolved into advanced image generation tools, especially for few-shot fine-tuning where a pretrained DM is fine-tuned on a small set of images to capture specific styles or objects. Many people upload these personalized checkpoints online, fostering communities such as Ci…

Cited by 0SourcePDFScholar
2025

Persona-Augmented Benchmarking: Evaluating LLMs Across Diverse Writing Styles

EMNLP 2025

Current benchmarks for evaluating Large Language Models (LLMs) often do not exhibit enough writing style diversity, with many adhering primarily to standardized conventions. Such benchmarks do not fully capture the rich variety of communication patterns exhibited by humans. Thus, it is possible that

Cited by 0SourcePDFScholar
2025

Predicting Language Models’ Success at Zero-Shot Probabilistic Prediction

EMNLP 2025

Recent work has investigated the capabilities of large language models (LLMs) as zero-shot models for generating individual-level characteristics (e.g., to serve as risk models or augment survey datasets). However, when should a user have confidence that an LLM will provide high-quality predictions

2025

Unlearned but Not Forgotten: Data Extraction after Exact Unlearning in LLM

NeurIPS 2025poster

Large Language Models are typically trained on datasets collected from the web, which may inadvertently contain harmful or sensitive personal information. To address growing privacy concerns, unlearning methods have been proposed to remove the influence of specific data from trained models. Of these…

Cited by 0SourcecodeScholar
2025

Unlearning or Obfuscating? Jogging the Memory of Unlearned LLMs via Benign Relearning

ICLR 2025poster

Machine unlearning is a promising approach to mitigate undesirable memorization of training data in ML models. However, in this work we show that existing approaches for unlearning in LLMs are surprisingly susceptible to a simple set of benign relearning attacks. With access to only a small and pote…

Cited by 1SourcePDFScholar
2025

Utility-Directed Conformal Prediction: A Decision-Aware Framework for Actionable Uncertainty Quantification

ICLR 2025poster

There is increasing interest in ``decision-focused" machine learning methods which train models to account for how their predictions are used in downstream optimization problems. Doing so can often improve performance on subsequent decision problems. However, current methods for uncertainty quantifi…

Cited by 0SourcePDFScholar
2025

Validating LLM-as-a-Judge Systems under Rating Indeterminacy

NeurIPS 2025poster

The LLM-as-a-judge paradigm, in which a judge LLM system replaces human raters in rating the outputs of other generative AI (GenAI) systems, plays a critical role in scaling and standardizing GenAI evaluations. To validate such judge systems, evaluators assess human--judge agreement by first collect…

Cited by 0SourceScholar
2024

A Minimaximalist Approach to Reinforcement Learning from Human Feedback

ICML 2024poster

We present *Self-Play Preference Optimization* (SPO), an algorithm for reinforcement learning from human feedback. Our approach is *minimalist* in that it does not require training a reward model nor unstable adversarial training and is therefore rather simple to implement. Our approach is *maximali…

Cited by 87SourcePDFScholar
2024

Bridging Multicalibration and Out-of-distribution Generalization Beyond Covariate Shift

NeurIPS 2024poster

We establish a new model-agnostic optimization framework for out-of-distribution generalization via multicalibration, a criterion that ensures a predictor is calibrated across a family of overlapping groups. Multicalibration is shown to be associated with robustness of statistical inference under co…

Cited by 1SourcePDFScholar
2024

Differentially Private SGD Without Clipping Bias: An Error-Feedback Approach

ICLR 2024poster

Differentially Private Stochastic Gradient Descent with Gradient Clipping (DPSGD-GC) is a powerful tool for training deep learning models using sensitive data, providing both a solid theoretical privacy guarantee and high efficiency. However, existing research has shown that DPSGD-GC only converges…

2024

Hybrid Inverse Reinforcement Learning

ICML 2024poster

The inverse reinforcement learning approach to imitation learning is a double-edged sword. On the one hand, it can enable learning from a smaller number of expert demonstrations with more robustness to error compounding than behavioral cloning approaches. On the other hand, it requires that the lear…

2024

Membership Inference Attacks on Diffusion Models via Quantile Regression

ICML 2024poster

Recently, diffusion models have become popular tools for image synthesis due to their high-quality outputs. However, like other large models, they may leak private information about their training data. Here, we demonstrate a privacy vulnerability of diffusion models through a *membership inference…

Cited by 9SourcePDFScholar
2024

On the Benefits of Public Representations for Private Transfer Learning under Distribution Shift

NeurIPS 2024poster

Public pretraining is a promising approach to improve differentially private model training. However, recent work has noted that many positive research results studying this paradigm only consider in-distribution tasks, and may not apply to settings where there is distribution shift between the pret…

Cited by 2SourcePDFScholar
2024

Oracle-Efficient Differentially Private Learning with Public Data

NeurIPS 2024poster

Due to statistical lower bounds on the learnability of many function classes under privacy constraints, there has been recent interest in leveraging public data to improve the performance of private learning algorithms. In this model, algorithms must always guarantee differential privacy with respec…

Cited by 8SourcePDFScholar
2024

Predictive Performance Comparison of Decision Policies Under Confounding

ICML 2024poster

Predictive models are often introduced to decision-making tasks under the rationale that they improve performance over an existing decision-making policy. However, it is challenging to compare predictive performance against an existing decision-making policy that is generally under-specified and dep…

2024

Reconstruction Attacks on Machine Unlearning: Simple Models are Vulnerable

NeurIPS 2024poster

Machine unlearning is motivated by principles of data autonomy. The premise is that a person can request to have their data's influence removed from deployed models, and those models should be updated as if they were retrained without the person's data. We show that these updates expose individuals…

Cited by 8SourcePDFScholar
2023

Adaptive Principal Component Regression with Applications to Panel Data

NeurIPS 2023poster

Principal component regression (PCR) is a popular technique for fixed-design error-in-variables regression, a generalization of the linear regression setting in which the observed covariates are corrupted with random noise. We provide the first time-uniform finite sample guarantees for online (regul…

Cited by 7SourcePDFScholar
2023

Adaptive Privacy Composition for Accuracy-first Mechanisms

NeurIPS 2023poster

Although there has been work to develop ex-post private mechanisms from Ligett et al. '17 and Whitehouse et al '22 that seeks to provide privacy guarantees subject to a target level of accuracy, there was not a way to use them in conjunction with differentially private mechanisms. Furthermore, ther…

Cited by 3SourcePDFScholar
2023

Generating Private Synthetic Data with Genetic Algorithms

ICML 2023poster

We study the problem of efficiently generating differentially private synthetic data that approximate the statistical properties of an underlying sensitive dataset. In recent years, there has been a growing line of work that approaches this problem using first-order optimization techniques. However,…

2023

Inverse Reinforcement Learning without Reinforcement Learning

ICML 2023poster

Inverse Reinforcement Learning (IRL) is a powerful set of techniques for imitation learning that aims to learn a reward function that rationalizes expert demonstrations. Unfortunately, traditional IRL methods suffer from a computational weakness: they require repeatedly solving a hard reinforcement…

2023

Learning Shared Safety Constraints from Multi-task Demonstrations

NeurIPS 2023poster

Regardless of the particular task we want to perform in an environment, there are often shared safety constraints we want our agents to respect. For example, regardless of whether it is making a sandwich or clearing the table, a kitchen robot should not break a plate. Manually specifying such a cons…

2023

Meta-Learning Adversarial Bandit Algorithms

NeurIPS 2023poster

We study online meta-learning with bandit feedback, with the goal of improving performance across multiple tasks if they are similar according to some natural similarity measure. As the first to target the adversarial online-within-online partial-information setting, we design meta-algorithms that…

Cited by 4SourcePDFScholar
2023

Meta-Learning in Games

ICLR 2023poster

In the literature on game-theoretic equilibrium finding, focus has mainly been on solving a single game in isolation. In practice, however, strategic interactions—ranging from routing problems to online advertising auctions—evolve dynamically, thereby leading to many similar games to be solved. To a…

Cited by 22SourcePDFScholar
2023

Nonparametric Extensions of Randomized Response for Private Confidence Sets

ICML 2023oral

This work derives methods for performing nonparametric, nonasymptotic statistical inference for population means under the constraint of local differential privacy (LDP). Given bounded observations $(X_1, \dots, X_n)$ with mean $\mu^\star$ that are privatized into $(Z_1, \dots, Z_n)$, we present con…

2023

Reinforcement Learning with Stepwise Fairness Constraints

AISTATS 2023poster

AI methods are used in societally important settings, ranging from credit to employment to housing, and it is crucial to provide fairness in regard to automated decision making. Moreover, many settings are dynamic, with populations responding to sequential decision policies. We introduce the study o…

Cited by 15SourcePDFScholar
2023

Scalable Membership Inference Attacks via Quantile Regression

NeurIPS 2023poster

Membership inference attacks are designed to determine, using black box access to trained models, whether a particular example was used in training or not. Membership inference can be formalized as a hypothesis testing problem. The most effective existing attacks estimate the distribution of some te…

2022

Bayesian Persuasion for Algorithmic Recourse

NeurIPS 2022accept

When subjected to automated decision-making, decision subjects may strategically modify their observable features in ways they believe will maximize their chances of receiving a favorable decision. In many practical situations, the underlying assessment rule is deliberately kept secret to avoid gami…

Cited by 18SourcePDFScholar
2022

Brownian Noise Reduction: Maximizing Privacy Subject to Accuracy Constraints

NeurIPS 2022accept

There is a disconnect between how researchers and practitioners handle privacy-utility tradeoffs. Researchers primarily operate from a privacy first perspective, setting strict privacy requirements and minimizing risk subject to these constraints. Practitioners often desire an accuracy first perspec…

Cited by 10SourcePDFScholar
2022

Causal Imitation Learning under Temporally Correlated Noise

ICML 2022oral

We develop algorithms for imitation learning from policy data that was corrupted by temporally correlated noise in expert actions. When noise affects multiple timesteps of recorded data, it can manifest as spurious correlations between states and actions that a learner might latch on to, leading to…

2022

Constrained Variational Policy Optimization for Safe Reinforcement Learning

ICML 2022spotlight

Safe reinforcement learning (RL) aims to learn policies that satisfy certain constraints before deploying them to safety-critical applications. Previous primal-dual style approaches suffer from instability issues and lack optimality guarantees. This paper overcomes the issues from the perspective of…

2022

Improved Regret for Differentially Private Exploration in Linear MDP

ICML 2022spotlight

We study privacy-preserving exploration in sequential decision-making for environments that rely on sensitive data such as medical records. In particular, we focus on solving the problem of reinforcement learning (RL) subject to the constraint of (joint) differential privacy in the linear MDP settin…

Cited by 10SourcePDFScholar
2022

Minimax Optimal Online Imitation Learning via Replay Estimation

NeurIPS 2022accept

Online imitation learning is the problem of how best to mimic expert demonstrations, given access to the environment or an accurate simulator. Prior work has shown that in the \textit{infinite} sample regime, exact moment matching achieves value equivalence to the expert policy. However, in the \tex…

2022

On Privacy and Personalization in Cross-Silo Federated Learning

NeurIPS 2022accept

While the application of differential privacy (DP) has been well-studied in cross-device federated learning (FL), there is a lack of work considering DP and its implications for cross-silo FL, a setting characterized by a limited number of clients each containing many data subjects. In cross-silo FL…

2022

Personalization Improves Privacy-Accuracy Tradeoffs in Federated Learning

ICML 2022spotlight

Large-scale machine learning systems often involve data distributed across a collection of users. Federated learning algorithms leverage this structure by communicating model updates to a central server, rather than entire datasets. In this paper, we study stochastic optimization algorithms for a pe…

2022

Private Synthetic Data for Multitask Learning and Marginal Queries

NeurIPS 2022accept

We provide a differentially private algorithm for producing synthetic data simultaneously useful for multiple tasks: marginal queries and multitask machine learning (ML). A key innovation in our algorithm is the ability to directly handle numerical features, in contrast to a number of related prior…

Cited by 35SourcePDFScholar
2022

Sequence Model Imitation Learning with Unobserved Contexts

NeurIPS 2022accept

We consider imitation learning problems where the learner's ability to mimic the expert increases throughout the course of an episode as more information is revealed. One example of this is when the expert has access to privileged information: while the learner might not be able to accurately reprod…

2022

Strategic Instrumental Variable Regression: Recovering Causal Relationships From Strategic Responses

ICML 2022spotlight

In settings where Machine Learning (ML) algorithms automate or inform consequential decisions about people, individual decision subjects are often incentivized to strategically modify their observable attributes to receive more favorable predictions. As a result, the distribution the assessment rule…

2022

Understanding Clipping for Federated Learning: Convergence and Client-Level Differential Privacy

ICML 2022spotlight

Providing privacy protection has been one of the primary motivations of Federated Learning (FL). Recently, there has been a line of work on incorporating the formal privacy notion of differential privacy with FL. To guarantee the client-level differential privacy in FL algorithms, the clients’ trans…

Cited by 133SourcePDFScholar
2021

Bypassing the Ambient Dimension: Private SGD with Gradient Subspace Identification

ICLR 2021poster

Differentially private SGD (DP-SGD) is one of the most popular methods for solving differentially private empirical risk minimization (ERM). Due to its noisy perturbation on each gradient update, the error rate of DP-SGD scales with the ambient dimension $p$, the number of parameters in the model. S…

Cited by 128SourcePDFScholar
2021

Gaming Helps! Learning from Strategic Interactions in Natural Dynamics

AISTATS 2021poster

We consider an online regression setting in which individuals adapt to the regression model: arriving individuals may access the model throughout the process, and invest strategically in modifying their own features so as to improve their predicted score. Such feature manipulation, or “gaming”, has…

Cited by 53SourcePDFScholar
2021

Incentivizing Compliance with Algorithmic Instruments

ICML 2021spotlight

Randomized experiments can be susceptible to selection bias due to potential non-compliance by the participants. While much of the existing work has studied compliance as a static behavior, we propose a game-theoretic model to study compliance as dynamic behavior that may change over time. In rounds…

2021

Iterative Methods for Private Synthetic Data: Unifying Framework and New Methods

NeurIPS 2021poster

We study private synthetic data generation for query release, where the goal is to construct a sanitized version of a sensitive dataset, subject to differential privacy, that approximately preserves the answers to a large collection of statistical queries. We first present an algorithmic framework t…

Cited by 76SourcePDFScholar
2021

Learn to Expect the Unexpected: Probably Approximately Correct Domain Generalization

AISTATS 2021poster

Domain generalization is the problem of machine learning when the training data and the test data come from different “domains” (data distributions). We propose an elementary theoretical model of the domain generalization problem, introducing the concept of a meta-distribution over domains. In our m…

Cited by 29SourcePDFScholar
2021

Leveraging Public Data for Practical Private Query Release

ICML 2021spotlight

In many statistical problems, incorporating priors can significantly improve performance. However, the use of prior knowledge in differentially private query release has remained underexplored, despite such priors commonly being available in the form of public datasets, such as previous US Census re…

2021

Of Moments and Matching: A Game-Theoretic Framework for Closing the Imitation Gap

ICML 2021spotlight

We provide a unifying view of a large family of previous imitation learning algorithms through the lens of moment matching. At its core, our classification scheme is based on whether the learner attempts to match (1) reward or (2) action-value moments of the expert’s behavior, with each option leadi…

2021

Towards the Unification and Robustness of Perturbation and Gradient Based Explanations

ICML 2021spotlight

As machine learning black boxes are increasingly being deployed in critical domains such as healthcare and criminal justice, there has been a growing emphasis on developing techniques for explaining these black boxes in a post hoc manner. In this work, we analyze two popular post hoc interpretation…

Cited by 82SourcePDFScholar
2020

New Oracle-Efficient Algorithms for Private Synthetic Data Release

ICML 2020poster

We present three new algorithms for constructing differentially private synthetic data—a sanitized version of a sensitive dataset that approximately preserves the answers to a large collection of statistical queries. All three algorithms are \emph{oracle-efficient} in the sense that they are computa…

2020

Private Query Release Assisted by Public Data

ICML 2020poster

We study the problem of differentially private query release assisted by access to public data. In this problem, the goal is to answer a large class $\mathcal{H}$ of statistical queries with error no more than $\alpha$ using a combination of public and private samples. The algorithm is required to s…

Cited by 68SourcePDFScholar
2020

Private Reinforcement Learning with PAC and Regret Guarantees

ICML 2020poster

Motivated by high-stakes decision-making domains like personalized medicine where user information is inherently sensitive, we design privacy preserving exploration policies for episodic reinforcement learning (RL). We first provide a meaningful privacy formulation using the notion of joint differen…

Cited by 74SourcePDFScholar
2020

Structured Linear Contextual Bandits: A Sharp and Geometric Smoothed Analysis

ICML 2020poster

Bandit learning algorithms typically involve the balance of exploration and exploitation. However, in many practical applications, worst-case scenarios needing systematic exploration are seldom encountered. In this work, we consider a smoothed setting for structured linear contextual bandits where t…

Cited by 24SourcePDFScholar