Revisiting Matrix Sketching in Linear Bandits: Achieving Sublinear Regret via Dyadic Block Sketching
Dongxie Wen, Hanyan Yin, Xiao Zhang, Peng Zhao, Lijun Zhang, Zhewei Wei
Abstract
Linear bandits have become a cornerstone of online learning and sequential decision-making, providing solid theoretical foundations for balancing exploration and exploitation. Within this domain, matrix sketching serves as a critical component for achieving computational efficiency, especially when confronting high-dimensional problem instances. The sketch-based approaches reduce per-round complexity from to , where is the dimension and is the sketch size. However, this computational efficiency comes with a fundamental pitfall: when the streaming matrix exhibits heavy spectral tails, such algorithms can incur vacuous linear regret. In this paper, we revisit the regret bounds and algorithmic design for sketch-based linear bandits. Our analysis reveals that inappropriate sketch sizes can lead to substantial spectral error, severely undermining regret guarantees. To overcome this issue, we propose Dyadic Block Sketching, a novel multi-scale matrix sketching approach that dynamically adjusts the sketch size during the learning process. We apply this technique to linear bandits and demonstrate that the new algorithm achieves sublinear regret bounds without requiring prior knowledge of the streaming matrix properties. It establishes a general framework for efficient sketch-based linear bandits, which can be integrated with any matrix sketching method that provides covariance guarantees. Comprehensive experimental evaluation demonstrates the superior utility-efficiency trade-off achieved by our approach.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itBuilds on7
- Counteracting User Attention Bias in Music Streaming Recommendation via Reward ModificationXiao Zhang, Sunhao Dai, Jun Xu, Zhenhua Dong et al.KDD 2022 · 26 citations
- Sketchy: Memory-efficient Adaptive Regularization with Frequent DirectionsVladimir Feinberg, Xinyi Chen, Y. Jennifer Sun, Rohan Anil et al.NeurIPS 2023 · 21 citations
- Generalized Linear Bandits: Almost Optimal Regret with One-Pass UpdateYu-Jie Zhang, Sheng-An Xu, Peng Zhao, Masashi SugiyamaNeurIPS 2025 · 17 citations
- Reward Imputation with Sketching for Contextual Batched BanditsXiao Zhang, Ninglu Shao, Zihua Si, Jun Xu et al.NeurIPS 2023 · 6 citations
- Optimal Matrix Sketching over Sliding WindowsHanyan Yin, Dongxie Wen, Jiajun Li, Zhewei Wei et al.VLDB 2024 · 5 citations
Related papers
- Linear Bandits with Partially Observable FeaturesWonyoung Kim, Sungwoo Park, Garud Iyengar, Assaf Zeevi et al.ICML 2025 · 3 citations
- Efficient Algorithms for Generalized Linear Bandits with Heavy-tailed RewardsBo Xue, Yimu Wang, Yuanyu Wan, Jinfeng Yi et al.NeurIPS 2023 · 16 citations
- Heavy-Tailed Linear Bandits: Huber Regression with One-Pass UpdateJing Wang, Yu-Jie Zhang, Peng Zhao, Zhi-Hua ZhouICML 2025
- Thresholded Lasso BanditKaito Ariu, Kenshi Abe, Alexandre ProutièreICML 2022 · 20 citations
- Delay and Cooperation in Nonstochastic Linear BanditsShinji Ito, Daisuke Hatano, Hanna Sumita, Kei Takemura et al.NeurIPS 2020 · 27 citations
