arXiv:2410.17835cs.LG2024-10ICML被引 18

提出高效流式算法,单内存单遍扫描下精准识别最优多臂

Optimal Streaming Algorithms for Multi-Armed Bandits

  • 基于单内存单遍扫描,实现对前k优臂的近似最优采样
  • 在ε-精度下,样本复杂度达O(n/ε² log(k/δ))
  • 适用于资源受限场景,适合在线学习与实时决策

本文研究两种流式模型下的最佳臂识别(BAI)问题。在流式设置中,有n个臂依次到来,每个臂的奖励分布支持在[0,1]上且均值未知,算法只能在有限内存中存储臂。首先研究流式ε-top-k臂识别问题:要求以至少1−δ的概率找出k个均值不超过第k佳臂均值ε的臂。针对一般ε∈(0,1),现有方法仅适用于k=1,需使用O(log*(n))内存并达到最优样本复杂度O(n/ε² log(1/δ))。本文提出一种新算法,适用于任意k,仅用单臂内存和单遍扫描即可实现最优样本复杂度O(n/ε² log(k/δ))。其次研究流式BAI问题,目标是以单臂内存和最少遍数,以至少1−δ的概率识别出均值最大的臂。我们提出一种单臂内存算法,在O(log Δ₂⁻¹)遍内达到近似实例依赖最优样本复杂度,其中Δ₂为最优臂与次优臂均值之差。

原文摘要 · Abstract (English)

This paper studies two variants of the best arm identification (BAI) problem under the streaming model, where we have a stream of $n$ arms with reward distributions supported on $[0,1]$ with unknown means. The arms in the stream are arriving one by one, and the algorithm cannot access an arm unless it is stored in a limited size memory. We first study the streaming \eps-$top$-$k$ arms identification problem, which asks for $k$ arms whose reward means are lower than that of the $k$-th best arm by at most $\eps$ with probability at least $1-δ$. For general $\eps \in (0,1)$, the existing solution for this problem assumes $k = 1$ and achieves the optimal sample complexity $O(\frac{n}{\eps^2} \log \frac{1}δ)$ using $O(\log^*(n))$ ($\log^*(n)$ equals the number of times that we need to apply the logarithm function on $n$ before the results is no more than 1.) memory and a single pass of the stream. We propose an algorithm that works for any $k$ and achieves the optimal sample complexity $O(\frac{n}{\eps^2} \log\frac{k}δ)$ using a single-arm memory and a single pass of the stream. Second, we study the streaming BAI problem, where the objective is to identify the arm with the maximum reward mean with at least $1-δ$ probability, using a single-arm memory and as few passes of the input stream as possible. We present a single-arm-memory algorithm that achieves a near instance-dependent optimal sample complexity within $O(\log Δ_2^{-1})$ passes, where $Δ_2$ is the gap between the mean of the best arm and that of the second best arm.

强化学习流式计算多臂赌博机

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。