Graph Dimensionality Reduction for Contextual Bandits: Structure-Specific Regret Bounds under Approximate Smoothness and Noisy Eigenspaces
About
Contextual bandits with graph-structured arms arise in recommendation, citation retrieval, and social advertising, where arms connected on a graph tend to share reward signal. Standard dimensionality reduction ignores this structure, inflating exploration cost by a factor of $d/k$. We propose GraphDR-LinUCB, which projects arm features onto the graph's low-frequency spectral subspace and runs linear UCB in the resulting $k$-dimensional space. We prove the first $\wtO(k\sqrt{T})$ regret bound for spectral-projection-based contextual bandits, reducing dimension dependence from $d$ to $k$; a perturbation argument extends this to noisy graphs, with an explicit penalty for reward-smoothness mismatch and graph-estimation error. Our central theoretical finding is that the high-frequency reward component need not incur a worst-case linear-in-$T$ penalty: its actual cost depends on its realized impact along the played path, not on its total energy. A simple spectral comparison between subspaces ($\Gamma_k$) predicts which reducer wins on a given dataset, correctly calling five of six real-dataset outcomes without any fitted threshold. Across a synthetic benchmark and six real datasets (MovieLens, Amazon, LastFM, ogbn-arxiv, MIND), GraphDR-LinUCB reduces cumulative regret by $15\times$ over full-dimensional LinUCB and outperforms competing graph-aware methods on five of six; the single failure is precisely where the graph's spectral subspace is misaligned with the reward.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Contextual Bandits | MovieLens 100k | Cumulative Regret122.7 | 15 | |
| Contextual Bandits | MovieLens 1M | Cumulative Regret147.7 | 9 | |
| Contextual Bandits | MIND small | Cumulative Regret169.4 | 9 | |
| Contextual Bandits | Ogbn-arxiv | Cumulative Regret277 | 9 | |
| Bandit Recommendation | MovieLens 1M top-1500 most-rated items | Cumulative Regret R(T)273.3 | 6 | |
| Contextual Bandits | Synthetic SBM | Cumulative Regret31.3 | 5 | |
| Contextual Bandits | AMAZON | Cumulative Regret R(T)148.6 | 5 | |
| Contextual Bandits | LastFM | Cumulative Regret186.4 | 5 | |
| Contextual Bandits | Amazon Dig. Music | Cumulative Regret309.9 | 4 | |
| Contextual Bandits | LastFM HetRec | Cumulative Regret412.1 | 4 |