English

Catching a Moving Subspace: Low-Rank Bandits Beyond Stationarity

Machine Learning 2026-05-21 v1 Artificial Intelligence Machine Learning

Abstract

Many bandit deployments (recommendation, clinical dosing, ad targeting) share two facts prior work handles only in isolation: rewards live on a low-dimensional latent subspace, and that subspace drifts. Stationary low-rank bandits exploit rank but break under subspace change; non-stationary linear bandits adapt to drift but pay ambient rate O~(dT)\widetilde{O}(d\sqrt{T}). We study piecewise-stationary low-rank linear contextual bandits with scalar feedback: θt=Bkwt\theta_t = B_k^\star w_t with rank-rr factor BkRd×rB_k^\star\in\mathbb{R}^{d\times r} constant within each of KK unknown segments and able to shift at boundaries. Our results are tight along three axes. (i) Identification boundary. With single-play scalar rewards, the moving subspace is recoverable through quadratic functionals of rewards iff three probe-side conditions hold: known noise variance, bounded state-noise coupling, and full-dimensional probe support. Each is necessary in the unrestricted-second-moment problem, and jointly they are sufficient, characterizing the boundary of the solvable region. (ii) Algorithm and dynamic regret. SPSC interleaves isotropic probes with windowed projected ridge-UCB exploitation inside the learned rr-dimensional subspace; a CUSUM-style variant discovers segment boundaries online. The costed dynamic regret is O~(rT)+O~(T2/3)+O(WVin)\widetilde{O}(r\sqrt{T})+\widetilde{O}(T^{2/3})+O(W\,V_{\mathrm{in}}), replacing the ambient dTd\sqrt{T} rate with the intrinsic rank. (iii) Empirics. On eleven benchmarks spanning synthetic, UCI/MovieLens, semi-synthetic clinical, and ZOZOTOWN production-log data, SPSC outperforms non-stationary and low-rank baselines whenever drT1/6d-r\gtrsim T^{1/6}, matching the analytical crossover. To our knowledge, this is the first work to characterize the identification boundary and attain the intrinsic-rank dynamic-regret rate in this setting.

Keywords

Cite

@article{arxiv.2605.20269,
  title  = {Catching a Moving Subspace: Low-Rank Bandits Beyond Stationarity},
  author = {Hamed Khosravi and Xiaoming Huo},
  journal= {arXiv preprint arXiv:2605.20269},
  year   = {2026}
}