Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Multi-Objective Reward and Preference Optimization: Theory and Algorithms

About

This thesis develops theoretical frameworks and algorithms that advance constrained reinforcement learning (RL) across control, preference learning, and alignment of large language models. The first contribution addresses constrained Markov Decision Processes (CMDPs) under the average-cost criterion through the Average-Constrained Policy Optimization (ACPO) algorithm. ACPO integrates sensitivity analysis with trust-region updates to ensure stable constraint handling, achieving state-of-the-art empirical performance with theoretical guarantees. Constrained RL is then extended to finite-horizon settings via e-COP, the first policy optimization method for episodic CMDPs. Built on an episodic policy difference lemma, e-COP offers provable performance, simplicity, and scalability in safety-critical environments. The thesis then investigates reinforcement learning from human preferences. warmPref-PS introduces a posterior sampling strategy for linear bandits that integrates offline preference data from heterogeneous raters into online learning. Explicit modeling of rater competence yields substantial regret reduction and more efficient data collection for RLHF. The PSPL algorithm further advances preference-based RL by jointly sampling reward models and transition dynamics from pairwise trajectory comparisons, providing Bayesian simple-regret guarantees and robust empirical identification of optimal policies. The final contribution applies these methods to large-scale model alignment. A multi-objective constrained optimization view yields MOPO, an iterative algorithm with closed-form updates that scales to multi-billion-parameter language models and remains robust across alignment settings. Collectively, the thesis unifies constrained RL across average-cost, episodic, and preference-driven paradigms, delivering theoretical advances and practical tools for safe and aligned decision-making.

Akhil Agnihotri• 2025

Related benchmarks

TaskDatasetResultRank
Helpful Assistant AlignmentHelpful Assistant normalized rewards (test)
Helpfulness Reward (r1)53
60
Reddit Summary AlignmentReddit Summary normalized rewards (test)
Faithfulness Reward0.55
60
Assistant Response Alignment (Helpfulness and Harmlessness)HH-RLHF (test)
Helpfulness Win Rate39
31
HelpfulnessAlpaca Eval
Alpaca Eval (%)16.33
22
Constrained Reinforcement LearningPointCircle
Episodic Reward110.5
8
Constrained Reinforcement LearningAntCircle
Episodic Reward198.6
8
Constrained Reinforcement LearningPointReach
Episodic Reward81.5
8
Constrained Reinforcement LearningGRID
Episodic Reward258.1
8
Constrained Reinforcement LearningBottleneck
Episodic Reward345.1
8
Constrained Reinforcement LearningHumanoid
Episodic Reward1.65e+3
8
Showing 10 of 13 rows

Other info

Follow for update