理解并改进双曲深度强化学习
Understanding and Improving Hyperbolic Deep Reinforcement Learning
基于摘要分析。本文研究双曲深度强化学习的优化失稳问题:双曲空间的指数体积增长有利于低失真地表示强化学习状态间的层级关系,但表示优势不一定能转化为稳定训练。作者通过分析 Poincaré Ball 与 Hyperboloid 模型中核心运算的梯度,提出稳定化智能体 Hyper++。 机制上,作者指出大范数嵌入会使梯度训练失稳,并导致 proximal policy optimization(PPO)的信赖域违规。Hyper++ 包含三项组件:保证特征范数有界、同时避免裁剪所致维度灾难的特征正则化;用于稳定 critic 训练的 categorical value loss;以及更利于优化的双曲网络层表述。具体正则项、损失形式和网络层实现需查阅全文,不能仅由摘要确定。 在 ProcGen 上,作者报告 Hyper++ 实现稳定学习、优于既有双曲智能体,并使实际运行时间约减少 30%;摘要未明确该时间降幅对应的具体基线、硬件与训练预算。在采用 Double DQN 的 Atari-5 实验中,作者报告其优于欧氏及双曲基线,但未提供具体分数。实验协议、消融及统计信息尚未验证。摘要提供了代码仓库,但运行依赖与复现完整性尚未核对。 平台推测(待验证):其潜在可迁移对象是采用强化学习的 BCI 自适应策略,而非 EEG 解码器本身。假设此类任务的状态关系具有适合双曲表示的层级结构,且出现嵌入范数增大与策略更新失稳,则有界特征正则化和双曲层表述可能值得复用;状态编码、动作空间及奖励设计仍需针对 BCI 改造。低信噪比、跨被试差异和小数据可能使层级结构难以可靠学习。可先在固定数据划分与交互预算下,对同一离线 EEG 策略任务比较欧氏表示、原双曲表示及加入稳定化组件的版本,检查嵌入范数、策略更新稳定性与任务回报。相关 EEG 工作尚未检索确认。
该研究将双曲表示的大范数梯度不稳定与 PPO 信赖域违规联系起来,并提出可核对的稳定化组件,值得关注其机制分析及组件消融,但其 EEG/BCI 适用性尚未验证。
来源:OpenReview · Representation learning · openreview.net