English

Federated Linear Bandits with Finite Adversarial Actions

Machine Learning 2023-11-03 v1 Information Theory math.IT Machine Learning

Abstract

We study a federated linear bandits model, where MM clients communicate with a central server to solve a linear contextual bandits problem with finite adversarial action sets that may be different across clients. To address the unique challenges of adversarial finite action sets, we propose the FedSupLinUCB algorithm, which extends the principles of SupLinUCB and OFUL algorithms in linear contextual bandits. We prove that FedSupLinUCB achieves a total regret of O~(dT)\tilde{O}(\sqrt{d T}), where TT is the total number of arm pulls from all clients, and dd is the ambient dimension of the linear model. This matches the minimax lower bound and thus is order-optimal (up to polylog terms). We study both asynchronous and synchronous cases and show that the communication cost can be controlled as O(dM2log(d)log(T))O(d M^2 \log(d)\log(T)) and O(d3M3log(d))O(\sqrt{d^3 M^3} \log(d)), respectively. The FedSupLinUCB design is further extended to two scenarios: (1) variance-adaptive, where a total regret of O~(dt=1Tσt2)\tilde{O} (\sqrt{d \sum \nolimits_{t=1}^{T} \sigma_t^2}) can be achieved with σt2\sigma_t^2 being the noise variance of round tt; and (2) adversarial corruption, where a total regret of O~(dT+dCp)\tilde{O}(\sqrt{dT} + d C_p) can be achieved with CpC_p being the total corruption budget. Experiment results corroborate the theoretical analysis and demonstrate the effectiveness of FedSupLinUCB on both synthetic and real-world datasets.

Keywords

Cite

@article{arxiv.2311.00973,
  title  = {Federated Linear Bandits with Finite Adversarial Actions},
  author = {Li Fan and Ruida Zhou and Chao Tian and Cong Shen},
  journal= {arXiv preprint arXiv:2311.00973},
  year   = {2023}
}

Comments

Accepted to NeurIPS 2023, camera-ready version