atlas

Research map / Large language models / Language, safety and interpretability

The Consensus Game: Language Model Generation via Equilibrium Search

ICLR 2024 · Large language models › Language, safety and interpretability · approach: Controlled text generation · 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 equilibrium-ranking, a training-free game-theoretic LM decoding method that reconciles generative and discriminative scoring by computing regularized equilibria of a signaling game solved with piKL no-regret dynamics.

Problem

Language models give inconsistent answers when queried generatively (sampling or ranking by likelihood) versus discriminatively (scoring candidate correctness), and existing ensembling or deliberation fixes are poorly calibrated or costly.

Approach

Casts LM decoding as a regularized imperfect-information signaling game between a generator and discriminator, initialized from LM-derived policies (with a false-prompt contrast), and computes approximate regularized equilibria via piKL no-regret dynamics to obtain consensus predictions without training.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply ER-D instead of discriminative rankingLLaMA-7B · MMLUaccuracy: 40.4 -> 39.9No clear effect
Apply ER-D instead of discriminative rankingLLaMA-13B · MMLUaccuracy: 41.9 -> 45.1Helped
Apply ER-D instead of generative rankingLLaMA-7B · ARC-Easyaccuracy: 68.2 -> 71.5Helped
Apply ER-D instead of self-contrastive rankingLLaMA-13B · ARC-Challengeaccuracy: 59.3 -> 61.4Helped
Apply ER-D instead of self-contrastive rankingLLaMA-13B · RACE-Middleaccuracy: 64.8 -> 68.6Helped
Apply ER-D instead of self-contrastive rankingLLaMA-7B · RACE-Highaccuracy: 53.1 -> 56.4Helped
Apply ER-D instead of discriminative rankingLLaMA-7B · HHHaccuracy: 70.1 -> 71.5Helped
Apply default ER-D instead of discriminative rankingLLaMA-13B · HHHaccuracy: 69.2 -> 61.1Hurt
Use ER-G instead of self-contrastive rankingLLaMA-7B · TruthfulQABLEU-Acc: 34.91 ± 0.57 -> 34.61 ± 0.99Hurt
Use ER-G instead of discriminative rankingLLaMA-13B · TruthfulQABLEU-Acc: 39.05 ± 1.42 -> 39.83 ± 2.20Helped
Use ER-D instead of majority voteLLaMA-7B · GSM8Kaccuracy: 14.7 ± 0.2 -> 15.1 ± 0.6No clear effect
Use ER-D instead of majority voteLLaMA-13B · GSM8Kaccuracy: 22.5 ± 0.5 -> 23.0 ± 0.5No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: MMLU ARC RACE HHH TruthfulQA GSM8K

Benchmarks: MMLU ARC-Easy ARC-Challenge RACE-middle RACE-high HHH TruthfulQA GSM8K

Metrics: accuracy BLEU-Acc

Baselines: Generative Ranking Mutual Information Ranking Self-Contrastive Ranking Discriminative Ranking Greedy decoding Majority Vote LLaMA-65B PaLM-540B

Builds on: piKL counterfactual regret minimization contrastive decoding self-consistency

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