修复强化学习合成数据缺陷,让模型更准更稳
A Forensic Analysis of Synthetic Data in RL: Diagnosing and Solving Algorithmic Failures in Model-Based Policy Optimization
- 分离目标归一化与直接预测下一状态,修复模型偏差
- 在7个任务中5个超越SAC,同时保持原版高效表现
- 适合研究高效强化学习或模型误判问题的从业者
合成数据是高效动态型模型基础强化学习的核心,但可能降低性能。本文研究模型基础策略优化(MBPO)中的这一问题:尽管在OpenAI Gym上表现出色,但在DeepMind Control Suite(DMC)上常弱于非动态基线算法SAC,即使两者均基于MuJoCo的连续控制环境。我们发现两大耦合原因:动力学与奖励目标规模不匹配,抑制奖励学习并导致评价网络低估;残余下一状态预测导致模型方差增大,生成不可靠合成转移。为此提出FTFL——通过独立目标归一化与直接下一状态预测实现最小修复。FTFL在7个曾失败的任务中5个优于SAC,同时保留原版在Gym上的高效表现。进一步表明,包括基于模型不确定性的过滤、惩罚或拒绝机制在内的MBPO系列算法,若未对共享模型主干应用FTFL,仍会继承这些缺陷。本工作揭示基准测试局限性可能将环境特异性假设嵌入算法设计,呼吁建立马尔可夫决策过程结构与算法失效模式的映射分类体系。
原文摘要 · Abstract (English)
Synthetic data is central to data-efficient Dyna-style model-based reinforcement learning, but it can also degrade performance. We study this failure in Model-Based Policy Optimization (MBPO), which performs actor-critic updates using model-generated synthetic state transitions. Although MBPO reports strong sample-efficiency gains on OpenAI Gym, recent work shows that it often underperforms Soft Actor-Critic (SAC), its non-Dyna base, in the DeepMind Control Suite (DMC), despite both suites involving MuJoCo-based proprioceptive continuous control. We identify two coupled causes of this collapse: scale mismatch between dynamics and reward targets, which suppresses reward learning and induces critic underestimation, and residual next-state prediction, which inflates model variance and produces unreliable synthetic transitions. We introduce Fixing That Free Lunch (FTFL), a minimal repair that combines independent target normalization with direct next-state prediction. FTFL outperforms SAC in five of seven previously failing DMC tasks while preserving MBPO's strong Gym performance. We further show that MBPO-lineage algorithms, including uncertainty-aware variants that filter, penalize, or reject synthetic transitions based on model uncertainty, still inherit these failures unless FTFL is applied to their shared learned-model backbone. More broadly, our results show how benchmark-limited evaluation can encode environment-specific assumptions into algorithm design, motivating taxonomies that map MDP structure to algorithmic failure modes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。