Model Evolution Framework with Genetic Algorithm for Multi-Task Reinforcement Learning

TL;DR

This paper introduces MEGA, a genetic algorithm-based framework that dynamically evolves multi-task reinforcement learning models by adding modules based on task difficulty, improving adaptability and performance.

Contribution

The paper presents a novel model evolution framework using genetic algorithms for multi-task reinforcement learning, enabling dynamic resource allocation and model adaptation.

Findings

Achieved state-of-the-art results on Meta-World benchmark tasks.

Demonstrated effective model adaptation to task difficulty.

Showed improved learning efficiency through dynamic module addition.

Abstract

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.