← Search

Xiaohan Wei

10 accepted papers

2023

Gradient-Variation Bound for Online Convex Optimization with Constraints

AAAI 2023technical

We study online convex optimization with constraints consisting of multiple functional constraints and a relatively simple constraint set, such as a Euclidean ball. As enforcing the constraints at each time step through projections is computationally challenging in general, we allow decisions to vio…

Cited by 1SourcePDFScholar
2022

Frequency-aware SGD for Efficient Embedding Learning with Provable Benefits

ICLR 2022poster

Embedding learning has found widespread applications in recommendation systems and natural language modeling, among other domains. To learn quality embeddings efficiently, adaptive learning rate algorithms have demonstrated superior empirical performance over SGD, largely accredited to their token-d…

Cited by 5SourcePDFScholar
2021

Provably Efficient Fictitious Play Policy Optimization for Zero-Sum Markov Games with Structured Transitions

ICML 2021oral

While single-agent policy optimization in a fixed environment has attracted a lot of research attention recently in the reinforcement learning community, much less is known theoretically when there are multiple agents playing in a potentially competitive environment. We take steps forward by proposi…

Cited by 15SourcePDFScholar
2021

Provably Efficient Safe Exploration via Primal-Dual Policy Optimization

AISTATS 2021poster

We study the safe reinforcement learning problem using the constrained Markov decision processes in which an agent aims to maximize the expected total reward subject to a safety constraint on the expected total value of a utility function. We focus on an episodic setting with the function approximat…

Cited by 200SourcePDFScholar
2020

Robust One-Bit Recovery via ReLU Generative Networks: Near-Optimal Statistical Rate and Global Landscape Analysis

ICML 2020poster

We study the robust one-bit compressed sensing problem whose goal is to design an algorithm that faithfully recovers any sparse target vector $\theta_0\in\mathbb{R}^d$ \emph{uniformly} via $m$ quantized noisy measurements. Specifically, we consider a new framework for this problem where the sparsity…

Cited by 22SourcePDFScholar
2020

Upper Confidence Primal-Dual Reinforcement Learning for CMDP with Adversarial Loss

NeurIPS 2020poster

We consider online learning for episodic stochastically constrained Markov decision processes (CMDP), which plays a central role in ensuring the safety of reinforcement learning. Here the loss function can vary arbitrarily across the episodes, whereas both the loss received and the budget consumptio…

Cited by 61SourcePDFScholar
2019

On the statistical rate of nonlinear recovery in generative models with heavy-tailed data

ICML 2019oral

We consider estimating a high-dimensional vector from non-linear measurements where the unknown vector is represented by a generative model $G:\mathbb{R}^k\rightarrow\mathbb{R}^d$ with $k\ll d$. Such a model poses structural priors on the unknown vector without having a dedicated basis, and in parti…

Cited by 31SourcePDFScholar
2018

Solving Non-smooth Constrained Programs with Lower Complexity than $\mathcal{O}(1/\varepsilon)$: A Primal-Dual Homotopy Smoothing Approach

NeurIPS 2018poster

We propose a new primal-dual homotopy smoothing algorithm for a linearly constrained convex program, where neither the primal nor the dual function has to be smooth or strongly convex. The best known iteration complexity solving such a non-smooth problem is $\mathcal{O}(\varepsilon^{-1})$. In this p…

Cited by 0SourcePDFScholar