When Greedy Sampling Explores: KL-Regularized Contextual Bandits without Eluder-Dimension Dependence
cs.LG
Submitted: 2026-09-11
Updated: 2026-09-20
License: http://creativecommons.org/licenses/by/4.0/
The gist: We study KL-regularized contextual bandits under both reward and preference feedback.
Terminology
Abstract
We study KL-regularized contextual bandits under both reward and preference feedback. We show that greedy sampling can achieve logarithmic regret without explicit dependence on the eluder dimension. For reward feedback, we establish an eluder-dimension-independent regret bound for a simple greedy algorithm that directly samples from the Gibbs policy induced by the estimated reward. We further extend this result to preference feedback under both the general preference and Bradley--Terry models, while also sharpening existing dimension-dependent guarantees. Our analysis reveals a trade-off between greedy sampling and upper confidence bound-style exploration: greedy sampling enjoys stronger guarantees when KL regularization is sufficiently strong, whereas additional exploration becomes preferable as the regularization weakens.
Sources
- On the Optimal Sample Complexity of Offline Multi-Armed Bandits with KL Regularization
- Upper Counterfactual Confidence Bounds: a New Optimism Principle for Contextual Bandits
- Fast Rates for Offline Contextual Bandits with Forward-KL Regularization under Single-Policy Concentrability
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks