arXiv:2505.19423cs.LGcs.AI2025-05被引 3

用自编码器与双曲神经网络提升强化学习进化算法的搜索效率。

Surrogate-Assisted Evolutionary Reinforcement Learning Based on Autoencoder and Hyperbolic Neural Network

  • 将高维策略压缩为低维嵌入,再用双曲神经网络构建代理模型预筛选。
  • 在10个Atari和4个Mujoco游戏中显著提升搜索效率与收敛速度。
  • 适合需要高效探索的复杂强化学习场景,如机器人控制与游戏智能体。

进化强化学习(ERL)通过进化算法训练强化学习策略,展现出比传统策略梯度更强的探索能力与鲁棒性。然而,ERL因进化算法需大量耗时模拟评估候选策略而面临高计算成本与低搜索效率问题,其中多数评估无效。为减少无效评估,可引入代理模型。但现有方法中策略多以深度神经网络形式存在,参数量达数百万,导致构建有效代理模型极为困难。本文提出一种新型代理辅助型ERL方法,融合自编码器(AE)与双曲神经网络(HNN)。AE将高维策略压缩为低维表示并提取关键特征作为代理输入;HNN作为基于分类的代理模型,能从采样数据中学习复杂非线性关系,实现无需真实评估的高效策略预筛选。在10个Atari和4个Mujoco游戏上的实验表明,该方法显著优于以往方法。可视化搜索轨迹显示,由AE与HNN引导的搜索在探索与收敛方面均更高效。本工作首次实现高维ERL策略的可学习嵌入与代理建模模块,并实证揭示其成功的关键条件与机制。

原文摘要 · Abstract (English)

Evolutionary Reinforcement Learning (ERL), training the Reinforcement Learning (RL) policies with Evolutionary Algorithms (EAs), have demonstrated enhanced exploration capabilities and greater robustness than using traditional policy gradient. However, ERL suffers from the high computational costs and low search efficiency, as EAs require evaluating numerous candidate policies with expensive simulations, many of which are ineffective and do not contribute meaningfully to the training. One intuitive way to reduce the ineffective evaluations is to adopt the surrogates. Unfortunately, existing ERL policies are often modeled as deep neural networks (DNNs) and thus naturally represented as high-dimensional vectors containing millions of weights, which makes the building of effective surrogates for ERL policies extremely challenging. This paper proposes a novel surrogate-assisted ERL that integrates Autoencoders (AE) and Hyperbolic Neural Networks (HNN). Specifically, AE compresses high-dimensional policies into low-dimensional representations while extracting key features as the inputs for the surrogate. HNN, functioning as a classification-based surrogate model, can learn complex nonlinear relationships from sampled data and enable more accurate pre-selection of the sampled policies without real evaluations. The experiments on 10 Atari and 4 Mujoco games have verified that the proposed method outperforms previous approaches significantly. The search trajectories guided by AE and HNN are also visually demonstrated to be more effective, in terms of both exploration and convergence. This paper not only presents the first learnable policy embedding and surrogate-modeling modules for high-dimensional ERL policies, but also empirically reveals when and why they can be successful.

强化学习进化算法代理模型双曲网络

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