Get our free extension to see links to code for papers anywhere online!

Chrome logo Add to Chrome

Firefox logo Add to Firefox
Picture for Jason D. Lee

Jason D. Lee

Going Beyond Linear RL: Sample Efficient Neural Function Approximation


Jul 14, 2021
Baihe Huang, Kaixuan Huang, Sham M. Kakade, Jason D. Lee, Qi Lei, Runzhe Wang, Jiaqi Yang

Figure 1 for Going Beyond Linear RL: Sample Efficient Neural Function Approximation
Figure 2 for Going Beyond Linear RL: Sample Efficient Neural Function Approximation

  Access Paper or Ask Questions

Optimal Gradient-based Algorithms for Non-concave Bandit Optimization


Jul 09, 2021
Baihe Huang, Kaixuan Huang, Sham M. Kakade, Jason D. Lee, Qi Lei, Runzhe Wang, Jiaqi Yang

Figure 1 for Optimal Gradient-based Algorithms for Non-concave Bandit Optimization
Figure 2 for Optimal Gradient-based Algorithms for Non-concave Bandit Optimization

  Access Paper or Ask Questions

A Short Note on the Relationship of Information Gain and Eluder Dimension


Jul 06, 2021
Kaixuan Huang, Sham M. Kakade, Jason D. Lee, Qi Lei


  Access Paper or Ask Questions

Near-Optimal Linear Regression under Distribution Shift


Jun 23, 2021
Qi Lei, Wei Hu, Jason D. Lee

Figure 1 for Near-Optimal Linear Regression under Distribution Shift
Figure 2 for Near-Optimal Linear Regression under Distribution Shift
Figure 3 for Near-Optimal Linear Regression under Distribution Shift
Figure 4 for Near-Optimal Linear Regression under Distribution Shift
* ICML 2021 
  Access Paper or Ask Questions

Policy Mirror Descent for Regularized Reinforcement Learning: A Generalized Framework with Linear Convergence


May 24, 2021
Wenhao Zhan, Shicong Cen, Baihe Huang, Yuxin Chen, Jason D. Lee, Yuejie Chi

Figure 1 for Policy Mirror Descent for Regularized Reinforcement Learning: A Generalized Framework with Linear Convergence

  Access Paper or Ask Questions

How Fine-Tuning Allows for Effective Meta-Learning


May 05, 2021
Kurtland Chua, Qi Lei, Jason D. Lee

Figure 1 for How Fine-Tuning Allows for Effective Meta-Learning
Figure 2 for How Fine-Tuning Allows for Effective Meta-Learning

  Access Paper or Ask Questions

Bilinear Classes: A Structural Framework for Provable Generalization in RL


Mar 19, 2021
Simon S. Du, Sham M. Kakade, Jason D. Lee, Shachar Lovett, Gaurav Mahajan, Wen Sun, Ruosong Wang

Figure 1 for Bilinear Classes: A Structural Framework for Provable Generalization in RL
Figure 2 for Bilinear Classes: A Structural Framework for Provable Generalization in RL

  Access Paper or Ask Questions

MUSBO: Model-based Uncertainty Regularized and Sample Efficient Batch Optimization for Deployment Constrained Reinforcement Learning


Feb 23, 2021
DiJia Su, Jason D. Lee, John M. Mulvey, H. Vincent Poor

Figure 1 for MUSBO: Model-based Uncertainty Regularized and Sample Efficient Batch Optimization for Deployment Constrained Reinforcement Learning
Figure 2 for MUSBO: Model-based Uncertainty Regularized and Sample Efficient Batch Optimization for Deployment Constrained Reinforcement Learning
Figure 3 for MUSBO: Model-based Uncertainty Regularized and Sample Efficient Batch Optimization for Deployment Constrained Reinforcement Learning
Figure 4 for MUSBO: Model-based Uncertainty Regularized and Sample Efficient Batch Optimization for Deployment Constrained Reinforcement Learning

  Access Paper or Ask Questions

A Theory of Label Propagation for Subpopulation Shift


Feb 22, 2021
Tianle Cai, Ruiqi Gao, Jason D. Lee, Qi Lei

Figure 1 for A Theory of Label Propagation for Subpopulation Shift
Figure 2 for A Theory of Label Propagation for Subpopulation Shift
Figure 3 for A Theory of Label Propagation for Subpopulation Shift
Figure 4 for A Theory of Label Propagation for Subpopulation Shift

  Access Paper or Ask Questions

Provably Efficient Policy Gradient Methods for Two-Player Zero-Sum Markov Games


Feb 17, 2021
Yulai Zhao, Yuandong Tian, Jason D. Lee, Simon S. Du


  Access Paper or Ask Questions

Beyond Lazy Training for Over-parameterized Tensor Decomposition


Oct 22, 2020
Xiang Wang, Chenwei Wu, Jason D. Lee, Tengyu Ma, Rong Ge

Figure 1 for Beyond Lazy Training for Over-parameterized Tensor Decomposition
* NeurIPS 2020; the first two authors contribute equally 
  Access Paper or Ask Questions

Provable Benefits of Representation Learning in Linear Bandits


Oct 13, 2020
Jiaqi Yang, Wei Hu, Jason D. Lee, Simon S. Du

Figure 1 for Provable Benefits of Representation Learning in Linear Bandits
Figure 2 for Provable Benefits of Representation Learning in Linear Bandits
Figure 3 for Provable Benefits of Representation Learning in Linear Bandits
Figure 4 for Provable Benefits of Representation Learning in Linear Bandits
* 28 pages, 6 figures 
  Access Paper or Ask Questions

How Important is the Train-Validation Split in Meta-Learning?


Oct 12, 2020
Yu Bai, Minshuo Chen, Pan Zhou, Tuo Zhao, Jason D. Lee, Sham Kakade, Huan Wang, Caiming Xiong

Figure 1 for How Important is the Train-Validation Split in Meta-Learning?
Figure 2 for How Important is the Train-Validation Split in Meta-Learning?
Figure 3 for How Important is the Train-Validation Split in Meta-Learning?

  Access Paper or Ask Questions

Sanity-Checking Pruning Methods: Random Tickets can Win the Jackpot


Sep 22, 2020
Jingtong Su, Yihang Chen, Tianle Cai, Tianhao Wu, Ruiqi Gao, Liwei Wang, Jason D. Lee

Figure 1 for Sanity-Checking Pruning Methods: Random Tickets can Win the Jackpot
Figure 2 for Sanity-Checking Pruning Methods: Random Tickets can Win the Jackpot
Figure 3 for Sanity-Checking Pruning Methods: Random Tickets can Win the Jackpot
Figure 4 for Sanity-Checking Pruning Methods: Random Tickets can Win the Jackpot
* Original submission to NeurIPS 2020. Under review 
  Access Paper or Ask Questions

Generalized Leverage Score Sampling for Neural Networks


Sep 21, 2020
Jason D. Lee, Ruoqi Shen, Zhao Song, Mengdi Wang, Zheng Yu

Figure 1 for Generalized Leverage Score Sampling for Neural Networks
Figure 2 for Generalized Leverage Score Sampling for Neural Networks
Figure 3 for Generalized Leverage Score Sampling for Neural Networks

  Access Paper or Ask Questions

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


Aug 03, 2020
Jason D. Lee, Qi Lei, Nikunj Saunshi, Jiacheng Zhuo

Figure 1 for Predicting What You Already Know Helps: Provable Self-Supervised Learning
Figure 2 for Predicting What You Already Know Helps: Provable Self-Supervised Learning
Figure 3 for Predicting What You Already Know Helps: Provable Self-Supervised Learning

  Access Paper or Ask Questions

Implicit Bias in Deep Linear Classification: Initialization Scale vs Training Accuracy


Jul 13, 2020
Edward Moroshko, Suriya Gunasekar, Blake Woodworth, Jason D. Lee, Nathan Srebro, Daniel Soudry

Figure 1 for Implicit Bias in Deep Linear Classification: Initialization Scale vs Training Accuracy
Figure 2 for Implicit Bias in Deep Linear Classification: Initialization Scale vs Training Accuracy
Figure 3 for Implicit Bias in Deep Linear Classification: Initialization Scale vs Training Accuracy
Figure 4 for Implicit Bias in Deep Linear Classification: Initialization Scale vs Training Accuracy

  Access Paper or Ask Questions

Modeling from Features: a Mean-field Framework for Over-parameterized Deep Neural Networks


Jul 03, 2020
Cong Fang, Jason D. Lee, Pengkun Yang, Tong Zhang


  Access Paper or Ask Questions

Towards Understanding Hierarchical Learning: Benefits of Neural Representations


Jun 24, 2020
Minshuo Chen, Yu Bai, Jason D. Lee, Tuo Zhao, Huan Wang, Caiming Xiong, Richard Socher


  Access Paper or Ask Questions

Shape Matters: Understanding the Implicit Bias of the Noise Covariance


Jun 18, 2020
Jeff Z. HaoChen, Colin Wei, Jason D. Lee, Tengyu Ma

Figure 1 for Shape Matters: Understanding the Implicit Bias of the Noise Covariance

  Access Paper or Ask Questions

Convergence of Meta-Learning with Task-Specific Adaptation over Partial Parameters


Jun 16, 2020
Kaiyi Ji, Jason D. Lee, Yingbin Liang, H. Vincent Poor

Figure 1 for Convergence of Meta-Learning with Task-Specific Adaptation over Partial Parameters
Figure 2 for Convergence of Meta-Learning with Task-Specific Adaptation over Partial Parameters
Figure 3 for Convergence of Meta-Learning with Task-Specific Adaptation over Partial Parameters
Figure 4 for Convergence of Meta-Learning with Task-Specific Adaptation over Partial Parameters
* 28 pages, 8 figures, 2 tables 
  Access Paper or Ask Questions

Distributed Estimation for Principal Component Analysis: a Gap-free Approach


Apr 05, 2020
Xi Chen, Jason D. Lee, He Li, Yun Yang

Figure 1 for Distributed Estimation for Principal Component Analysis: a Gap-free Approach
Figure 2 for Distributed Estimation for Principal Component Analysis: a Gap-free Approach
Figure 3 for Distributed Estimation for Principal Component Analysis: a Gap-free Approach
Figure 4 for Distributed Estimation for Principal Component Analysis: a Gap-free Approach

  Access Paper or Ask Questions

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


Mar 23, 2020
Lemeng Wu, Mao Ye, Qi Lei, Jason D. Lee, Qiang Liu

Figure 1 for Steepest Descent Neural Architecture Optimization: Escaping Local Optimum with Signed Neural Splitting
Figure 2 for Steepest Descent Neural Architecture Optimization: Escaping Local Optimum with Signed Neural Splitting
Figure 3 for Steepest Descent Neural Architecture Optimization: Escaping Local Optimum with Signed Neural Splitting
Figure 4 for Steepest Descent Neural Architecture Optimization: Escaping Local Optimum with Signed Neural Splitting

  Access Paper or Ask Questions

Kernel and Rich Regimes in Overparametrized Models


Feb 24, 2020
Blake Woodworth, Suriya Gunasekar, Jason D. Lee, Edward Moroshko, Pedro Savarese, Itay Golan, Daniel Soudry, Nathan Srebro

Figure 1 for Kernel and Rich Regimes in Overparametrized Models
Figure 2 for Kernel and Rich Regimes in Overparametrized Models
Figure 3 for Kernel and Rich Regimes in Overparametrized Models
Figure 4 for Kernel and Rich Regimes in Overparametrized Models
* This updates and significantly extends a previous article (arXiv:1906.05827), Sections 6 and 7.1 are the most major additions. 30 pages. arXiv admin note: text overlap with arXiv:1906.05827 
  Access Paper or Ask Questions

Few-Shot Learning via Learning the Representation, Provably


Feb 21, 2020
Simon S. Du, Wei Hu, Sham M. Kakade, Jason D. Lee, Qi Lei


  Access Paper or Ask Questions

Agnostic Q-learning with Function Approximation in Deterministic Systems: Tight Bounds on Approximation Error and Sample Complexity


Feb 17, 2020
Simon S. Du, Jason D. Lee, Gaurav Mahajan, Ruosong Wang


  Access Paper or Ask Questions

When Does Non-Orthogonal Tensor Decomposition Have No Spurious Local Minima?


Nov 22, 2019
Maziar Sanjabi, Sina Baharlouei, Meisam Razaviyayn, Jason D. Lee


  Access Paper or Ask Questions

SGD Learns One-Layer Networks in WGANs


Oct 15, 2019
Qi Lei, Jason D. Lee, Alexandros G. Dimakis, Constantinos Daskalakis

Figure 1 for SGD Learns One-Layer Networks in WGANs
Figure 2 for SGD Learns One-Layer Networks in WGANs
* 21 pages, 4 figures 
  Access Paper or Ask Questions

Beyond Linearization: On Quadratic and Higher-Order Approximation of Wide Neural Networks


Oct 03, 2019
Yu Bai, Jason D. Lee


  Access Paper or Ask Questions

Optimality and Approximation with Policy Gradient Methods in Markov Decision Processes


Aug 29, 2019
Alekh Agarwal, Sham M. Kakade, Jason D. Lee, Gaurav Mahajan

Figure 1 for Optimality and Approximation with Policy Gradient Methods in Markov Decision Processes
Figure 2 for Optimality and Approximation with Policy Gradient Methods in Markov Decision Processes
Figure 3 for Optimality and Approximation with Policy Gradient Methods in Markov Decision Processes
Figure 4 for Optimality and Approximation with Policy Gradient Methods in Markov Decision Processes
* Additional references and discussion of prior work 
  Access Paper or Ask Questions