Table of Contents
Fetching ...

Multi-Objective Alignment of Large Language Models Through Hypervolume Maximization

Subhojyoti Mukherjee, Anusha Lalitha, Sailik Sengupta, Aniket Deshmukh, Branislav Kveton

TL;DR

This work tackles multi-objective alignment of large language models (MOAHF) when human preferences are unknown or hard to quantify. It proposes HaM, an a-posteriori MO approach that learns $K$ diverse LLM policies by directly maximizing the hypervolume across $J$ objectives, ensuring coverage of the Pareto front. Key innovations include a shared-transformer backbone with per-policy heads, mini-batch based hypervolume estimation, and a random hypervolume scalarization option to reduce computational burden. Empirical results across harmlessness, helpfulness, humor, faithfulness, and hallucination demonstrate that HaM yields stronger and more diverse Pareto fronts than baselines like RiC and SCA, with robust performance in both offline and online fine-tuning regimes. Overall, HaM offers a scalable, principled pathway to balance conflicting human preferences in MOAHF while maintaining practical resource requirements.

Abstract

Multi-objective alignment from human feedback (MOAHF) in large language models (LLMs) is a challenging problem as human preferences are complex, multifaceted, and often conflicting. Recent works on MOAHF considered a-priori multi-objective optimization (MOO), where human preferences are known at training or inference time. In contrast, when human preferences are unknown or difficult to quantify, a natural approach is to cover the Pareto front by multiple diverse solutions. We propose an algorithm HaM for learning diverse LLM policies that maximizes their hypervolume. This is the first application of a-posteriori MOO to MOAHF. HaM is computationally and space efficient, and empirically superior across objectives such as harmlessness, helpfulness, humor, faithfulness, and hallucination, on various datasets.

Multi-Objective Alignment of Large Language Models Through Hypervolume Maximization

TL;DR

This work tackles multi-objective alignment of large language models (MOAHF) when human preferences are unknown or hard to quantify. It proposes HaM, an a-posteriori MO approach that learns diverse LLM policies by directly maximizing the hypervolume across objectives, ensuring coverage of the Pareto front. Key innovations include a shared-transformer backbone with per-policy heads, mini-batch based hypervolume estimation, and a random hypervolume scalarization option to reduce computational burden. Empirical results across harmlessness, helpfulness, humor, faithfulness, and hallucination demonstrate that HaM yields stronger and more diverse Pareto fronts than baselines like RiC and SCA, with robust performance in both offline and online fine-tuning regimes. Overall, HaM offers a scalable, principled pathway to balance conflicting human preferences in MOAHF while maintaining practical resource requirements.

Abstract

Multi-objective alignment from human feedback (MOAHF) in large language models (LLMs) is a challenging problem as human preferences are complex, multifaceted, and often conflicting. Recent works on MOAHF considered a-priori multi-objective optimization (MOO), where human preferences are known at training or inference time. In contrast, when human preferences are unknown or difficult to quantify, a natural approach is to cover the Pareto front by multiple diverse solutions. We propose an algorithm HaM for learning diverse LLM policies that maximizes their hypervolume. This is the first application of a-posteriori MOO to MOAHF. HaM is computationally and space efficient, and empirically superior across objectives such as harmlessness, helpfulness, humor, faithfulness, and hallucination, on various datasets.

Paper Structure

This paper contains 24 sections, 2 theorems, 32 equations, 8 figures, 6 tables.

Key Result

Proposition 1

Suppose that $\mathcal{V} = \{v_k\}_{k \in [K]}$ and $v_k = (\bar{\mathcal{L}}_j(\theta_k))_{j = 1}^J$. Then $\mathcal{L}_\textsc{ham}(\Theta) = \mathrm{vol}(\mathcal{V})$. $\mathcal{L}_\textsc{ham}(\Theta)$ is also monotone and submodular in $\mathcal{V}$.

Figures (8)

  • Figure 1: The shaded area depicts the union of the rectangles corresponding to two policies $\theta_1$ and $\theta_2$ in $J = 2$ dimensions. The gray line is the Pareto front.
  • Figure 2: Multi-headed architecture in \ref{['sec:policy representation']}.
  • Figure 3: Pareto fronts in the harmless-helpful task (\ref{['sec:harmless-helful task']}).
  • Figure 4: Pareto fronts in the harmless-humor task (\ref{['sec:harmless-humor task']}).
  • Figure 5: Pareto fronts in the faithful-hallucination task (\ref{['sec:faithful-hallucination task']}).
  • ...and 3 more figures

Theorems & Definitions (2)

  • Proposition 1
  • Theorem 1