cs.LGOct 5, 2026

Bellman-Centric Learning: Near-Optimal Regret for Linear Bandits with Memory

Authors: Jingyuan Liu, Huiwen Jia

Organizations: Department of Industrial Engineering and Operations Research, University of California, Berkeley

Abstract

We study linear bandits with memory, where past actions induce endogenous nonstationarity through an arbitrary known, bounded matrix-valued memory map. To trade off exploration and exploitation while accounting for the memory dynamics, we develop RSM-LinUCB, a Bellman-centric algorithm that learns as in linear bandits and plans as in reinforcement learning. This design admits a novel regret decomposition which separates the memory-induced error from the cumulative reward estimation error along the learner's trajectory. We prove a high-probability regret bound of O~(dRS(M+1)+σdT)\widetilde O\big(dRS(M+1)+σd\sqrt T\big), where TT is the learning horizon, dd is the parameter dimension, MM is the memory length, RR and SS bound the memory-map operator norm and reward-parameter norm, respectively, and σσ is the sub-Gaussian noise scale. Our results reveal that the multiplicative memory-horizon coupling in prior bounds is not intrinsic: memory only contributes an additive cost, up to logarithmic factors. We also prove a matching minimax lower bound, establishing near-optimality. We further extend the algorithm to generalized linear rewards, preserving this separation with near-optimal memory and leading statistical dependence. Our algorithms outperform the baselines in numerical experiments on synthetic instances and semi-synthetic KV- and semantic-cache tasks.

Figures & tables

Appendix figures & tables1 asset

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Block Optimism for Nonstationary Bandits with Latent Linear Dynamics

    Oct 1, 2026Taehyun Hwang, Hyunjun Choi, Heesang Ann +1PessimismLatent Dynamics

  2. Stochastic Linear Contextual Bandits with Bounded Noise: A Set-Membership Approach

    Jun 18, 2026Haonan Xu, Yingying LiContextual Bandit FrameworkOptimal Regret

  3. Online Learning with Recency: Algorithms for Sliding-window Streaming Multi-armed Bandits

    Jun 8, 2026Vladimir Braverman, Chen Wang, Liudeng Wang +1Stochastic Multi-Armed BanditsBidirectional Long Short-Term Memory