用遗传算法让模型自动进化,动态增删模块应对不同难度任务。
Model Evolution Framework with Genetic Algorithm for Multi-Task Reinforcement Learning
- 通过基因型策略控制模块增减,实现模型动态演化。
- 在Meta-World上达到当前最优性能,适应多任务复杂度差异。
- 适合需要自适应结构的多任务强化学习研究者使用。
多任务强化学习通过单一策略完成多种任务,旨在提升智能体在不同场景下的泛化能力。由于任务间存在共性,参数共享可提高学习效率。现有方法通常使用路由网络为每项任务生成特定路径,并重构一组模块以同时完成多个任务。然而,任务间的本质差异要求根据任务难度合理分配资源,这受到模型结构的限制。为此,我们提出基于遗传算法的模型演化框架(MEGA),使模型在训练过程中根据任务难度自动演化。当现有模型无法胜任某些任务时,框架将自动引入额外模块以增强能力。此外,为适配该框架,我们引入了模块级基因型模型,采用二进制序列作为基因型策略进行模型重构,并利用非梯度遗传算法优化这些策略。与输出维度固定的路由网络不同,本方法支持基因型策略长度的动态调整,可兼容模块数量变化的模型。我们在Meta-World基准上的各类机器人操作任务上进行了实验,结果表明MEGA框架具有显著有效性。源代码将公开发布。
原文摘要 · Abstract (English)
Multi-task reinforcement learning employs a single policy to complete various tasks, aiming to develop an agent with generalizability across different scenarios. Given the shared characteristics of tasks, the agent's learning efficiency can be enhanced through parameter sharing. Existing approaches typically use a routing network to generate specific routes for each task and reconstruct a set of modules into diverse models to complete multiple tasks simultaneously. However, due to the inherent difference between tasks, it is crucial to allocate resources based on task difficulty, which is constrained by the model's structure. To this end, we propose a Model Evolution framework with Genetic Algorithm (MEGA), which enables the model to evolve during training according to the difficulty of the tasks. When the current model is insufficient for certain tasks, the framework will automatically incorporate additional modules, enhancing the model's capabilities. Moreover, to adapt to our model evolution framework, we introduce a genotype module-level model, using binary sequences as genotype policies for model reconstruction, while leveraging a non-gradient genetic algorithm to optimize these genotype policies. Unlike routing networks with fixed output dimensions, our approach allows for the dynamic adjustment of the genotype policy length, enabling it to accommodate models with a varying number of modules. We conducted experiments on various robotics manipulation tasks in the Meta-World benchmark. Our state-of-the-art performance demonstrated the effectiveness of the MEGA framework. We will release our source code to the public.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。