atlas

Research map / Large language models / Alignment and preferences

Pragmatic Feature Preferences: Learning Reward-Relevant Preferences from Human Input

ICML 2024 · Large language models › Alignment and preferences · approach: Reward model design and preference models · 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 pragmatic feature preference queries that augment example-level pairwise comparisons with feature-level preferences and language-derived irrelevant-feature data augmentation to learn linear reward models more efficiently.

Problem

Reward models learned from example-level pairwise preferences (RLHF) lose fine-grained information about which features matter and why, making reward learning inefficient with limited comparisons.

Approach

Enrich binary preference queries with feature-level preferences and language descriptions of reward-relevant features; infer that unmentioned features are irrelevant and synthesize new preference pairs by swapping their values, with a joint loss combining example-level and feature-level Bradley-Terry terms.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Combine feature-level and example-level comparisons with pragmatic augmentationReward learning · Mushroom foragingprobability assigned to ground-truth best examplesHelped
Combine feature-level and example-level comparisons with pragmatic augmentationReward learning · Flight bookingprobability assigned to ground-truth best examplesHelped
Combine feature-level and example-level comparisons with pragmatic augmentationReward learning · Mushroom foraging, real user responsesprobability assigned to ground-truth best examplesHelped
Add feature-level comparisons to example-level comparisons (ablation)Pragmatic feature preference learning · Mushroom foragingprobability assigned to ground-truth best examplesHelped
Add pragmatic augmentation of comparisons using linguistic descriptions (ablation)Pragmatic feature preference learning · Mushroom foragingprobability assigned to ground-truth best examplesHelped
Use example-level comparisons only, without feature-level comparisons or pragmatic augmentation (ablation)Reward learning · Mushroom foragingprobability assigned to ground-truth best examplesHurt
Add feature-level comparisons to example-level comparisons (ablation)Pragmatic feature preference learning · Mushroom foraging, sparse reward featuresprobability assigned to ground-truth best examplesHelped
Use pragmatic feature preference queries rather than example-only queriesReward learning with real users · Mushroom foraging user studyperceived challenge describing preferences: t(11)=0.16, p=0.43No clear effect
Use pragmatic feature preference queries rather than example-only queriesReward learning with real users · Mushroom foraging user studyfrustration with providing labels: t(11)=-0.87, p=0.19No clear effect
Use pragmatic feature preference queries rather than feature-only queriesReward learning with real users · Mushroom foraging user studytime spent on task: t(11)= -0.24, p=0.41No clear effect
Use feature-only queries rather than example-only queries (ablation)Reward learning with real users · Mushroom foraging user studyperceived challenge describing preferences: t(11)=2.31, p=0.02Hurt
Use feature-only queries rather than example-only queries (ablation)Reward learning with real users · Mushroom foraging user studytime spent on task: t(11)=-1.86, p=0.03Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Mushroom foraging simulated task (own) Flight booking human descriptions (Lin et al. rewards-from-language dataset) Mushroom foraging user study responses (Prolific, 36 participants)

Benchmarks: mushroom foraging (simulated) flight booking (human language) mushroom foraging user study

Metrics: probability assigned to ground-truth best example Likert survey ratings (effort, frustration, communication) time on task

Baselines: RLHF (example-level comparisons) feature preference (FP) ablation pragmatic comparison (RLHF-prag) ablation

Builds on: RLHF / Bradley-Terry pairwise preference loss linear contextual bandits rewards-from-language (Lin et al.)

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