[音乐] 大家好,今天我们要深入探讨一篇论文,它提出了一个问题:强化学习能否成为终极的多任务专家?本期的焦点是一种名为 MRQ 的方法。让我们先聊聊它的名字,是 Mr. Q、MRQ,还是像特工代号一样的 MRQ?我们就选听起来更酷的 MRQ 吧。那么它的核心理念是什么?MRQ 是“基于模型的 Q 学习表示”(Model-based Representations for Q learning)的缩写。它是一种模型无关的强化学习方法,借鉴了基于模型方法的一些技巧(如更智能的奖励结构),但省去了繁重的规划开销。其目标是创建一个通用的强化学习算法,无需大量调整即可在不同任务中发挥作用。你可以把它想象成瑞士军刀般的强化学习工具,随时准备大显身手。让我们深入探讨吧!

我们常听说强化学习是解决复杂问题的终极工具箱,但在实践中并非如此简单。大多数强化学习算法最终都针对特定任务或基准进行了精细调整,因此在迁移到新领域时泛化能力往往不佳。目前有一种有前景的方法叫做基于模型的强化学习,系统在其中构建内部世界模型以辅助规划。这些方法效果显著,但速度慢且复杂,难以广泛应用。因此,这篇论文解决了一个核心问题:我们能否创建一个既通用又无需显式规划开销的模型无关强化学习算法?这就是 MRQ 登场的原因。MRQ 是一种深度强化学习方法,借鉴了基于模型方法的一些技巧(如更丰富的奖励结构),但省去了显式规划的开销。其理念是兼收并蓄:在无需额外计算成本的情况下获得强大性能,结果相当令人鼓舞。

请看图一,展示了 MRQ 在四个不同强化学习环境中的表现,每个环境都有独特的挑战。与 Dreamer V3 和 TD-MPC2 相比,MRQ 所需的参数更少、训练速度更快且评估效率更高。基准测试涵盖了从经典的移动任务到高维像素游戏,展示了其多功能性。结论是:模型无关的强化学习不必低效;通过适当的调整,它既可以强大又实用。这篇论文为思考强化学习的新方式奠定了基础,推动着真正通用智能体的发展。但现实情况是:大多数强化学习算法实际上并非通用的,它们是专为特定问题构建的。

例如,有些算法擅长处理离散动作,而另一些则在连续动作方面表现出色;有些与基于向量的观察数据配合良好,而另一些则专为像素输入设计。因此,每个类别都需要其特有的算法调整和超参数微调。以 Rainbow 和 TD3 为例:Rainbow 专为 Atari 游戏构建,而 TD3 则为 MuJoCo 任务设计。尽管它们都是模型无关的强化学习方法,但共同点并不多。请看表一,这里比较了它们的超参数:你会发现 TD3 倾向于更大的批次大小和更快的更新频率,而 Rainbow 则采用更渐进的优化策略。

虽然确实存在一些通用强化学习方法(如策略梯度方法和进化算法),但这些方法通常对环境有强烈的假设,且性能往往不如领域特定方法。最近的基于模型的方法(如 Dreamer V3 和 TD-MPC2)在泛化方面取得了进展,但它们带来了额外的复杂性。那么这里的思路是什么?作者希望借鉴基于模型强化学习的优势(如其如何构建学习目标),但将其应用于模型无关的设定中。他们的 MRQ 方法学习表示,捕捉状态-动作对与价值估计之间的近乎线性关系。如果这一思路可行,它将消除在不同环境中微调超参数的需求。

为了严谨地验证这一想法,MRQ 在四个不同的强化学习基准和超过 100个环境中进行了评估。其目标是看它是否能在不需要算法或超参数更改的情况下,达到最先进方法的性能水平。为了理解 MRQ 的工作原理,让我们退一步谈谈强化学习的基础知识。核心而言,强化学习依赖于一个称为马尔可夫决策过程(MDP)的数学框架。该框架定义了智能体做出决策所需的一切:其可能处于的状态、可采取的动作、这些动作如何改变环境以及奖励的给予方式。智能体的目标是学习一种策略,以选择最大化长期回报的动作。

实现这一目标的一种方法是基于价值的方法。这些方法估计一个函数,预测智能体从给定状态和动作中期望获得的未来回报数。但棘手之处在于:这个函数需要随时间学习以保持训练稳定。深度强化学习方法通常使用所谓的“目标网络”(Target networks)。这些是价值函数的副本,其更新速度比主网络慢。这就像拥有一张安全网,防止情况失控。在此基础上,论文引入了一种称为“基于动力学的表示学习”的技术。这有助于强化学习智能体通过结构化其学习方式来实现状态-动作关系的泛化。

这种方法不是从头开始学习策略,而是构建对环境行为方式的表示。作者认为,这种学习方式可以取代显式的基于模型的规划,同时仍能捕捉数据中的有用结构。这里另一个重要概念是“状态抽象”。与将每个可能的状态视为独特不同,抽象方法寻找相似性。这在复杂环境中非常有用,因为学习每一个微小细节是不切实际的。通过将类似状态分组在一起,强化学习智能体可以更好地泛化,并避免对特定情况的过拟合。

最后,作者过渡到讨论他们的方法 MRQ。他们认为该方法与现有的状态-动作表示方法非常相似,但在如何结构化奖励和转换函数中的信息损失方面有所不同。其目标是在保持强大性能的同时提高效率。现在让我们谈谈 MRQ 背后的关键思想之一:中间嵌入(Intermediate embeddings)。大多数强化学习算法直接将状态和动作映射到价值,而不考虑更深层的结构。但 MRQ 采取了不同的方法:它不直接使用原始输入,而是首先将状态和状态-动作对转换为更抽象的表示。这些嵌入随后作为策略和价值函数的输入。

价值函数被分解为两个组成部分:特征表示和权重参数。他们比较了两种学习这些权重的方法。无模型方法使用标准的强化学习技术来更新权重,而基于模型的方法则通过滚动预测未来奖励和环境动态来进行估计。这里变得有趣起来的是,定理1表明这两种方法最终都会达到相同的解。这表明无模型学习可以在不需要完整环境建模的情况下,近似基于模型学习的优势。

定理2进一步通过证明基于模型的解的准确性取决于系统预测环境动态的能力,将这一观点推向深入。得出的结论是:你并不总是需要显式模型才能从结构化学习中受益。通过使用嵌入和结构化表示学习,你可以以更轻量级的方式实现类似的结果。深度强化学习通常面临训练不稳定的问题,其中一个主要问题源于策略与所学表示之间的相互作用。

如果一个强化学习模型在学习策略的同时也在学习环境内部表示,这两个过程可能会相互干扰,导致学习不一致。为了解决这个问题,论文引入了一种损失函数来平衡两个目标:学习与奖励对齐的表示,以及捕捉环境变化的表示。但这里存在一个问题:原始方法存在问题。学习到的特征依赖于策略选择的未来动作,这使得训练不稳定。

此外,如果所有内容都一起优化,系统可能会陷入糟糕的局部极小值,导致平凡的解。那么解决方案是什么?首先,与其让学习到的特征依赖于动作,不如让它们仅依赖于状态。这使得训练更加稳定,同时仍能捕捉重要的环境细节。其次,他们使用目标网络(Target Network),而不是基于最新数据不断进行更新。系统通过缓慢调整单独的参考模型来稳定学习过程。

这里的一个关键结果是定理3,它表明即使学习到的特征不能完美匹配真实价值函数,它们仍然对学习最优策略有用。这表明深度强化学习系统不需要对环境拥有完美准确的模型才能成功,只需要一种从数据中学习的结构化方式。这为论文的下一部分奠定了基础,在第二部分中他们展示了该方法在实际中的表现。

现在让我们深入MRQ的核心算法。关键思想是学习一种嵌入,以与真实价值函数对齐的方式捕捉状态-动作关系。由于完美近似是不可能的,该方法使用非线性函数逼近从这些嵌入中估计值。学习过程分为几个步骤进行:首先,状态通过状态编码器转换为嵌入;其次,状态-动作对分别被嵌入,允许系统表示不同的输入模态,同时保持统一的学习结构。

这些嵌入传递给预测器,以估计未来状态、奖励以及剧集是否会结束。为了使学习稳定,MRQ应用了目标网络策略,而不是同时更新所有内容。某些参数会定期更新以防止不稳定性。注意他们使用的另一种技术是奖励缩放,它随时间规范化奖励值。这防止极端值主导学习过程。

编码器的损失函数包含三个组成部分:它确保正确预测奖励,状态之间的转换合理,并且剧集终止被准确建模。最终损失函数以不同的权重平衡这些项。这种设置允许MRQ在处理不同类型输入的同时保持稳定的训练。通过以这种方式结构化学习,它避免了常见的强化学习陷阱,如不稳定的更新和糟糕的局部极小值。

强化学习中最大的挑战之一是确保智能体学习稳定且可靠的转换。本节重点介绍有助于实现这一目标的两个关键损失函数。首先是动态损失(Dynamics loss),它确保模型准确预测状态随时间的转换方式。与使用完整的状态-动作基于的预测不同,它仅依赖于状态本身,从而使学习更加稳定。

第二个是终止损失(Terminal loss),它帮助模型预测剧集是否正在结束,防止其对何时停止做出错误假设。对于价值学习,MRQ建立在TD3的基础上,TD3因通过训练两个独立的价值函数并使用较小的一个来指导学习而以提高稳定性著称。这防止了强化学习中可能出现的过估计误差。

与使用单步奖励不同,MRQ预测未来多个步骤,使其更具样本效率。作者还应用了Huber损失而不是简单的平方损失,以防止极端值干扰训练。最后,策略更新使用确定性策略梯度方法,意味着它直接优化动作选择过程。添加了一个小的正则化项以防止策略对特定状态过拟合,从而处理不同类型的动作空间。

作者针对离散动作应用了不同的技巧:他们使用Gumbel Softmax技巧;对于连续动作,则使用Tanh激活函数以保持动作在有效范围内。这些技术共同提高了训练的稳定性和效率,确保MRQ能够泛化到不同类型的环境中。现在谈谈实验部分。这里的目标不仅仅是击败现有方法,而是展示MRQ能否在相同的一组超参数下在各种任务中表现良好。

请查看图2,它绘制了不同强化学习方法随时间学习的情况。基准测试涵盖了广泛的强化学习挑战。首先是Gym Locomotion,专注于MuJoCo环境中的模拟行走和跑步任务。其次是DMC Proprioceptive(本体感觉),涉及使用传感器数据而非图像进行机器人控制。第三组是DMC Visual,它采用相同的任务,但用基于像素的图像替换了基于向量的输入,使学习变得更加困难。

最后是Atari,它在经典视频游戏任务上评估MRQ,要求离散动作控制。关键结论是:MRQ的表现与领域特定算法和通用方法具有竞争力。它针对Dreamer V3、TD-MPC2和PPO等强大的基线进行了测试。结果显示,MRQ可以匹配或超越这些方法,而无需为不同的基准测试设置不同的超参数。

为了更直观地理解这一点,训练时间在环境之间差异很大。在DMC任务中,50万时间步对应模拟器中的100万帧;而在Atari中,250万时间步对应1,000万帧。当你查看学习曲线时,请注

总体而言,这些结果表明 MRQ 是一种高效的无模型替代方案,能够在多种类型的强化学习任务中实现良好的泛化能力。这些实验的一个重要启示是:强化学习并没有一种放之四海而皆准的解决方案,不同的方法在不同的任务上表现各异。这一现象在机器学习中常被称为“没有免费午餐”定理。MRQ 在 DeepMind Control (DMC) 基准测试中表现最佳,展示了其在基于向量和基于像素的观察空间中进行泛化的能力。

在 Gym Locomotion 任务中,MRQ 的表现略逊于 TD7,但总体而言,它是 Atari 连续控制任务中最强的方法。Dreamer V3 的表现优于 MRQ,但这是以拥有比 PPO、DQN 和 Rainbow 等无模型基线多 40 倍的参数为代价的。MRQ 在离散动作设置中表现更好。为了更深入地理解哪些设计选择影响了 MRQ 的性能,作者进行了一项总结在表二中的设计研究。

这项研究测试了 MRQ 架构和损失函数的各种变体,以观察它们对性能的影响程度。一些关键发现包括:将非线性值函数切换为线性值函数会显著降低性能;改变动态目标,使其依赖状态动作编码器而非分离的状态嵌入,引入了不稳定性;移除目标编码器导致整体性能下降。

使用单步回报而非多步回报会削弱长期学习,特别是在 Atari 房间任务中;移除奖励缩放会在某些环境中引入不稳定性。底线是:MRQ 的设计经过精心调整,以平衡稳定性和泛化能力。即使是微小的改动,如奖励的缩放方式或值函数的结构,也可能导致性能的巨大差异。

这里的讨论强调了一个重要的启示:在强化学习中,平衡理论思想与实际调整至关重要。设计研究证实了 MRQ 的架构选择是经过深思熟虑的。两个关键发现脱颖而出:首先,简单地增加模型的复杂性(在非模型版本中)并没有提高性能;相反,保持与值函数的近似线性关系更为重要。

其次,Gym 和 Atari 基准测试的表现不同,某些在一个环境中效果良好的调整在另一个环境中却损害了性能。这表明强化学习算法可能会过拟合特定环境,使得广泛泛化变得困难。结论总结了 MRQ 的贡献:一种通用的无模型强化学习方法,无需针对特定任务进行调整即可在不同类型的环境中表现良好。

通过使用基于模型的表示而不进行完整的基于模型的计划,MRQ 有效地模糊了传统无模型和基于模型强化学习之间的界限。作者认为,在某些情况下,完整的基于模型的推理甚至可能不是必需的;仅凭结构化的表示学习就能产生类似的好处。另一个主要的启示是:在强化学习基准测试中缺乏强烈的正向迁移效应。

仅仅因为一种算法在 MuJoCo Gym 中表现良好,并不意味着它会在 Atari 中脱颖而出,反之亦然。这加强了在评估新的强化学习方法时进行更广泛和更多样化测试的必要性。话虽如此,MRQ 并非完美无缺;它在需要长视界推理或重度探索策略的任务中表现不佳,且尚未在视频游戏或无人机竞速等大型任务中与 Po 等方法进行过测试。

未来的工作需要扩大评估范围,以确保这些发现能够进一步推广。结论以乐观的语调结束:更好的通用强化学习方法最终可能让任何人都能轻松训练自己的智能体,也许甚至只需轻轻点击一个按钮。感谢大家今天的参与,祝大家学习愉快并共同进步。[音乐]