[2604.00531] Learning Shared Representations for Multi-Task Linear Bandits

Computer Science > Machine Learning arXiv:2604.00531 (cs) [Submitted on 1 Apr 2026] Title:Learning Shared Representations for Multi-Task Linear Bandits Authors:Jiabin Lin, Shana Moothedath View a PDF of the paper titled Learning Shared Representations for Multi-Task Linear Bandits, by Jiabin Lin and Shana Moothedath View PDF HTML (experimental) Abstract:Multi-task representation learning is an approach that learns shared latent representations across related tasks, facilitating knowledge transfer and improving sample efficiency. This paper introduces a novel approach to multi-task representation learning in linear bandits. We consider a setting with T concurrent linear bandit tasks, each with feature dimension d, that share a common latent representation of dimension r \ll min{d,T}$, capturing their underlying relatedness. We propose a new Optimism in the Face of Uncertainty Linear (OFUL) algorithm that leverages shared low-rank representations to enhance decision-making in a sample-efficient manner. Our algorithm first collects data through an exploration phase, estimates the shared model via spectral initialization, and then conducts OFUL based learning over a newly constructed confidence set. We provide theoretical guarantees for the confidence set and prove that the unknown reward vectors lie within the confidence set with high probability. We derive cumulative regret bounds and show that the proposed approach achieves \tilde{O}(\sqrt{drNT}), a significant improvement ...