利用离散信息瓶颈方法改进视觉强化学习
Improving Visual Reinforcement Learning with Discrete Information Bottleneck Approach
基于摘要分析。本文研究视觉强化学习中基于对比学习的状态表征在小批量训练下不稳定的问题,提出离散信息瓶颈方法 DIB,将离散化与信息瓶颈引入对比学习,以学习简洁的离散状态表征,并用于策略学习。 机制与研究动机:作者将既有方法的训练不稳定归因于对比学习对大批量的需求。摘要以 4096 或更大的批量为例,对比现有相关强化学习方法通常采用的 512 批量;这是作者对问题背景的描述,不能视为所有对比学习方法的通用要求,也不能据此确定 DIB 的实际实验批量。DIB 的核心思路是通过离散状态表征压缩视觉信息,但离散化方式、信息瓶颈目标、损失组合及策略训练细节尚未验证。 评估与结果:作者报告,在 DeepMind Control suite 的若干连续控制任务上,DIB 相比既有基于模型及无模型的视觉强化学习方法,具有更好的任务表现和样本效率,并在小批量条件下实现更稳定的训练。摘要未提供具体任务、基线名称、指标数值或重复实验统计;实验协议、消融及统计信息尚未验证。 平台推测(待验证):可检验的迁移假设是,离散信息瓶颈可能有助于在 EEG 闭环强化学习中压缩高维观测、降低小批量表征学习的波动。原方法依赖视觉观测与环境交互的连续控制结构,并非已验证的 EEG 解码方法;迁移需改造输入编码器及状态定义,且必须核对其对交互数据规模的依赖。低信噪比可能使离散化丢失弱任务信号,被试差异和通道变化也可能导致离散状态不稳定。一个小规模检验是在固定 EEG 闭环任务、数据预算和策略学习器的条件下,对比连续表征与 DIB 表征,观察多次运行的回报波动和样本效率。已有 EEG 相关工作尚未检索确认。
值得关注 DIB 如何用离散信息瓶颈缓解视觉强化学习中小批量对比学习的训练不稳定,但其具体实现、对照协议及向 EEG 场景的适用性尚未验证。
来源:OpenReview · Representation learning · openreview.net