← Search

Chicheng Zhang

27 accepted papers

2024

Beyond task diversity: provable representation transfer for sequential multitask linear bandits

NeurIPS 2024poster

We study lifelong learning in linear bandits, where a learner interacts with a sequence of linear bandit tasks whose parameters lie in an $m$-dimensional subspace of $\mathbb{R}^d$, thereby sharing a low-rank representation. Current literature typically assumes that the tasks are diverse, i.e., thei…

2024

Efficient Active Learning Halfspaces with Tsybakov Noise: A Non-convex Optimization Approach

AISTATS 2024poster

We study the problem of computationally and label efficient PAC active learning $d$-dimensional halfspaces with Tsybakov Noise (Tsybakov, 2004) under structured unlabeled data distributions. Inspired by Diakonikolas et al., (2020c), we prove that any approximate first-order stationary point of a smo…

Cited by 1SourcePDFScholar
2024

Efficient Low-Rank Matrix Estimation, Experimental Design, and Arm-Set-Dependent Low-Rank Bandits

ICML 2024poster

We study low-rank matrix trace regression and the related problem of low-rank matrix bandits. Assuming access to the distribution of the covariates, we propose a novel low-rank matrix estimation method called *LowPopArt* and provide its recovery guarantee that depends on a novel quantity denoted by…

2023

Kullback-Leibler Maillard Sampling for Multi-armed Bandits with Bounded Rewards

NeurIPS 2023poster

We study $K$-armed bandit problems where the reward distributions of the arms are all supported on the $[0,1]$ interval. Maillard sampling\cite{maillard13apprentissage}, an attractive alternative to Thompson sampling, has recently been shown to achieve competitive regret guarantees in the sub-Gaussi…

2022

PopArt: Efficient Sparse Regression and Experimental Design for Optimal Sparse Linear Bandits

NeurIPS 2022accept

In sparse linear bandits, a learning agent sequentially selects an action from a fixed action set and receives reward feedback, and the reward function depends linearly on a few coordinates of the covariates of the actions. This has applications in many real-world sequential decision making problems…

2022

Thompson Sampling for Robust Transfer in Multi-Task Bandits

ICML 2022spotlight

We study the problem of online multi-task learning where the tasks are performed within similar but not necessarily identical multi-armed bandit environments. In particular, we study how a learner can improve its overall performance across multiple related tasks through robust transfer of knowledge.…

2021

Multitask Bandit Learning Through Heterogeneous Feedback Aggregation

AISTATS 2021poster

In many real-world applications, multiple agents seek to learn how to perform highly related yet slightly different tasks in an online bandit learning protocol. We formulate this problem as the $\epsilon$-multi-player multi-armed bandit problem, in which a set of players concurrently interact with a…

2020

Deep Batch Active Learning by Diverse, Uncertain Gradient Lower Bounds

ICLR 2020talk

We design a new algorithm for batch active learning with deep neural network models. Our algorithm, Batch Active learning by Diverse Gradient Embeddings (BADGE), samples groups of points that are disparate and high-magnitude when represented in a hallucinated gradient space, a strategy designed to i…

Cited by 957SourcecodeScholar
2020

Efficient Contextual Bandits with Continuous Actions

NeurIPS 2020poster

We create a computationally tractable learning algorithm for contextual bandits with continuous actions having unknown structure. The new reduction-style algorithm composes with most supervised learning representations. We prove that this algorithm works in a general sense and verify the new funct…

Cited by 43SourcePDFScholar
2020

Efficient active learning of sparse halfspaces with arbitrary bounded noise

NeurIPS 2020oral

We study active learning of homogeneous $s$-sparse halfspaces in $\mathbb{R}^d$ under the setting where the unlabeled data distribution is isotropic log-concave and each label is flipped with probability at most $\eta$ for a parameter $\eta \in \big[0, \frac12\big)$, known as the bounded noise. Even…

Cited by 45SourcePDFScholar
2019

Bandit Multiclass Linear Classification: Efficient Algorithms for the Separable Case

ICML 2019oral

We study the problem of efficient online multiclass linear classification with bandit feedback, where all examples belong to one of $K$ classes and lie in the $d$-dimensional Euclidean space. Previous works have left open the challenge of designing efficient algorithms with finite mistake bounds whe…

Cited by 18SourcePDFScholar
2019

Warm-starting Contextual Bandits: Robustly Combining Supervised and Bandit Feedback

ICML 2019oral

We investigate the feasibility of learning from both fully-labeled supervised data and contextual bandit data. We specifically consider settings in which the underlying learning signal may be different between these two data sources. Theoretically, we state and prove no-regret algorithms for learnin…

2017

Efficient Online Bandit Multiclass Learning with $\tilde{O}(\sqrt{T})$ Regret

ICML 2017poster

We present an efficient second-order algorithm with $\tilde{O}(1/\eta \sqrt{T})$ regret for the bandit online multiclass problem. The regret bound holds simultaneously with respect to a family of loss functions parameterized by $\eta$, ranging from hinge loss ($\eta=0$) to squared hinge loss ($\eta=…

Cited by 0SourcePDFScholar
2015

Spectral Learning of Large Structured HMMs for Comparative Epigenomics

NeurIPS 2015poster

We develop a latent variable model and an efficient spectral algorithm motivated by the recent emergence of very large data sets of chromatin marks from multiple human cell types. A natural model for chromatin data in one cell type is a Hidden Markov Model (HMM); we model the relationship between mu…

Cited by 4SourcePDFScholar