Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Joyanta Jyoti Mondal, Ibne Farabi Shihab, Anuj Sharma• 2026

Related benchmarks

TaskDatasetResultRank
Contextual BanditsMovieLens 100k
Cumulative Regret122.7
15
Contextual BanditsMovieLens 1M
Cumulative Regret147.7
9
Contextual BanditsMIND small
Cumulative Regret169.4
9
Contextual BanditsOgbn-arxiv
Cumulative Regret277
9
Bandit RecommendationMovieLens 1M top-1500 most-rated items
Cumulative Regret R(T)273.3
6
Contextual BanditsSynthetic SBM
Cumulative Regret31.3
5
Contextual BanditsAMAZON
Cumulative Regret R(T)148.6
5
Contextual BanditsLastFM
Cumulative Regret186.4
5
Contextual BanditsAmazon Dig. Music
Cumulative Regret309.9
4
Contextual BanditsLastFM HetRec
Cumulative Regret412.1
4
Showing 10 of 10 rows

Other info

Follow for update