arXiv:2501.17384cs.LGcs.AI2025-01被引 3

通过双智能体对抗学习,提升强化学习模型的泛化能力。

A Dual-Agent Adversarial Framework for Robust Generalization in Deep Reinforcement Learning

  • 设计双智能体对抗机制,自动学习无关特征的鲁棒表示。
  • 在Procgen基准上显著提升硬任务表现,超越基线方法。
  • 适用于PPO等主流算法,无需人工先验知识。

近年来,得益于神经网络的强大能力,强化学习(RL)已成功解决诸多挑战性任务。然而,这些模型虽决策能力增强,却日益面临过拟合问题,例如在背景颜色等微小变化下便无法泛化。为此,我们提出一种双智能体对抗策略学习框架,使智能体在不引入人类先验知识的前提下,自发学习底层语义。该框架包含两个智能体之间的博弈:每个智能体通过生成相同状态下的表征差异来最大化对对手策略的干扰,同时保持自身对扰动的稳定性。这种交互促使智能体学习到可处理高维观测中无关特征的泛化策略。在Procgen基准上的大量实验表明,对抗过程显著提升了双方的泛化性能,并可应用于PPO等多种强化学习算法。使用该框架后,智能体在硬级别任务中显著优于基线方法,标志着深度强化学习泛化能力的重要进展。

原文摘要 · Abstract (English)

Recently, empowered with the powerful capabilities of neural networks, reinforcement learning (RL) has successfully tackled numerous challenging tasks. However, while these models demonstrate enhanced decision-making abilities, they are increasingly prone to overfitting. For instance, a trained RL model often fails to generalize to even minor variations of the same task, such as a change in background color or other minor semantic differences. To address this issue, we propose a dual-agent adversarial policy learning framework, which allows agents to spontaneously learn the underlying semantics without introducing any human prior knowledge. Specifically, our framework involves a game process between two agents: each agent seeks to maximize the impact of perturbing on the opponent's policy by producing representation differences for the same state, while maintaining its own stability against such perturbations. This interaction encourages agents to learn generalizable policies, capable of handling irrelevant features from the high-dimensional observations. Extensive experimental results on the Procgen benchmark demonstrate that the adversarial process significantly improves the generalization performance of both agents, while also being applied to various RL algorithms, e.g., Proximal Policy Optimization (PPO). With the adversarial framework, the RL agent outperforms the baseline methods by a significant margin, especially in hard-level tasks, marking a significant step forward in the generalization capabilities of deep reinforcement learning.

强化学习对抗学习泛化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。