activity
20152023
most citedHessian-based Analysis of Large Batch Training and Robustness to Adversaries

95 citations · 421 across the 33 of their papers we have counts for

collaborators
Showing 2020Show all

7 papers · 1 filter

cs.LG2020★ 1 cited

Fast Convergence of Langevin Dynamics on Manifold: Geodesics meet Log-Sobolev

Xiao Wang, Qi Lei, Ioannis Panageas

Sampling is a fundamental and arguably very important task with numerous applications in Machine Learning. One approach to sample from a high dimensional distribution for…

cs.LG2020★ 48 cited

Predicting What You Already Know Helps: Provable Self-Supervised Learning

Jason D. Lee, Qi Lei, Nikunj Saunshi +1

Self-supervised representation learning solves auxiliary prediction tasks (known as pretext tasks) without requiring labeled data to learn useful semantic representations. These pr…

cs.LG2020★ 4 cited

Steepest Descent Neural Architecture Optimization: Escaping Local Optimum with Signed Neural Splitting

Lemeng Wu, Mao Ye, Qi Lei +2

Developing efficient and principled neural architecture optimization methods is a critical challenge of modern deep learning. Recently, Liu et al.[19] proposed a splitting steepest…

cs.LG2020★ 1 cited

Solving Inverse Problems with a Flow-based Noise Model

Jay Whang, Qi Lei, Alexandros G. Dimakis

We study image inverse problems with a normalizing flow prior. Our formulation views the solution as the maximum a posteriori estimate of the image conditioned on the measurements.…

cs.LG2020★ 27 cited

CAT: Customized Adversarial Training for Improved Robustness

Minhao Cheng, Qi Lei, Pin-Yu Chen +2

Adversarial training has become one of the most effective methods for improving robustness of neural networks. However, it often suffers from poor generalization on both clean and…

cs.LG2020★ 54 cited

Few-Shot Learning via Learning the Representation, Provably

Simon S. Du, Wei Hu, Sham M. Kakade +2

This paper studies few-shot learning via representation learning, where one uses source tasks with data per task to learn a representation in order to reduce the sample c…

cs.LG2020★ 14 cited

Last iterate convergence in no-regret learning: constrained min-max optimization for convex-concave landscapes

Qi Lei, Sai Ganesh Nagarajan, Ioannis Panageas +1

In a recent series of papers it has been established that variants of Gradient Descent/Ascent and Mirror Descent exhibit last iterate convergence in convex-concave zero-sum games.…