activity
20172022
most citedR-SPIDER: A Fast Riemannian Stochastic Optimization Algorithm with Curvature Independent Rate

26 citations · 55 across the 6 of their papers we have counts for

collaborators

18 papers

cs.LG20221 cited

Online Policy Optimization for Robust MDP

Jing Dong, Jingwei Li, Baoxiang Wang +1

Reinforcement learning (RL) has exceeded human performance in many synthetic settings such as video games and Go. However, real-world deployment of end-to-end RL models is less com…

cs.LG20212 cited

Provably Efficient Algorithms for Multi-Objective Competitive RL

Tiancheng Yu, Yi Tian, Jingzhao Zhang +1

We study multi-objective reinforcement learning (RL) where an agent's reward is represented as a vector. In settings where an agent competes against opponents, its performance is m…

cs.LG2020

Coping with Label Shift via Distributionally Robust Optimisation

Jingzhao Zhang, Aditya Menon, Andreas Veit +3

The label shift problem refers to the supervised learning setting where the train and test label distributions do not match. Existing work addressing label shift usually assumes ac…

math.OC2020

Complexity of Finding Stationary Points of Nonsmooth Nonconvex Functions

Jingzhao Zhang, Hongzhou Lin, Stefanie Jegelka +2

We provide the first non-asymptotic analysis for finding stationary points of nonsmooth, nonconvex functions. In particular, we study the class of Hadamard semi-differentiable func…

math.OC2019

Why are Adaptive Methods Good for Attention Models?

Jingzhao Zhang, Sai Praneeth Karimireddy, Andreas Veit +4

While stochastic gradient descent (SGD) is still the \emph{de facto} algorithm in deep learning, adaptive methods like Clipped SGD/Adam have been observed to outperform SGD across…

math.OC20193 cited

Acceleration in First Order Quasi-strongly Convex Optimization by ODE Discretization

Jingzhao Zhang, Suvrit Sra, Ali Jadbabaie

We study gradient-based optimization methods obtained by direct Runge-Kutta discretization of the ordinary differential equation (ODE) describing the movement of a heavy-ball under…