Federated X-Armed Bandit

Abstract

This work establishes the first framework of federated X-armed bandit, where different clients face heterogeneous local objective functions defined on the same domain and are required to collaboratively figure out the global optimum. We propose the first federated algorithm for such problems, named Fed-PNE. By utilizing the topological structure of the global objective inside the hierarchical partitioning and the weak smoothness property, our algorithm achieves sublinear cumulative regret with respect to both the number of clients and the evaluation budget. Meanwhile, it only requires logarithmic communications between the central server and clients, protecting the client privacy. Experimental results on synthetic functions and real datasets validate the advantages of Fed-PNE over various centralized and federated baseline algorithms.

Cite

Text

Li et al. "Federated X-Armed Bandit." AAAI Conference on Artificial Intelligence, 2024. doi:10.1609/AAAI.V38I12.29267

Markdown

[Li et al. "Federated X-Armed Bandit." AAAI Conference on Artificial Intelligence, 2024.](https://mlanthology.org/aaai/2024/li2024aaai-federated/) doi:10.1609/AAAI.V38I12.29267

BibTeX

@inproceedings{li2024aaai-federated,
  title     = {{Federated X-Armed Bandit}},
  author    = {Li, Wenjie and Song, Qifan and Honorio, Jean and Lin, Guang},
  booktitle = {AAAI Conference on Artificial Intelligence},
  year      = {2024},
  pages     = {13628-13636},
  doi       = {10.1609/AAAI.V38I12.29267},
  url       = {https://mlanthology.org/aaai/2024/li2024aaai-federated/}
}