atlas

Research map / Large language models / Alignment and preferences

Group Preference Optimization: Few-Shot Alignment of Large Language Models

ICLR 2024 · Large language models › Alignment and preferences · approach: Robust and calibrated DPO · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Group Preference Optimization (GPO), a meta-learned in-context transformer preference module attached to a frozen LLM that predicts group preferences from a few examples, and evaluates it on OpinionQA and GlobalOpinionQA alignment.

Problem

Aligning LLMs to the subjective preferences of many distinct groups when per-group preference data is scarce and per-group fine-tuning or prompting is expensive or ineffective.

Approach

Augment a base LLM with an independent in-context transformer, meta-trained across groups, that maps LLM embeddings of (prompt, response) pairs plus a few context preference labels to predicted group preferences; the predictor then drives re-ranking or policy optimization without updating the base LLM.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace GPO's architecture with a standard autoregressive transformer using a causal mask and positional encoding (ablation)GPO · GlobalOpinionQAAlignment Score: 0.798 ± 0.007 -> 0.780 ± 0.009; 0.820 ± 0.004 -> 0.782 ± 0.004; 0.799 ± 0.015 -> 0.772 ± 0.006Hurt
use only the last-token embedding instead of averaging token embeddings (ablation)GPO with Alpaca-7B embeddings · OpinionQA, meta train split of 80%Alignment Score: 0.946 ± 0.007 -> 0.903 ± 0.014Hurt
concatenate last-token and average-token embeddings instead of using average-token embeddings (ablation)GPO with Alpaca-7B embeddings · OpinionQA, meta train split of 80%Alignment Score: 0.946 ± 0.007 -> 0.942 ± 0.009Hurt
add group meta-data embeddings (ablation)GPO with Alpaca-7B · OpinionQA, meta train on 40% groupsAlignment Score: 0.920 ± 0.003 -> 0.900 ± 0.003Hurt
add group meta-data embeddings (ablation)GPO with Alpaca-7B · OpinionQA, meta train on 60% groupsAlignment Score: 0.926 ± 0.013 -> 0.916 ± 0.017Hurt
add group meta-data embeddings (ablation)GPO with Alpaca-7B · OpinionQA, meta train on 80% groupsAlignment Score: 0.946 ± 0.007 -> 0.926 ± 0.006Hurt
model target preference dependencies rather than assuming conditional independence (ablation)GPO · preliminary experimentsAlignment performanceNo clear effect
use GPO instead of In-Context FinetuneAlpaca-7B and Llama2-13B-chat · OpinionQAAlignment Score: 7.1%Helped
use GPO instead of the second-best baselineAlpaca-7B and Llama2-13B-chat · GlobalOpinionQAAlignment Score: 8.4%Helped
use GPO instead of In-Context FinetuneGPO and In-Context Finetune · GlobalOpinionQAtraining time: approximately 4.7 times more training timeHelped

Surprising findings

Limitations

Experimental setup

Datasets: OpinionQA GlobalOpinionQA

Benchmarks: OpinionQA GlobalOpinionQA

Metrics: alignment score (1-Wasserstein for OpinionQA, 1-Jensen-Shann individual alignment accuracy

Baselines: LM Base LM Steered Few-shot Prompt SFT per group Reward Model In-Context Finetune Uniform Distribution

Builds on: Neural processes / in-context meta-learning over datasets LoRA PPO DPO

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy