Contextual Bandits for Unbounded Context Distributions
Puning Zhao, Rongfei Fan, Shaowei Wang, Li Shen, Qixin Zhang, Zong Ke, Tianhang Zheng
Abstract
Nonparametric contextual bandit is an important model of sequential decision making problems. Under α-Tsybakov margin condition, existing research has established a regret bound of Õ T 1-α+1 d+2 for bounded supports. However, the optimal regret with unbounded contexts has not been analyzed. The challenge of solving contextual bandit problems with unbounded support is to achieve both exploration-exploitation tradeoff and bias-variance tradeoff simultaneously. In this paper, we solve the nonparametric contextual bandit problem with unbounded contexts. We propose two nearest neighbor methods combined with UCB exploration. The first method uses a fixed k. Our analysis shows that this method achieves minimax optimal regret under a weak margin condition and relatively light-tailed context distributions. The second method uses adaptive k. By a proper data-driven selection of k, this method achieves an expected regret of Õ T 1-(α+1)β α+(d+2)β + T 1-β , in which β is a parameter describing the tail strength. This bound matches the minimax lower bound up to logarithm factors, indicating that the second method is approximately optimal.
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 papers2
- Multinoulli Extension: A Lossless Yet Effective Probabilistic Framework for Subset Selection over Partition ConstraintsQixin Zhang, Wei Huang, Can Jin, Puning Zhao et al.ICML 2025
- Taming the Long Tail: Efficient Item-wise Sharpness-Aware Minimization for LLM-based Recommender SystemsJiaming Zhang, Yuyuan Li, Xiaohua Feng, Li Zhang et al.WWW 2026
Builds on4
- Stability and Convergence of Stochastic Gradient Clipping: Beyond Lipschitz Continuity and SmoothnessVien V. Mai, Mikael JohanssonICML 2021 · 53 citations
- Efficient Classification with Adaptive KNNPuning Zhao, Lifeng LaiAAAI 2021 · 13 citations
- Robust Nonparametric Regression under Poisoning AttackPuning Zhao, Zhiguo WanAAAI 2024 · 13 citations
- Tracking Most Significant Shifts in Nonparametric Contextual BanditsJoe Suk, Samory KpotufeNeurIPS 2023 · 10 citations
Related papers
- An Asymptotically Optimal Primal-Dual Incremental Algorithm for Contextual Linear BanditsAndrea Tirinzoni, Matteo Pirotta, Marcello Restelli, Alessandro LazaricNeurIPS 2020 · 37 citations
- Syndicated Bandits: A Framework for Auto Tuning Hyper-parameters in Contextual Bandit AlgorithmsQin Ding, Yue Kang, Yi-Wei Liu, Thomas Chun Man Lee et al.NeurIPS 2022 · 12 citations
- On the Minimax Regret for Contextual Linear Bandits and Multi-Armed Bandits with Expert AdviceShinji ItoNeurIPS 2024 · 3 citations
- Universal and data-adaptive algorithms for model selection in linear contextual banditsVidya K. Muthukumar, Akshay KrishnamurthyICML 2022 · 5 citations
- EE-Net: Exploitation-Exploration Neural Networks in Contextual BanditsYikun Ban, Yuchen Yan, Arindam Banerjee, Jingrui HeICLR 2022 · 62 citations
