多智能体协作下不共享数据的线性对决强化学习算法
Federated Linear Dueling Bandits
- 用在线梯度下降结合联邦学习,解决参数无闭式解难题
- 累积后悔呈次线性增长,通信开销与性能可权衡
- 适合隐私敏感场景下的推荐系统与大模型优化
上下文线性对决强化学习因在推荐系统和大语言模型等领域的广泛应用而备受关注。传统对决强化学习算法仅适用于单智能体,但实际应用中常需多个不愿共享数据的智能体协同提升性能。受联邦学习启发,我们提出融合在线梯度下降(OGD)与联邦学习的联邦线性对决强化学习(FLDB-OGD)算法。由于线性对决强化学习中参数无闭式表达,需通过最小化损失函数估计,传统依赖闭式更新的方法不适用。本工作通过将OGD与联邦学习结合,首次实现该场景下的有效协作。理论分析证明,该算法累积后悔具有次线性上界,并揭示了后悔与通信复杂度之间的权衡关系。实验验证了算法有效性,发现更多智能体带来更好性能,且进一步揭示了通信频率对结果的影响。
原文摘要 · Abstract (English)
Contextual linear dueling bandits have recently garnered significant attention due to their widespread applications in important domains such as recommender systems and large language models. Classical dueling bandit algorithms are typically only applicable to a single agent. However, many applications of dueling bandits involve multiple agents who wish to collaborate for improved performance yet are unwilling to share their data. This motivates us to draw inspirations from federated learning, which involves multiple agents aiming to collaboratively train their neural networks via gradient descent (GD) without sharing their raw data. Previous works have developed federated linear bandit algorithms which rely on closed-form updates of the bandit parameters (e.g., the linear function parameters) to achieve collaboration. However, in linear dueling bandits, the linear function parameters lack a closed-form expression and their estimation requires minimizing a loss function. This renders these previous methods inapplicable. In this work, we overcome this challenge through an innovative and principled combination of online gradient descent (OGD, for minimizing the loss function to estimate the linear function parameters) and federated learning, hence introducing our federated linear dueling bandit with OGD (FLDB-OGD) algorithm. Through rigorous theoretical analysis, we prove that FLDB-OGD enjoys a sub-linear upper bound on its cumulative regret and demonstrate a theoretical trade-off between regret and communication complexity. We conduct empirical experiments to demonstrate the effectiveness of FLDB-OGD and reveal valuable insights, such as the benefit of a larger number of agents, the regret-communication trade-off, among others.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。