atlas

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

Self-playing Adversarial Language Game Enhances LLM Reasoning

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: LLM unlearning and benchmarks · 8 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 SPAG, where LLMs play the adversarial Taboo word game against a copy of themselves, then are trained via imitation of GPT-4 games and offline RL on winning episodes, improving reasoning benchmarks over three epochs.

Problem

Improving the general reasoning ability of LLMs without relying on curated reasoning datasets or subjective self-judgments, which are costly and can amplify model biases.

Approach

Model the Adversarial Taboo game as a zero-sum two-player Markov game where an LLM plays both attacker and defender via self-play, with game outcomes (automatically checkable) providing rewards; winners' episodes are used for offline RL (ReST-style) plus an SFT regularizer.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Train with SPAG self-play through epoch 3 rather than use IM-AlpacaSFTLLaMA-2-7B · Reasoning benchmarks, GM (Avg.)geometric mean: 51.20 -> 52.58Helped
Train with SPAG self-play through epoch 3 rather than use IM+AlpacaSFTBaichuan-2-13B · Reasoning benchmarks, GM (Avg.)geometric mean: 55.80 -> 56.75Helped
Increase SPAG training from epoch 1 to epoch 3LLaMA-2-7B · Reasoning benchmarks, GM (Avg.)geometric mean: 51.69 -> 52.58Helped
Increase SPAG training from epoch 1 to epoch 3Baichuan-2-13B · Reasoning benchmarks, GM (Avg.)geometric mean: 56.09 -> 56.75Helped
Use SPAG self-play instead of continuous SFTLLaMA-2-7B IM model · Reasoning benchmarks, GM (Avg.)geometric mean: 51.20 vs 50.22Helped
Use SPAG self-play instead of continuous SFTBaichuan-2-13B IM model · Reasoning benchmarks, GM (Avg.)geometric mean: 56.09 vs 55.80Helped
Increase SPAG training from epoch 1 to epoch 3Baichuan-2-13B · MMLUaccuracy: 57.93 -> 57.75Hurt
Increase SPAG training from epoch 1 to epoch 3LLaMA-2-7B · WinoGrandeaccuracy: 68.90 -> 69.93Helped
Use SPAG-3 instead of AlpacaSFT-3 (ablation)LLaMA-2-7B · Reasoning benchmarksbenchmark scoresHelped
Increase imitation data beyond 5K episodes (ablation)GPT-4 imitation learning · Reasoning benchmarksreasoning performanceNo clear effect
Set imitation KL coefficient around 0.1 (ablation)GPT-4 imitation learning · Reasoning benchmarksreasoning performanceHelped
Increase SPAG SFT coefficient above 0.5 (ablation)SPAG · Reasoning benchmarksreasoning performanceNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: CoCA top-50K word frequency list Alpaca 52K instruction data GPT-4 self-play Adversarial Taboo episodes 20-Question keyword list (168 test words) BBH ARC Mutual WinoGrande LogiQA2 PIQA MMLU

Benchmarks: MMLU BBH Mutual ARC-e ARC-c LogiQA2 WinoGrande PIQA

Metrics: accuracy geometric mean across benchmarks game win rate

Baselines: Chain-of-Thought prompting AlpacaSFT Imitation-20Q Imitation-GuessCity SP-20Q SP-GuessCity IM+AlpacaSFT GPT-4

Builds on: ReST AlphaGo Zero Adversarial Taboo A-LoL PPO

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