Search papers, labs, and topics across Lattice.
This paper introduces MGDA-Decoupled, a geometry-aware multi-objective optimization algorithm designed to align large language models (LLMs) with human values by balancing conflicting objectives such as helpfulness, truthfulness, and harmlessness. Unlike traditional methods that rely on fixed scalarization or reinforcement learning, MGDA-Decoupled accounts for the convergence dynamics of each objective, promoting fairer trade-offs in the optimization process. Experimental results on the UltraFeedback dataset demonstrate that this approach significantly outperforms existing methods, achieving the highest win rates against golden responses across all objectives.
Geometry-aware optimization can dramatically improve LLM alignment by ensuring fairer trade-offs among conflicting human values.
Aligning large language models (LLMs) to desirable human values requires balancing multiple, potentially conflicting objectives such as helpfulness, truthfulness, and harmlessness, which presents a multi-objective optimisation challenge. Most alignment pipelines rely on a fixed scalarisation of these objectives, which can introduce procedural unfairness by systematically under-weighting harder-to-optimise or minority objectives. To promote more equitable trade-offs, we introduce MGDA-Decoupled, a geometry-based multi-objective optimisation algorithm that finds a shared descent direction while explicitly accounting for each objective's convergence dynamics. In contrast to prior methods that depend on reinforcement learning (e.g., GAPO) or explicit reward models (e.g., MODPO), our approach operates entirely within the lightweight Direct Preference Optimisation (DPO) paradigm. Experiments on the UltraFeedback dataset show that geometry-aware methods -- and MGDA-Decoupled in particular -- achieve the highest win rates against golden responses, both overall and per objective.
