Calibration-Gated LLM Pseudo-Observations for Online Contextual Bandits
Not on the map. Why?
Abstract (arXiv)
Contextual bandit algorithms suffer from high regret during cold-start, when the learner has insufficient data to distinguish good arms from bad. We propose augmenting Disjoint LinUCB with LLM pseudo-observations: after each round, a large language model predicts counterfactual rewards for the unplayed arms, and these predictions are injected into the learner as weighted pseudo-observations. The injection weight is controlled by a calibration-gated decay schedule that tracks the LLM's prediction accuracy on played arms via an exponential moving average; high calibration error suppresses the LLM's influence, while accurate predictions receive higher weight during the critical early rounds. We evaluate on two contextual bandit environments - UCI Mushroom (2-arm, asymmetric rewards) and MIND-small (5-arm news recommendation) - and find that when equipped with a task-specific prompt, LLM pseudo-observations reduce cumulative regret by 19% on MIND relative to pure LinUCB. However, generic counterfactual prompt framing increases regret on both environments, demonstrating that prompt design is the dominant factor, more important than the choice of decay schedule or calibration gating parameters. We analyze the failure modes of calibration gating on domains with small prediction errors and provide a theoretical motivation for the bias-variance trade-off governing pseudo-observation weight.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 8 of its 24 references that are in the map. First posted on arXiv 16 Apr 2026.
Builds on
8 of 24 references.
References · 8
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2022Language Models (Mostly) Know What They Knowcited by 1,856
- 2023Bypassing the Simulator: Near-Optimal Adversarial Linear Contextual Banditscited by 19
- 2024Efficient Sequential Decision Making with Large Language Modelscited by 16
- 2024Jump Starting Bandits with LLM-Generated Prior Knowledgecited by 12
- 2023LLMs-augmented Contextual Banditcited by 8
- 2025Large Language Model-Enhanced Multi-Armed Banditscited by 7
- 2023RoME: A Robust Mixed-Effects Bandit Algorithm for Optimizing Mobile Health Interventionscited by 5
Built on by
Not cited yet.