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), where T is the learning horizon, d is the parameter dimension, M is the memory length, R and S 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
Comparisons for LBM
Algorithm
Regret
Jaksch et al. (2010)
UCRL2
O(RSMnM+1/2T+σn(M+1)/2T)
Clerici et al. (2023)
OFUL-memory
O(R(S+σ)dMT3/4)
Ann et al. (2026)
OFUL-memory (improved)
O(RSdMT+σmax{1,S}dmax{d,R2}T)
Ours (Upper Bound)
RSM-LinUCB Theorem 4.3
O(dRSM+σdT)
Ours (Lower Bound)
Theorem 4.5
Ω(dRSM+σdT)
Comparisons for GLBM
Algorithm
Regret
Table 1 : Comparison of regret bounds for the basic LBM setting and the extended GLBM setting.
Figure 1 : Cumulative pseudo-regret for synthetic (A–D), KV-cache (E–F), and semantic-cache (G–H). Shading denotes one standard deviation and some baseline curves exceed the displayed range.
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
Figure 2 : Cumulative pseudo-regret for additional synthetic experiments with additive memory. Shading denotes one sample standard deviation. Some baseline curves exceed the displayed range.