English

Linear Bandits with Partially Observable Features

Machine Learning 2025-08-19 v3 Machine Learning

Abstract

We study the linear bandit problem that accounts for partially observable features. Without proper handling, unobserved features can lead to linear regret in the decision horizon TT, as their influence on rewards is unknown. To tackle this challenge, we propose a novel theoretical framework and an algorithm with sublinear regret guarantees. The core of our algorithm consists of (i) feature augmentation, by appending basis vectors that are orthogonal to the row space of the observed features; and (ii) the introduction of a doubly robust estimator. Our approach achieves a regret bound of O~((d+dh)T)\tilde{O}(\sqrt{(d + d_h)T}), where dd is the dimension of the observed features and dhd_h depends on the extent to which the unobserved feature space is contained in the observed one, thereby capturing the intrinsic difficulty of the problem. Notably, our algorithm requires no prior knowledge of the unobserved feature space, which may expand as more features become hidden. Numerical experiments confirm that our algorithm outperforms both non-contextual multi-armed bandits and linear bandit algorithms depending solely on observed features.

Keywords

Cite

@article{arxiv.2502.06142,
  title  = {Linear Bandits with Partially Observable Features},
  author = {Wonyoung Kim and Sungwoo Park and Garud Iyengar and Assaf Zeevi and Min-hwan Oh},
  journal= {arXiv preprint arXiv:2502.06142},
  year   = {2025}
}

Comments

Accepted in ICML 2025

R2 v1 2026-06-28T21:38:05.345Z