activity
20182022
most citedCAT: Customized Adversarial Training for Improved Robustness

27 citations · 41 across the 10 of their papers we have counts for

collaborators
Showing cs.LGShow all

17 papers · 1 filter

cs.LG2022

Nearly Minimax Algorithms for Linear Bandits with Shared Representation

Jiaqi Yang, Qi Lei, Jason D. Lee +1

We give novel algorithms for multi-task and lifelong linear bandits with shared representation. Specifically, we consider the setting where we play linear bandits with dimensio…

cs.LG2021

Provable Hierarchy-Based Meta-Reinforcement Learning

Kurtland Chua, Qi Lei, Jason D. Lee

Hierarchical reinforcement learning (HRL) has seen widespread interest as an approach to tractable learning of complex modular behaviors. However, existing work either assume acces…

cs.LG20212 cited

Optimal Gradient-based Algorithms for Non-concave Bandit Optimization

Baihe Huang, Kaixuan Huang, Sham M. Kakade +4

Bandit problems with linear or concave reward have been extensively studied, but relatively few works have studied bandits with non-concave reward. This work considers a large fami…

cs.LG20212 cited

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

Kaixuan Huang, Sham M. Kakade, Jason D. Lee +1

Eluder dimension and information gain are two widely used methods of complexity measures in bandit and reinforcement learning. Eluder dimension was originally proposed as a general…

cs.LG20214 cited

Near-Optimal Linear Regression under Distribution Shift

Qi Lei, Wei Hu, Jason D. Lee

Transfer learning is essential when sufficient data comes from the source domain, with scarce labeled data from the target domain. We develop estimators that achieve minimax linear…

cs.LG2021

How Fine-Tuning Allows for Effective Meta-Learning

Kurtland Chua, Qi Lei, Jason D. Lee

Representation learning has been widely studied in the context of meta-learning, enabling rapid learning of new tasks through shared representations. Recent works such as MAML have…