跳到正文
OpenReview · State space models· Minsong Liu; Yuanheng Zhu; Yaran Chen; Dongbin Zhao·· 2024-01-01精选AI 评分70

通过基于 Transformer 的状态预测表征增强强化学习

Enhancing Reinforcement Learning via Transformer-Based State Predictive Representations

AI 导读

基于摘要分析。本文研究高维输入环境中强化学习(RL)样本效率偏低的问题,提出基于 Transformer 的状态预测表征辅助任务 TSPR,以自监督预测目标改善序列表征学习,并辅助策略训练;主要研究对象是连续与离散控制任务,而非 EEG 或 BCI。 核心机制是在潜在空间中预测状态表征:作者设计基于 Transformer 的预测模型,结合单向与双向预测任务,利用序列上下文学习更有信息量的状态表征,再与离策略 RL 算法结合。摘要未说明预测目标的具体构造、损失形式、序列长度、目标表征更新方式及与策略学习的训练配比,这些实现条件尚未验证。 作者报告,在 DMControl 连续控制与 Atari 离散控制任务中,TSPR 与离策略 RL 算法结合可提高样本效率,并优于所比较的先进样本高效 RL 方法。摘要未提供具体任务清单、交互预算、基线名称、数值结果或统计信息,因此尚不能核对优势幅度及其适用范围;实验协议、消融及统计信息尚未验证。作者表示将发布代码,材料不足以确认当前代码可用性。 平台推测(待验证):若面向具有连续交互轨迹的自适应 BCI,TSPR 的潜在状态预测辅助任务可能用于学习 EEG 时序上下文,但原领域结果不等于 BCI 有效。可复用的是序列预测辅助学习思路;需改造 EEG 编码器、预测时间尺度及策略接口,并明确轨迹、动作与奖励的来源。低信噪比、跨被试与通道差异、小数据条件可能使模型优先学习伪迹或个体特征。一个可检验的小实验是在固定数据划分和交互预算的 EEG 闭环任务中,比较相同 RL 基线加入与不加入 TSPR 的表现,并分别评估单向、双向预测的贡献,同时核对预测上下文与在线可用信息的一致性。已有 EEG 相关工作尚未检索确认。

阅读价值

值得关注其利用单向与双向潜在状态预测辅助离策略 RL 表征学习的机制,但样本效率优势的具体协议及其对 EEG/BCI 的适用性尚未验证。

来源:OpenReview · State space models · openreview.net