Skip to content

算法Algorithm

UCB 算法

施工中 · 内容逐步修订

UCB1 · upper confidence bound algorithm

在有界随机多臂赌博机中选择经验均值加置信半径最大的手臂。

形式陈述 ​

模型与算法 ​

有整数 K≥1 个臂,臂 i 的奖励独立同分布于 [0,1],均值为 μi。UCB1 以累计随机赌博机遗憾为目标,先把每臂拉一次;若时域在初始化中结束,就在该处停止。初始化后在时刻 t 选择

It∈arg⁡maxi{μ^i(t−1)+2log⁡tNi(t−1)},

其中 Ni(t−1) 是此前拉臂次数,μ^i 是对应奖励均值。平票规则可任意固定。指数中的常数与证明采用的置信事件必须配套;“UCB”还包括许多方差、自归一化或 KL 版本,本页只陈述 UCB1。

直接实现维护 K 个计数与均值,占 O(K) 空间;每轮重算并扫描 K 个指数,需 O(K) 算术操作,收到奖励后仅所选臂的计数和均值作 O(1) 更新。即使其余臂的观测不变,log⁡t 仍会改变它们的半径;算法可运行到给定时域 T,也可在线继续而不预先知道终止轮。

次优臂拉取界 ​

令 T≥2、μ∗=maxiμi、Δi=μ∗−μi>0。固定一个最优臂。若次优臂 i 在已经拉取 n 次后仍被选中,则至少发生以下之一:最优臂均值向下偏离超过其半径、臂 i 均值向上偏离超过其半径,或 22log⁡t/n≥Δi。对前两类,先固定臂内样本数 s≤t−1,Hoeffding 不等式给出每个单侧事件概率至多 t−4;对这两个臂的所有 s 取并集界,每轮概率至多 2(t−1)t−4≤2t−3,对时间可求和。后一项在 n>8log⁡T/Δi2 后不再可能;整数取整和初始化只增加常数。由此得到

ENi(T)≤8log⁡TΔi2+O(1),

以及伪遗憾

Reg―T=∑i:Δi>0ΔiENi(T)=O(∑i:Δi>0log⁡TΔi).
直觉

UCB 把不确定性下的乐观原则写成一个可比较指数:经验均值表示已经观测到的收益,置信半径表示仍未排除的上升空间。常拉的臂半径收缩,少拉的臂仍可能显得最好,于是探索与利用在每轮由同一条上界共同决定。

次优臂只有在估计发生罕见偏差,或它的半径仍大到足以覆盖 gap 时才会被选中。前两类事件的概率可求和,后一类在样本数达到 O((log⁡T)/Δi2) 后消失,这正是拉取界的结构来源。

最高置信上界选臂
例子与边界

一轮实际选择 ​

设第 101 轮开始前,三个臂的拉取次数为 (80,15,5),合计恰为此前的 100 轮;经验均值为 (0.7,0.5,0.2)。代入本页公式:

臂 经验均值 置信半径 UCB 指数
1 0.7 0.33967 1.03967
2 0.5 0.78444 1.28444
3 0.2 1.35869 1.55869

算法选择第三臂:它的已观测收益最低,但仅有五次观测,尚未排除的上升空间最大。指数可以大于奖励上限 1,因为它是用于排序的置信上界,并非奖励预测值;本页算法按公式直接比较,不截断这些数。

若这次收到奖励 0,第三臂次数变为 6,奖励总和仍为 5×0.2=1,新均值为 1/6。下一轮用 t=102 和更新后的次数重新计算指数,其他两臂的经验均值保持不变。

自适应样本数的严谨点 ​

Ni(t) 由过去奖励决定,不能把它当作预先固定样本量无条件套 Hoeffding。常见处理是先为每个确定的臂内样本数 n 建立事件,再对 i,t,n 使用并集界,或使用置信序列先控制每个臂内所有样本数的覆盖,再分配跨臂错误预算。这样在随机时刻读取经验均值仍有保证。

边界 ​

Δi 很小时,上述 gap-dependent 界可能比 minimax KT 量级更差;它不是所有实例上的最佳表达。奖励范围未知、重尾、非平稳或对抗生成时,Hoeffding 半径失效,需要稳健均值或另一 bandit 模型。UCB1 控制累计 regret,也不是最佳臂识别的最优停止算法。

推论与应用

拉取界还能帮助理解探索预算:一个臂与最优均值的差距缩小一半,标准分析中为排除它所需的观测数就约增为四倍。因此预算主要花在难以区分的近邻上;这也解释了为何只看经验均值接近,并不足以宣布某个臂已经获胜。

UCB 是一族算法而非单一公式。方差感知、KL-UCB、线性 UCB 与置信序列版本会改变半径和证明事件;迁移时应从奖励范围、尾部模型和自适应覆盖要求重新推导,不应只替换根号中的常数。

参考资料
关系图谱11 个相邻概念 · 3 类关系

拖动节点调整位置。

显示关系

显示:依赖

  1. 前置三跳
  2. 前置二跳
  3. 前置一跳
  4. 当前条目
  5. 后续一跳
  6. 后续二跳
  7. 后续三跳
文字版关系按与当前条目的最短距离分组
类型化关系