Multi-objective reinforcement learning (MORL) is a powerful tool to learn Pareto-optimal policy families across conflicting objectives. However, unlike traditional RL algorithms, existing MORL algorithms do not effectively leverage large-scale parallelization to concurrently simulate thousands of environments, resulting in vastly increased computation time. Ultimately, this has limited MORL's application towards complex multi-objective robotics problems. To address these challenges, we present 1) MORLAX, a new GPU-native, fast MORL algorithm, and 2) MO-Playground, a pip-installable playground of GPU-accelerated multi-objective environments. Together, MORLAX and MO-Playground approximate Pareto sets within minutes, offering 25-270x speed-ups compared to legacy CPU-based approaches whilst achieving superior Pareto front hypervolumes. We demonstrate the versatility of our approach by implementing a custom BRUCE humanoid robot environment using MO-Playground and learning Pareto-optimal locomotion policies across 6 realistic objectives for BRUCE, such as smoothness, efficiency and arm swinging.
Figures & tables
Framework
Real Robots
GPU
Multi-Obj.
Ref.
Gymnasium
✗
✗
✗
[ 25 , 26 ]
MO-Gymnasium
✗
✗
✓
[ 12 ]
MuJoCo Playground
✓
✓
✗
[ 19 ]
MO-Playground
✓
✓
✓
Ours
TABLE I : A comparison of open-source RL toolboxes for robotics.
Fig. 2 : MORLAX Architecture. (a) showcases evaluating the policy hypernetwork Hπ by picking a trade-off w , evaluating Hπ , and rolling out the extracted policy in the environment. (b) showcases how MORLAX trains Hπ via a parallelized sample - rollout - update procedure.
Fig. 3 : MORLaX finds Pareto sets that achieve better-performing, diverse behavior compared to the baseline, HYPER-MORL [ 16 ] , but significantly (21-270x) faster. To visualize each environment’s objectives, Pareto fronts along with composite figures of a single MORLAX policy hypernetwork are shown across extreme input trade-offs (e.g. maximum speed vs. maximum energy efficiency in MOCheetah). The displayed trajectories are generated via policies highlighted on the Pareto Fronts.
Environment
MORLAX
HYPER-MORL
Improvement
Maximum Hypervolume Possible
Hypervol. Increase
Cheetah
5.78⋅106
5.58⋅106
1.03x
Walker
6.48⋅106
5.28⋅106
1.23x
Ant
2.81⋅107
2.70⋅107
1.04x
Hopper
1.48⋅107
0.60⋅107
2.48x
Humanoid
5.45⋅106
0.74⋅105
7.33x
TABLE II : This table presents the comparative results between MORLAX and an existing MORL algorithm, HYPER-MORL [ 16 ] . Top : the maximum hypervolume each algorithm attained when run until convergence (5-45 min for MORLAX ). Bottom : the time taken for each algorithm to reach the lower of the two maximum hypervolumes in Top . Across all environments, MORLAX achieved larger hypervolumes faster than HYPER-MORL.
Fig. 4 : We project the learned six-dimensional Pareto front representing 30,720 policies tracking a moderate, forward velocity onto two 3D plots, each with different objectives. The colors represent trade-off vectors, and are blended according to the color-coded objectives along each axis. π1 represents a policy with arm swinging, π2 represents a policy with rigid arms, and π3 represents a policy with maximum smoothness. One interesting finding is that π1 walks faster and has higher efficiency, likely an emergent benefit of arm-swing.