中文

多臂老虎机的最优流式算法

机器学习 2024-10-24 v1

摘要

本文研究了流式模型下最佳臂识别(BAI)问题的两个变体,其中我们有一个包含nn个臂的流,其奖励分布支撑在[0,1][0,1]上且均值未知。流中的臂逐个到达,算法无法访问未存储在有限大小内存中的臂。我们首先研究流式\eps-toptop-kk臂识别问题,该问题要求以至少1δ1-\delta的概率找出kk个臂,其奖励均值与第kk优臂的均值之差不超过\eps\eps。对于一般的\eps(0,1)\eps \in (0,1),该问题的现有解决方案假设k=1k = 1,并使用O(log(n))O(\log^*(n))log(n)\log^*(n)等于对nn重复应用对数函数直到结果不大于1的次数)内存和单遍流,实现了最优样本复杂度O(n\eps2log1δ)O(\frac{n}{\eps^2} \log \frac{1}{\delta})。我们提出了一种适用于任意kk的算法,使用单臂内存和单遍流,实现了最优样本复杂度O(n\eps2logkδ)O(\frac{n}{\eps^2} \log\frac{k}{\delta})。其次,我们研究了流式BAI问题,其目标是以至少1δ1-\delta的概率识别出具有最大奖励均值的臂,使用单臂内存并尽可能少地遍历输入流。我们提出了一种单臂内存算法,在O(logΔ21)O(\log \Delta_2^{-1})遍内实现了接近实例相关的最优样本复杂度,其中Δ2\Delta_2是最优臂均值与次优臂均值之间的差距。

关键词

引用

@article{arxiv.2410.17835,
  title  = {Optimal Streaming Algorithms for Multi-Armed Bandits},
  author = {Tianyuan Jin and Keke Huang and Jing Tang and Xiaokui Xiao},
  journal= {arXiv preprint arXiv:2410.17835},
  year   = {2024}
}

备注

24pages