如何从数学上理解“探索与利用”?一篇关于概率决策算法理论分析
不确定性环境中的智能决策:贝叶斯优化 BO、蒙特卡洛树搜索 MCTS 和老虎机 Bandits算法
走向未来
概率决策算法构成了现代人工智能系统在不确定性环境中运行的核心。这些算法在材料科学、药物发现等实验成本高昂的领域展现出显著价值,它们通过自适应地收集信息,实现了数据高效的工作流。理解这些算法的行为、评估其效率,并指导下一代算法的开发,依赖于对它们进行深入的理论分析。一份技术专著详细阐述了这一分析过程(本文所分析的报告全文可以从“走向未来”知识星球中获取),其内容从决策理论的基础出发,系统性地构建了一套理解智能体如何在不确定性中学习和行动的理论框架。

添加图片注释,不超过 140 字(可选)
该分析的核心起点是决策理论本身。决策制定被抽象为在给定数据D和某个未知的潜在变量f的情况下,从一组可能的行动A中选择最优行动a。面对这一问题,存在两种主要的统计决策视角:贝叶斯主义和频率主义。
贝叶斯决策理论将概率视为信念的度量。决策的目标是基于观测到的数据D更新我们对未知变量f的后验信念p(f|D),然后选择一个行动a,以最大化在该信念下的预期效用。这种方法的核心是主观信念的更新和基于信念的效用最大化。
相对地,频率主义决策理论将概率视为长期运行的相对频率。在此框架下,未知变量f被假定为固定的,而数据D是随机生成的。决策的目标是找到一个最优的函数或策略δ(D),这个策略能将任意观测到的数据D映射到一个行动。由于真实的f未知,频率主义者常常转而寻求最小化最坏情况风险的“极小化极大”策略。该专著明确指出,这两种视角并非相互排斥,而是服务于不同目标的有效工具。这种包容性的观点为后续混合使用贝叶斯方法(如高斯过程)和频率主义方法(如置信区间)奠定了基础。
理解算法性能保证的关键,在于掌握用于分析的数学工具,即集中不等式。这些不等式是理论分析的基石,它们提供了关于随机变量偏离其期望值的概率边界。从马尔可夫不等式(仅使用均值)到切比雪夫不等式(使用均值和方差),再到更紧密的切诺夫界和霍夫丁不等式(提供指数级衰减的边界),这些工具让分析者能够量化不确定性。
这些数学工具在分析“K-臂老虎机”问题时立即显示出其威力。老虎机问题是决策理论中最简化的形式之一:在一个有K个选项的环境中,每一步选择一个行动,并获得一个随机奖励,目标是最大化长期总奖励。为了衡量算法的性能,引入了“懊悔”的概念,即算法在T步内所选行动的总奖励与始终选择最优行动所获总奖励之间的差额。一个数据高效的算法应具有“无懊悔”特性,即其平均懊悔随时间推移收敛于零。
频率主义的老虎机算法,如“上置信边界”(UCB)算法,直接体现了集中不等式的应用。UCB算法为每个臂a维护一个经验平均奖励μ(a)和一个置信项,后者通常形式为sqrt(log T / N(a)),其中T是总步数,N(a)是臂a被选择的次数。算法在每一步选择的,是最大化“经验均值”与“置信项”之和的臂。这个置信项本质上就是霍夫丁不等式导出的一个高概率边界。它精确地量化了探索(选择N(a)较小的臂以减少不确定性)和利用(选择μ(a)较高的臂以获取奖励)之间的权衡。理论分析证明,这种策略的累积懊悔是次线性的(例如,按sqrt(T log T)增长),因此保证了平均懊悔收敛于零。
当决策问题从K个离散选项扩展到优化一个未知的、昂贵的黑盒函数f(x)时,问题就进入了贝(BO)的领域。贝叶斯优化不再为每个离散的“臂”估计一个简单的均值,而是需要对整个函数f(x)建立一个信念模型。
高斯过程(GP)为此提供了强大的工具。GP是函数空间上的概率分布,它由一个均值函数和一个协方差(核)函数k完全定义。核函数k至关重要,它编码了关于函数f的先验假设,例如f的平滑程度。通过高斯过程,我们可以在观测到一组数据D后,计算出关于f的后验分布,即后验均值μ(x)和后验协方差(或标准差σ(x))。
基于GP的决策算法,如GP-UCB,与老虎机UCB算法在逻辑上惊人地一致。GP-UCB在每一步选择的评估点x,是最大化采集函数μ(x) + βσ(x)的点。这里,μ(x)代表当前的最佳估计(利用),而σ(x)代表该点的不确定性(探索)。贝叶斯框架下的GP-UCB算法,其形式与频率主义的老虎机UCB算法如出一辙,两者都实现了“在乐观情况下最大化”的策略。
对GP-UCB的懊悔分析则更为深入。理论证明其累积懊悔R(T)受到一个关键量的约束:信息容量γ(T)。信息容量γ(T)衡量了在T次观测中,我们通过GP模型最多能获得多少关于未知函数f的信息(通过互信息定义)。这个值取决于核函数k的性质。例如,对于平滑的RBF核,γ(T)增长缓慢(按(log T)的多项式增长);对于较粗糙的Matérn核,γ(T)增长较快。最终,GP-UCB的懊悔被证明为R(T) ≤ O(sqrt(T * γ(T)))。这个界限意义非凡:它清晰地表明,算法的数据效率(懊悔)直接取决于问题本身的内在复杂度(通过γ(T)量化)。一个更平滑、结构更简单的函数(γ(T)更小),是可被证明能被更快优化的。
现实世界中的优化问题往往发生在连续空间,即f(x)的定义域X是R^d中的一个子集,这是一个无限空间。理论分析通过一个精巧的桥梁将离散空间的结论扩展到了连续空间。
这个桥梁建立在两个假设之上:一是函数f满足利普S希茨连续性,即函数的“陡峭”程度有界;二是算法采用自适应离D散化的策略。分析显示,通过在时间t构建一个越来越精细的网格X(t),其网格点的数量随t增长(例如,按t^(2d)增长),算法可以将连续优化问题转化为一系列离散优化问题。
总懊悔被分解为两部分:一部分是由于离散化带来的误差,即真实最优点f(x*)与网格最优点f([x*])之间的差异,这部分懊悔被利普S希茨假设和逐渐加密的网格所约束,其总和收敛到一个常数。另一部分是在该动态网格上执行离散GP-UCB算法所产生的懊悔。通过应用离散情况下的分析,并代入动态变化的网格大小|X(t)|,最终证明了在连续空间中,GP-UCB的懊悔仍然是次线性的,从而保证了算法的收敛性和数据效率。
最后,该理论框架从单步决策(老虎机和BO)推进到多步序列决策,即规划问题。马尔可夫决策过程(MDP)是描述此类问题的标准语言。在MDP中,智能体在一系列状态s中转移,通过执行行动a来获取奖励r。规划的目标是找到一个最优策略π(a|s),以最大化累积期望奖励。
当MDP的状态空间巨大,特别是当其结构化为一棵深广的树时(例如,语言模型中的词序列生成),精确求解变得不可行。语言模型生成一个句子,可以被视为在词汇树上进行一次深度为T的搜索,其目标是找到一条具有最高“奖励”的路径(例如,由一个奖励模型评估的句子质量)。
这种将语言生成视为规划问题的视角,虽然在理论上是完备的,但在实践中揭示了一个更深层次的挑战。仅靠MCTS等算法优化一个固定的奖励模型,并不能解决大模型固有的“幻觉”和“知识陈旧”问题。正如人工智能与大模型技术专家王文广在其灯塔书《知识增强大模型》中指出的,大模型的决策过程不仅是“规划”问题,更是“知识”问题。该书深入探讨了(如第4章“检索增强生成”和第9章“知识图谱增强生成与GraphRAG”所述),如何通过检索增强生成(RAG)和知识图谱等技术,将外部的、可验证的实时知识注入到决策(生成)的每一步。因此,本文所探讨的概率决策算法(如MCTS)解决了“如何优化生成路径以符合目标”的问题,而知识增强技术则确保了这条路径上的每一步都“有据可依”,两者共同构成了构建高可靠性智能系统的关键支柱。
面对这种指数级复杂的搜索,专著介绍了两种核心的 tree search 算法。第一种是A搜索,这是一种基于启发式的搜索。A算法通过一个启发式函数h(s)来估计从状态s到达目标的未来价值V*(s)。它优先探索“已行走代价”g(s)与“未来预估代价”h(s)之和最小的节点。A*的理论保证依赖于一个关键特性:启发式函数h(s)的“可容许性”(即h(s)必须乐观地估计未来价值)。这在本质上是一种将先验领域知识(以h(s)的形式)注入搜索过程的方法。
第二种,也是更具适应性的方法,是蒙特卡洛 tree search(MCTS)。MCTS巧妙地将棘手的规划问题转变为一系列局部的、频率主义的决策问题。MCTS的核心思想是,在搜索树的每一个节点s上,选择下一个子节点s'的问题,本身就是一个“老虎机问题”。智能体需要平衡对已知高价值子树的“利用”和对未充分探索子树的“探索”。
MCTS通过UCT(树上的UCB)算法来实现这一平衡。UCT的策略与老虎机UCB完全相同,它选择的子节点k最大化了μ(k) + C * sqrt(log N(s) / N(k)),其中μ(k)是通过“rollout”(随机模拟)估计的平均累积奖励,N(s)和N(k)分别是父节点和子节点的访问次数。MCTS通过四个步骤——选择(使用UCT)、扩展(添加新节点)、模拟(rollout采样)、反向传播(更新μ和N)——迭代地构建和评估搜索树。
至此,整个理论分析形成了一个完整的闭环。它从最简单的频率主义决策(老虎机UCB)出发,展示了其理论保证如何依赖于集中不等式;接着,它将这一逻辑平行移植到贝叶斯领域(GP-UCB),用于解决更复杂的函数优化问题,并将其性能与问题的内在信息复杂度(γ(T))相联系;最后,在处理指数级复杂的序列规划问题时,MCTS算法通过将规划问题递归地分解为一系列局部的老虎机问题,再次回归到了UCB这一频率主义的基石算法。
这份专著的深刻价值在于,它不仅介绍了单一的算法,而是揭示了算法之间深刻的理论联系,以及它们共享的数学基础。它证明了这些自适应算法为何高效,并提供了量化其性能的工具,为构建更强大、更可靠的智能决策系统提供了坚实的理论基石。

添加图片注释,不超过 140 字(可选)
