Saved in:
Bibliographic Details
Main Authors: Lee, Harin, Hwang, Taehyun, Oh, Min-hwan
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2406.00823
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917941430190080
author Lee, Harin
Hwang, Taehyun
Oh, Min-hwan
author_facet Lee, Harin
Hwang, Taehyun
Oh, Min-hwan
contents We consider a stochastic sparse linear bandit problem where only a sparse subset of context features affects the expected reward function, i.e., the unknown reward parameter has a sparse structure. In the existing Lasso bandit literature, the compatibility conditions, together with additional diversity conditions on the context features are imposed to achieve regret bounds that only depend logarithmically on the ambient dimension $d$. In this paper, we demonstrate that even without the additional diversity assumptions, the \textit{compatibility condition on the optimal arm} is sufficient to derive a regret bound that depends logarithmically on $d$, and our assumption is strictly weaker than those used in the lasso bandit literature under the single-parameter setting. We propose an algorithm that adapts the forced-sampling technique and prove that the proposed algorithm achieves $O(\text{poly}\log dT)$ regret under the margin condition. To our knowledge, the proposed algorithm requires the weakest assumptions among Lasso bandit algorithms under the single-parameter setting that achieve $O(\text{poly}\log dT)$ regret. Through numerical experiments, we confirm the superior performance of our proposed algorithm.
format Preprint
id arxiv_https___arxiv_org_abs_2406_00823
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Lasso Bandit with Compatibility Condition on Optimal Arm
Lee, Harin
Hwang, Taehyun
Oh, Min-hwan
Machine Learning
We consider a stochastic sparse linear bandit problem where only a sparse subset of context features affects the expected reward function, i.e., the unknown reward parameter has a sparse structure. In the existing Lasso bandit literature, the compatibility conditions, together with additional diversity conditions on the context features are imposed to achieve regret bounds that only depend logarithmically on the ambient dimension $d$. In this paper, we demonstrate that even without the additional diversity assumptions, the \textit{compatibility condition on the optimal arm} is sufficient to derive a regret bound that depends logarithmically on $d$, and our assumption is strictly weaker than those used in the lasso bandit literature under the single-parameter setting. We propose an algorithm that adapts the forced-sampling technique and prove that the proposed algorithm achieves $O(\text{poly}\log dT)$ regret under the margin condition. To our knowledge, the proposed algorithm requires the weakest assumptions among Lasso bandit algorithms under the single-parameter setting that achieve $O(\text{poly}\log dT)$ regret. Through numerical experiments, we confirm the superior performance of our proposed algorithm.
title Lasso Bandit with Compatibility Condition on Optimal Arm
topic Machine Learning
url https://arxiv.org/abs/2406.00823