← Search

Stephen Wright

20 accepted papers

2024

Convex and Bilevel Optimization for Neural-Symbolic Inference and Learning

ICML 2024poster

We leverage convex and bilevel optimization techniques to develop a general gradient-based parameter learning framework for neural-symbolic (NeSy) systems. We demonstrate our framework with NeuPSL, a state-of-the-art NeSy architecture. To achieve this, we propose a smooth primal and dual formulation…

Cited by 6SourcePDFScholar
2024

How to Make the Gradients Small Privately: Improved Rates for Differentially Private Non-Convex Optimization

ICML 2024poster

We provide a simple and flexible framework for designing differentially private algorithms to find approximate stationary points of non-convex loss functions. Our framework is based on using a private approximate risk minimizer to "warm start" another private algorithm for finding stationary points.…

2024

On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximation

ICLR 2024spotlight

In this work, we study first-order algorithms for solving Bilevel Optimization (BO) where the objective functions are smooth but possibly nonconvex in both levels and the variables are restricted to closed convex sets. As a first step, we study the landscape of BO through the lens of penalty methods…

Cited by 27SourcePDFScholar
2024

On the Complexity of Teaching a Family of Linear Behavior Cloning Learners

NeurIPS 2024poster

We study optimal teaching for a family of Behavior Cloning learners that learn using a linear hypothesis class. In this setup, a knowledgeable teacher can demonstrate a dataset of state and action tuples and is required to teach an optimal policy to an entire family of BC learners using the smallest…

Cited by 0SourcePDFScholar
2024

Private Heterogeneous Federated Learning Without a Trusted Server Revisited: Error-Optimal and Communication-Efficient Algorithms for Convex Losses

ICML 2024poster

We revisit the problem of federated learning (FL) with private data from people who do not trust the server or other silos/clients. In this context, every silo (e.g. hospital) has data from several people (e.g. patients) and needs to protect the privacy of each person's data (e.g. health records), e…

Cited by 6SourcePDFScholar
2024

Revisiting Inexact Fixed-Point Iterations for Min-Max Problems: Stochasticity and Structured Nonconvexity

ICML 2024poster

We focus on constrained, $L$-smooth, potentially stochastic and nonconvex-nonconcave min-max problems either satisfying $\rho$-cohypomonotonicity or admitting a solution to the $\rho$-weakly Minty Variational Inequality (MVI), where larger values of the parameter $\rho>0$ correspond to a greater deg…

Cited by 2SourcePDFScholar
2023

A Fully First-Order Method for Stochastic Bilevel Optimization

ICML 2023oral

We consider stochastic unconstrained bilevel optimization problems when only the first-order gradient oracles are available. While numerous optimization methods have been proposed for tackling bilevel problems, existing methods either tend to require possibly expensive calculations regarding Hessian…

Cited by 82SourcePDFScholar
2023

Cyclic Block Coordinate Descent With Variance Reduction for Composite Nonconvex Optimization

ICML 2023poster

Nonconvex optimization is central in solving many machine learning problems, in which block-wise structure is commonly encountered. In this work, we propose cyclic block coordinate methods for nonconvex optimization problems with non-asymptotic gradient norm guarantees. Our convergence analysis is b…

Cited by 21SourcePDFScholar
2023

Robust Second-Order Nonconvex Optimization and Its Application to Low Rank Matrix Sensing

NeurIPS 2023poster

Finding an approximate second-order stationary point (SOSP) is a well-studied and fundamental problem in stochastic nonconvex optimization with many applications in machine learning. However, this problem is poorly understood in the presence of outliers, limiting the use of existing nonconvex algor…

Cited by 3SourcePDFScholar
2022

BOME! Bilevel Optimization Made Easy: A Simple First-Order Approach

NeurIPS 2022accept

Bilevel optimization (BO) is useful for solving a variety of important machine learning problems including but not limited to hyperparameter optimization, meta-learning, continual learning, and reinforcement learning. Conventional BO methods need to differentiate through the low-level optimization p…

Cited by 94SourcePDFScholar
2022

Coordinate Linear Variance Reduction for Generalized Linear Programming

NeurIPS 2022accept

We study a class of generalized linear programs (GLP) in a large-scale setting, which includes simple, possibly nonsmooth convex regularizer and simple convex set constraints. By reformulating (GLP) as an equivalent convex-concave min-max problem, we show that the linear structure in the problem can…

2018

ATOMO: Communication-efficient Learning via Atomic Sparsification

NeurIPS 2018poster

Distributed model training suffers from communication overheads due to frequent gradient updates transmitted between compute nodes. To mitigate these overheads, several studies propose the use of sparsified stochastic gradients. We argue that these are facets of a general sparsification method that…

2018

Dissipativity Theory for Accelerating Stochastic Variance Reduction: A Unified Analysis of SVRG and Katyusha Using Semidefinite Programs

ICML 2018oral

Techniques for reducing the variance of gradient estimates used in stochastic programming algorithms for convex finite-sum problems have received a great deal of attention in recent years. By leveraging dissipativity theory from control, we provide a new perspective on two important variance-reducti…

Cited by 26SourcePDFScholar
2017

Improved Strongly Adaptive Online Learning using Coin Betting

AISTATS 2017poster

This paper describes a new parameter-free online learning algorithm for changing environments. In comparing against algorithms with the same time complexity as ours, we obtain a strongly adaptive regret bound that is a factor of at least $\sqrt\log(T)$ better, where $T$ is the time horizon. Empiri…

Cited by 85SourcePDFScholar
2017

k-Support and Ordered Weighted Sparsity for Overlapping Groups: Hardness and Algorithms

NeurIPS 2017poster

The k-support and OWL norms generalize the l1 norm, providing better prediction accuracy and better handling of correlated variables. We study the norms obtained from extending the k-support norm and OWL norms to the setting in which there are overlapping groups. The resulting norms are in general N…

Cited by 6SourcePDFScholar