Preference Conditioned Multi-Objective Reinforcement Learning: Decomposed, Diversity-Driven Policy Optimization
Authors:
- Tanmay AmbadkarTATanmay Ambadkar
- Sourav PandaSPSourav Panda
- Shreyash KaleSKShreyash Kale
- Jonathan DodgeJDJonathan Dodge
- Abhinav VermaAVAbhinav Verma
Preprint
Abstract
Multi-objective reinforcement learning (MORL) seeks to learn policies that balance multiple, often conflicting objectives. Although a single preference-conditioned policy is the most flexible and scalable solution, existing approaches remain brittle in practice, frequently failing to recover complete Pareto fronts. We show that this failure stems from two structural issues in current methods: destructive gradient interference caused by premature scalarization and representational collapse across the preference space. We introduce $D^3PO$, a PPO-based framework that reorganizes multi-objective policy optimization to address these issues directly. $D^3PO$ preserves per-objective learning signals through a decomposed optimization pipeline and integrates preferences only after stabilization, enabling reliable credit assignment. In addition, a scaled diversity regularizer enforces sensitivity of policy behavior to preference changes, preventing collapse. Across standard MORL benchmarks, including high-dimensional and many-objective control tasks, $D^3PO$ consistently discovers broader and higher-quality Pareto fronts than prior single- and multi-policy methods, matching or exceeding state-of-the-art hypervolume and expected utility while using a single deployable policy.
Keywords
All Related Versions
This publication has 6 references indexed in Scilit:
- Multi-objective Reinforcement Learning with Nonlinear Preferences: Provable Approximation for Maximizing Expected Scalarized Return2025
- Pareto Set Learning for Multi-Objective Reinforcement LearningProceedings of the AAAI Conference on Artificial Intelligence, 2025
- PA2D-MORL: Pareto Ascent Directional Decomposition Based Multi-Objective Reinforcement LearningProceedings of the AAAI Conference on Artificial Intelligence, 2024
- Multi-Objective Reinforcement Learning Based on Decomposition: A Taxonomy and FrameworkJournal of Artificial Intelligence Research, 2024
- Latent-Conditioned Policy Gradient for Multi-Objective Deep Reinforcement Learning2023
- Sample-Efficient Multi-Objective Learning via Generalized Policy Improvement Prioritization2023