Publications

For a full list, see Google Scholar.

What are we trying to measure?

Constructs and valid operationalizations.

ICML 2026ICML 2026

Position: AI Evaluations Should be Grounded on a Theory of Capability

with Nathan Jo. arXiv · ✓ Evaluation

with Nathan Jo

Benchmark scores are often treated as direct measurements of what an AI system can do, but interpreting performance as evidence of capability requires assumptions about how an underlying capability produces observed behavior. We argue that evaluations should make this inferential step explicit by beginning with a theory of capability and developing methods that quantify uncertainty about the capability being measured.

Benchmark scores are often treated as direct measurements of what an AI system can do, but interpreting performance as evidence of capability requires assumptions about how a latent capability produces observed behavior. We argue that evaluations should make this inferential step explicit by beginning with a theory of capability and developing methods that quantify uncertainty about the capability being measured.

NEURIPS BIALIGN WORKSHOP 2026NEURIPS BIALIGN WORKSHOP 2026
ORAL
ICLR HCAIR WORKSHOPICLR HCAIR WORKSHOP

Position: Alignment has a Fantasia Problem

with Nathanael Jo, Zoe De Simone, Mitchell Gordon. arXiv · ✓ Evaluation

with Nathanael Jo, Zoe De Simone, Mitchell Gordon

AI assistants are typically evaluated by how well they satisfy a user’s request, but users often approach complex tasks before their goals and intentions are fully formed. We argue that effective assistance should instead account for the user’s evolving cognitive process, motivating evaluations that ask not only whether a task was completed, but whether the system preserved appropriate human judgment and agency throughout the interaction.

AI assistants are typically evaluated by whether they satisfy a user’s request, but this operationalization assumes that users enter complex tasks with fixed and fully formed goals. We argue that successful assistance should instead be defined in relation to the user’s evolving cognitive process, including whether the system preserves appropriate human judgment and agency.

What does the evidence establish?

Context dependence, robustness, and the limits of inference.

AI4GOOD WORKSHOPAI4GOOD WORKSHOP
SPOTLIGHT

Evaluation without Generation: Non-Generative Assessment of Harmful Model Specialization with Applications to CSAM

with Vinith Suriyakumar, Ayush Sekhari, Lena Stempfle, Robertson Wang, Michael Simpson, Rebecca Portnoff, Marzyeh Ghassemi. arXiv · ◆ Safety · ✓ Evaluation

with Vinith Suriyakumar, Ayush Sekhari, Lena Stempfle, Robertson Wang, Michael Simpson, Rebecca Portnoff, Marzyeh Ghassemi

Some harmful model capabilities are difficult or unsafe to evaluate by generating outputs, especially when producing the content itself is legally or ethically constrained, as with child sexual abuse material (CSAM). We develop methods for detecting harmful model specialization from a model's internal representations, enabling evaluation without generation.

Some harmful model capabilities cannot be safely or lawfully evaluated by generating outputs, making the usual behavioral evidence unavailable. We develop Evaluation without Generation, the first machine-learning screening method evaluated on CSAM-specialized LoRAs found in the wild, detecting harmful model specialization without generating or inspecting CSAM.

NEURIPS 2026

When Are Semivalue-Based Decisions Identifiable? Robust Data Selection under Utility Ambiguity

with Hannah Diehl, Justin Steil · ✓ Evaluation

with Hannah Diehl, Justin Steil

Shapley values and other semivalues are often used to rank training examples for data selection, but these rankings can depend on arbitrary choices in how model utility is specified. We characterize when the resulting selection decisions are invariant to this utility ambiguity, and develop algorithms that either certify robustness or produce explicit witnesses showing that the selected data are not uniquely determined by the underlying task.

CHI 2026CHI 2026
HONORABLE MENTION

Interaction Context Often Increases Sycophancy in LLMs

with Shomik Jain, Charlotte Park, Matheus Vianna, Dana Calacci. arXiv · ◆ Safety · ✓ Evaluation

with Shomik Jain, Charlotte Park, Matheus Vianna, Dana Calacci

AI systems increasingly personalize their responses using information about users and their prior interactions, but this context can also amplify problematic forms of agreement and mirroring. We show that interaction context often increases sycophancy, highlighting how context-free evaluations can underestimate behaviors that raise concerns about user autonomy and psychological harm in real-world interactions.

Context-free evaluations treat behavior observed on isolated prompts as evidence of how a system will behave during interaction, but information about the user and prior exchanges can change the behavior being measured. We show that interaction context often increases sycophancy, demonstrating that context-free protocols can systematically understate risks to truthfulness and user autonomy.

TMLR 2026TMLR 2026

Unstable Unlearning: The Hidden Risk of Concept Resurgence in Diffusion Models

with Vinith Suriyakumar, Rohan Alur, Ayush Sekhari, Manish Raghavan. arXiv · ◆ Safety · ✓ Evaluation

with Vinith Suriyakumar, Rohan Alur, Ayush Sekhari, Manish Raghavan

Machine unlearning methods are typically evaluated immediately after a concept has been removed, implicitly treating successful forgetting as a static property of the model. We show that subsequent benign fine-tuning can cause supposedly unlearned concepts to reappear, revealing that evaluations of unlearning must test whether safety interventions remain effective under realistic sequences of model updates.

Unlearning is commonly evaluated immediately after an intervention, implicitly treating successful suppression as evidence of durable forgetting. We show that subsequent benign fine-tuning can cause unlearned concepts to reappear for diffusion models, demonstrating that evaluations must test whether safety properties persist through realistic sequences of model updates.

TMLR 2026TMLR 2026

Influence Estimation in Statistical Models Using the Fisher Information Matrix

with Omri Lev. paper · ✓ Evaluation

with Omri Lev

Influence functions estimate how perturbations to training data affect a learned model, but widely used Fisher-information-based approximations have lacked a clear account of when they are accurate. We characterize the asymptotic accuracy of Fisher-based influence estimators, compare them with Hessian-based methods, and extend the analysis to learning problems with nondifferentiable regularization.

What outcomes actually matter?

Decision-relevant metrics and downstream consequences.

AISTATS 2026AISTATS 2026

A Consequentialist Critique of Binary Classification Evaluation: Theory, Practice, and Tools

with Gerardo Flores, Abby Schiff, Alyssa Smith, Julia Fukuyama. arXiv · ✓ Evaluation

with Gerardo Flores, Abby Schiff, Alyssa Smith, Julia Fukuyama

Classification models are often compared using fixed thresholds or aggregate metrics that can be poorly matched to how their predictions will actually be used. We develop a decision-theoretic framework for choosing evaluation metrics based on the downstream consequences of prediction, showing how properties of the deployment setting determine which measures meaningfully distinguish between models.

Classification models are often compared using accuracy, AUC, or fixed thresholds, even when those metrics are poorly matched to how predictions will actually be used. We develop a decision-theoretic framework that derives evaluation metrics from downstream consequences, showing how the deployment setting determines which distinctions between models matter.

AISTATS OPTIMAL WORKSHOP 2026AISTATS OPTIMAL WORKSHOP 2026

Pandora's Regret: A Proper Scoring Rule for Evaluating Sequential Search

with Gerardo Flores, Yash Deshpande, Jannis R. Brea. arXiv · ✓ Evaluation

with Gerardo Flores, Yash Deshpande, Jannis R. Brea

Standard classification metrics evaluate predictions independently, even when predictions are used to determine the order in which alternatives are searched or tested. We develop Pandora’s Regret, a proper scoring rule derived from the downstream cost of sequential search, aligning model evaluation with the decisions that predictions ultimately support.

Standard scoring rules evaluate predictions independently, but predictions used for sequential search matter through the order and cost of the resulting search process. We develop Pandora’s Regret, a proper scoring rule derived from downstream search costs, aligning model evaluation with the decision problem that predictions support.

NEURIPS 2025NEURIPS 2025

Aligning Evaluation with Clinical Priorities: Calibration, Label Shift, and Error Costs

with Gerardo Flores, Alyssa Smith, Julia Fukuyama. arXiv · ✓ Evaluation

with Gerardo Flores, Alyssa Smith, Julia Fukuyama

Clinical prediction systems operate under changing disease prevalence and asymmetric costs of different errors, yet common metrics such as accuracy and AUC can ignore these deployment realities. We develop a decision-theoretic evaluation framework that incorporates calibration, label shift, and clinically meaningful error costs, enabling model comparisons that better reflect the conditions and priorities of real clinical use.

Accuracy and AUC can obscure calibration, label shift, and asymmetric error costs that determine clinical value. Building on the theory of proper scoring rules, we derive new bounded, cost-sensitive log scores that connect calibration to decision performance across clinically relevant ranges of class prevalence, providing principled metrics for comparing models under uncertain deployment conditions.

FACCT 2024FACCT 2024
BEST PAPER

Algorithmic Pluralism: A Structural Approach to Equal Opportunity

with Shomik Jain, Vinith Suriyakumar, Kathleen Creel. arXiv · ◆ Safety

with Shomik Jain, Vinith Suriyakumar, Kathleen Creel

What equal opportunity requires cannot always be determined by evaluating individual algorithmic decisions in isolation; it depends on how systems collectively structure access to opportunity. We develop algorithmic pluralism as a normative and evaluative framework, arguing that systems should preserve multiple viable pathways to opportunity and be evaluated by the severity and legitimacy of the bottlenecks they create.

Equal opportunity cannot always be evaluated from individual decisions in isolation because multiple systems collectively determine which pathways to opportunity remain available. We develop algorithmic pluralism as a framework for evaluating the diversity and legitimacy of these pathways, shifting attention from isolated predictions to the structural bottlenecks created by an algorithmic ecosystem.

FACCT 2025FACCT 2025

Allocation Multiplicity: Evaluating the Promises of the Rashomon Set

with Shomik Jain, Margaret Wang, Kathleen Creel. arXiv · ◆ Safety · ✓ Evaluation

with Shomik Jain, Margaret Wang, Kathleen Creel

When many models perform equally well, model multiplicity is often taken to imply greater flexibility, fairness, or diversity in downstream decisions. We show that multiplicity in predictive models need not translate into multiplicity in the allocations that ultimately matter, motivating evaluation at the level of downstream opportunities rather than model predictions alone.

Model multiplicity is often used as evidence that decision-makers have meaningful flexibility among equally accurate models, but differences between models need not produce differences in allocations. We show that evaluation must examine the opportunities people actually receive, distinguishing predictive multiplicity from allocation multiplicity.

FACCT 2026FACCT 2026

Impacts of Semivalue-Based Data Curation on Fair Machine Learning Outcomes

with Hannah Diehl, Justin Steil. paper · ✓ Evaluation · ◆ Safety

with Hannah Diehl, Justin Steil

Data-valuation methods are often evaluated through aggregate model performance, even though using those values to curate training data can redistribute errors and benefits across groups. We study how semivalue-based curation affects predictive performance and fairness, evaluating data-selection methods through the downstream outcomes they produce.

Data-valuation methods are often evaluated through aggregate model performance, even though using those values to curate training data can redistribute errors and benefits across groups. We study how semivalue-based curation affects predictive performance and fairness, evaluating data-selection methods through the downstream outcomes they produce.

How do models learn efficiently?

Geometry and dynamics for optimization and sampling.

COLT 2026COLT 2026

Accelerated Convex Optimization via Hamiltonian Dynamics with Deterministic Integration Time

with Xiuyuan Wang, Vishwak Srinivasan, Qiang Fu, Siddharth Mitra, Andre Wibisono. arXiv · ∇ Optimization

with Xiuyuan Wang, Vishwak Srinivasan, Qiang Fu, Siddharth Mitra, Andre Wibisono

Optimization and sampling are closely connected through the continuous-time dynamics that underlie their algorithms. We show that Hamiltonian dynamics, a central tool in modern sampling, can also serve as a primitive for deterministic accelerated optimization by exploiting contraction of averaged trajectories, yielding practical algorithms with optimal first-order complexity.

ICLR 2025ICLR 2025

Adaptive Backtracking Line Search

with Laurent Lessard, Joao Cavalcanti. arXiv · ∇ Optimization

with Laurent Lessard, Joao Cavalcanti

The performance of optimization algorithms often depends critically on choosing appropriate step sizes, yet standard backtracking adjusts them using a fixed rule that ignores how badly a candidate step fails. We develop an adaptive line search that uses the magnitude of this violation to choose better steps, improving practical optimization speed without sacrificing the convergence guarantees of classical methods.

ALT 2025ALT 2025

High-accuracy sampling from constrained spaces with the Metropolis-adjusted Preconditioned Langevin Algorithm

with Vishwak Srinivasan, Andre Wibisono. arXiv · ∇ Optimization

with Vishwak Srinivasan, Andre Wibisono

Many inference problems require sampling from high-dimensional distributions subject to complex constraints, where standard Euclidean sampling methods can be inefficient. We develop a geometry-aware Langevin sampler inspired by preconditioned and natural-gradient optimization, showing how optimization geometry can be transferred to sampling to obtain high-accuracy guarantees for distributions supported on constrained spaces.

COLT 2024COLT 2024

Fast Sampling from Constrained Spaces Using the Metropolis-Adjusted Mirror Langevin Algorithm

with Vishwak Srinivasan, Andre Wibisono. paper · ∇ Optimization

with Vishwak Srinivasan, Andre Wibisono

Sampling from distributions constrained to compact convex sets is challenging because standard Euclidean methods can handle boundary geometry poorly, while existing Mirror Langevin discretizations introduce asymptotic bias. We develop the Metropolis-adjusted Mirror Langevin algorithm, which combines mirror geometry with an accept-reject correction to obtain unbiased sampling and mixing guarantees with exponentially better dependence on the target accuracy.

ICML 2024ICML 2024

Mean-field Underdamped Langevin Dynamics and its Spacetime Discretization

with Qiang Fu. arXiv · ∇ Optimization

with Qiang Fu

Mean-field optimization provides a useful formulation for problems such as training mean-field neural networks, but turning continuous-time dynamics into practical algorithms requires both particle and time discretization. We develop an N-particle underdamped Langevin algorithm, establish fast mixing for the underlying dynamics, and prove global convergence of its spacetime discretization in total variation.

ICML 2023ICML 2023

Accelerated Stochastic Optimization Methods under Quasar-convexity

with Qiang Fu, Dongchu Xu. paper · ∇ Optimization

with Qiang Fu, Dongchu Xu

Many nonconvex learning problems have additional structure that makes global optimization possible even without convexity. We develop accelerated stochastic methods for smooth quasar-convex objectives, providing improved convergence guarantees and a unified analysis that also covers existing deterministic methods, with applications including learning linear dynamical systems.

How can learning remain effective under constraints?

Preserving utility under privacy and resource constraints.

PREPRINT 2026PREPRINT 2026

VERSE: Verified Self-Evolving Optimizer for Agent Harnesses

with Zekai Wang, Yingqiang Ge, Zekun Wang, Hai Wang, Yuhui Xu, Joshua Frandsen, Shancong Fu, Chandan K. Reddy. arXiv · ∇ Optimization

with Zekai Wang, Yingqiang Ge, Zekun Wang, Hai Wang, Yuhui Xu, Joshua Frandsen, Shancong Fu, Chandan K. Reddy

AI agents increasingly rely on prompts, tools, and workflows that can themselves be improved over time, but they must learn from imperfect feedback: observing a failure does not reveal which proposed change will actually help. We show that simply revising these components based on past failures can reinforce incorrect lessons. VERSE uses execution-based verification to turn uncertain feedback into reliable evidence before incorporating changes, leading to more reliable gains across several agent-optimization methods.

PREPRINT 2026PREPRINT 2026

SoftWater: Class-Aware Rate Allocation for Softmax Quantization

with Joao Cavalcanti. arXiv · ∇ Optimization

with Joao Cavalcanti

Modern language models can be expensive to store and deploy, and standard quantization methods often leave the large softmax output layer in high precision. We formulate softmax quantization as a class-aware rate-distortion problem and develop SoftWater, which allocates precision according to how different classes affect the model’s output distribution, substantially reducing memory while preserving predictive quality.

PREPRINT 2026PREPRINT 2026

The Fast Mixing Mechanism for Differential Privacy

with Omri Lev, Moshe Shenfeld, Vishwak Srinivasan, Katrina Ligett. arXiv · ∇ Optimization · ◆ Safety

with Omri Lev, Moshe Shenfeld, Vishwak Srinivasan, Katrina Ligett

Randomized sketching can make large optimization problems cheaper to solve while also providing privacy through its inherent randomness, but existing private sketches can remain computationally expensive. We develop a privacy mechanism based on fast randomized transforms that retains strong privacy and accuracy guarantees while enabling faster private linear regression, connecting computational efficiency and privacy within a common algorithmic primitive.

ICML 2026ICML 2026
SPOTLIGHT

Near-Optimal Private Linear Regression via Iterative Hessian Mixing

with Omri Lev, Vishwak Srinivasan, Moshe Shenfeld, Katrina Ligett. arXiv · ∇ Optimization · ◆ Safety

with Omri Lev, Vishwak Srinivasan, Moshe Shenfeld, Katrina Ligett

Differential privacy makes it possible to learn from sensitive data while providing formal guarantees about individual privacy, but these guarantees can come at a substantial cost to statistical accuracy. We develop Iterative Hessian Mixing, a Gaussian-sketching approach to private linear regression that achieves near-optimal accuracy and improves over state-of-the-art methods.

NEURIPS 2025NEURIPS 2025

The Gaussian Mixing Mechanism: Renyi Differential Privacy via Gaussian Sketches

with Omri Lev, Vishwak Srinivasan, Moshe Shenfeld, Katrina Ligett, Ayush Sekhari. arXiv · ◆ Safety · ∇ Optimization

with Omri Lev, Vishwak Srinivasan, Moshe Shenfeld, Katrina Ligett, Ayush Sekhari

Randomization used to make optimization more computationally efficient can also provide formal privacy guarantees. We develop a sharper privacy analysis of Gaussian sketching and show how the same randomized compression operation can simultaneously support efficient computation and differential privacy, leading to improved accuracy for private linear regression and, in some settings, reduced runtime.

NEURIPS 2022NEURIPS 2022

Algorithms that Approximate Data Removal: New Results and Limitations

with Vinith Suriyakumar. paper · ∇ Optimization

with Vinith Suriyakumar

Machine unlearning aims to remove the effect of deleted training examples without retraining a model from scratch, but existing approximate methods can be computationally costly and often assume smooth learning objectives. We develop an efficient online unlearning method based on the infinitesimal jackknife that extends to nonsmooth regularizers, and show how standard model-selection procedures such as cross-validation can invalidate existing approximate unlearning guarantees.

How can harmful model behavior be detected and controlled?

Non-generative auditing, model steering, and durable unlearning.

SPOTLIGHT

Evaluation without Generation: Non-Generative Assessment of Harmful Model Specialization with Applications to CSAM

with Vinith Suriyakumar, Ayush Sekhari, Lena Stempfle, Robertson Wang, Michael Simpson, Rebecca Portnoff, Marzyeh Ghassemi

Some harmful model capabilities cannot be safely or lawfully evaluated by generating outputs, making the usual behavioral evidence unavailable. We develop Evaluation without Generation, the first machine-learning screening method evaluated on CSAM-specialized LoRAs found in the wild, detecting harmful model specialization without generating or inspecting CSAM.

Efficient and Accurate Steering of Large Language Models through Attention-Guided Feature Learning

with Parmida Davarmanesh, Adityanarayanan Radhakrishnan

Activation steering offers an efficient way to control LLM behavior without retraining, but existing methods are brittle because concept-related features can vary across tokens, activations, and layers. We develop an attention-guided framework that automatically identifies the features and layers most relevant for steering, nearly doubling the number of successfully steered concepts across model architectures and sizes up to 70 billion parameters.

Unstable Unlearning: The Hidden Risk of Concept Resurgence in Diffusion Models

with Vinith Suriyakumar, Rohan Alur, Ayush Sekhari, Manish Raghavan

Machine unlearning methods are typically evaluated immediately after a concept has been removed, implicitly treating successful forgetting as a static property of the model. We show that subsequent benign fine-tuning can cause supposedly unlearned concepts to reappear, revealing that evaluations of unlearning must test whether safety interventions remain effective under realistic sequences of model updates.

UCD: Unlearning in LLMs via Contrastive Decoding

with Vinith M. Suriyakumar, Ayush Sekhari

Removing sensitive or undesirable information from large language models can require expensive retraining and can degrade performance on unrelated tasks. We develop UCD, an inference-time unlearning method that uses contrastive decoding between smaller auxiliary models to suppress targeted information while preserving the original model’s utility, improving both forgetting and retained performance on standard unlearning benchmarks.

How do AI systems affect human agency and collective knowledge?

Sycophancy, co-creation, and epistemic diversity.

Position: Alignment has a Fantasia Problem

with Nathanael Jo, Zoe De Simone, Mitchell Gordon

AI assistants are typically evaluated by how well they satisfy a user’s request, but users often approach complex tasks before their goals and intentions are fully formed. We argue that effective assistance should instead account for the user’s evolving cognitive process, motivating evaluations that ask not only whether a task was completed, but whether the system preserved appropriate human judgment and agency throughout the interaction.

Creo: From One-Shot Image Generation to Progressive, Co-Creative Ideation

with Zoe De Simone, Angie Boggust, Fredo Durand, Arvind Satyanarayan

One-shot image generation can make visual decisions before a user’s intentions are fully formed, prematurely anchoring the creative process and reducing the user’s control over subsequent edits. We develop Creo, a multi-stage system that progresses from editable sketches to high-fidelity images while preserving prior decisions, enabling greater exploration, stronger user ownership, and less homogeneous outputs than conventional one-shot generation.

HONORABLE MENTION

Interaction Context Often Increases Sycophancy in LLMs

with Shomik Jain, Charlotte Park, Matheus Vianna, Dana Calacci

AI systems increasingly personalize their responses using information about users and their prior interactions, but this context can also amplify problematic forms of agreement and mirroring. We show that interaction context often increases sycophancy, highlighting how context-free evaluations can underestimate behaviors that raise concerns about user autonomy and psychological harm in real-world interactions.

Task-Dependent Evaluation of LLM Output Homogenization: A Taxonomy-Guided Framework

with Shomik Jain, Jack Lanchantin, Maximilian Nickel, Karen Ullrich, Candace Ross, Jamelle Watson-Daniels

Whether homogeneous language-model outputs are harmful depends on the task: creative writing may benefit from variation in narrative content, while mathematical reasoning may require agreement on a verifiable answer. We develop and validate a taxonomy of task-dependent functional diversity, then introduce a sampling method that increases diversity where it is valuable without imposing a task-agnostic tradeoff between diversity and quality.

How do algorithms reshape access to opportunity?

Pluralism, competition, and consequential allocation.

Impacts of Semivalue-Based Data Curation on Fair Machine Learning Outcomes

with Hannah Diehl, Justin Steil

Data-valuation methods are often evaluated through aggregate model performance, even though using those values to curate training data can redistribute errors and benefits across groups. We study how semivalue-based curation affects predictive performance and fairness, evaluating data-selection methods through the downstream outcomes they produce.

Homogeneous Algorithms Can Reduce Competition in Personalized Pricing

with Nathan Jo, Kathleen Creel, Manish Raghavan

Firms increasingly rely on similar data, objectives, and pretrained models, causing their pricing algorithms to produce correlated predictions even without direct coordination. We model this phenomenon in personalized pricing and show that greater algorithmic correlation reduces consumer welfare and can incentivize firms to sacrifice predictive accuracy for implicit coordination as consumers become more price sensitive.

Allocation Multiplicity: Evaluating the Promises of the Rashomon Set

with Shomik Jain, Margaret Wang, Kathleen Creel

When many models perform equally well, model multiplicity is often taken to imply greater flexibility, fairness, or diversity in downstream decisions. We show that multiplicity in predictive models need not translate into multiplicity in the allocations that ultimately matter, motivating evaluation at the level of downstream opportunities rather than model predictions alone.

BEST PAPER

Algorithmic Pluralism: A Structural Approach to Equal Opportunity

with Shomik Jain, Vinith Suriyakumar, Kathleen Creel

What equal opportunity requires cannot always be determined by evaluating individual algorithmic decisions in isolation; it depends on how systems collectively structure access to opportunity. We develop algorithmic pluralism as a normative and evaluative framework, arguing that systems should preserve multiple viable pathways to opportunity and be evaluated by the severity and legitimacy of the bottlenecks they create.

Scarce Resource Allocations That Rely on Machine Learning Should Be Randomized

with Shomik Jain, Kathleen Creel

Machine-learning systems used to allocate scarce resources often convert uncertain predictions into deterministic decisions, denying individuals with legitimate competing claims any meaningful chance of receiving an opportunity. We argue that fair allocation can require randomization and develop stochastic procedures that better account for the claims individuals have to scarce social goods and opportunities.

PREPRINT 2026

Creo: From One-Shot Image Generation to Progressive, Co-Creative Ideation

with Zoe De Simone, Angie Boggust, Fredo Durand, Arvind Satyanarayan. arXiv · ◆ Safety

One-shot image generation can make visual decisions before a user’s intentions are fully formed, prematurely anchoring the creative process and reducing the user’s control over subsequent edits. We develop Creo, a multi-stage system that progresses from editable sketches to high-fidelity images while preserving prior decisions, enabling greater exploration, stronger user ownership, and less homogeneous outputs than conventional one-shot generation.

PNAS 2026

Efficient and accurate steering of Large Language Models through attention-guided feature learning

with Parmida Davarmanesh, Adityanarayanan Radhakrishnan. arXiv · ◆ Safety

Activation steering offers an efficient way to control LLM behavior without retraining, but existing methods are brittle because concept-related features can vary across tokens, activations, and layers. We develop an attention-guided framework that automatically identifies the features and layers most relevant for steering, nearly doubling the number of successfully steered concepts across model architectures and sizes up to 70 billion parameters.

NEURIPS 2025

Homogeneous Algorithms Can Reduce Competition in Personalized Pricing

with Nathan Jo, Kathleen Creel, Manish Raghavan. arXiv · ◆ Safety

Firms increasingly rely on similar data, objectives, and pretrained models, causing their pricing algorithms to produce correlated predictions even without direct coordination. We model this phenomenon in personalized pricing and show that greater algorithmic correlation reduces consumer welfare and can incentivize firms to sacrifice predictive accuracy for implicit coordination as consumers become more price sensitive.

ICML 2024

Scarce Resource Allocations That Rely On Machine Learning Should Be Randomized

with Shomik Jain, Kathleen Creel. arXiv · ◆ Safety

Machine-learning systems used to allocate scarce resources often convert uncertain predictions into deterministic decisions, denying individuals with legitimate competing claims any meaningful chance of receiving an opportunity. We argue that fair allocation can require randomization and develop stochastic procedures that better account for the claims individuals have to scarce social goods and opportunities.

PREPRINT 2025

Task-Dependent Evaluation of LLM Output Homogenization: A Taxonomy-Guided Framework

with Shomik Jain, Jack Lanchantin, Maximilian Nickel, Karen Ullrich, Candace Ross, Jamelle Watson-Daniels. arXiv · ◆ Safety

Whether homogeneous language-model outputs are harmful depends on the task: creative writing may benefit from variation in narrative content, while mathematical reasoning may require agreement on a verifiable answer. We develop and validate a taxonomy of task-dependent functional diversity, then introduce a sampling method that increases diversity where it is valuable without imposing a task-agnostic tradeoff between diversity and quality.

PREPRINT 2025

UCD: Unlearning in LLMs via Contrastive Decoding

with Vinith M. Suriyakumar, Ayush Sekhari. arXiv · ◆ Safety

Removing sensitive or undesirable information from large language models can require expensive retraining and can degrade performance on unrelated tasks. We develop UCD, an inference-time unlearning method that uses contrastive decoding between smaller auxiliary models to suppress targeted information while preserving the original model’s utility, improving both forgetting and retained performance on standard unlearning benchmarks.